OpenCV 3.0 无法加载神经网络

Question

我需要在OpenCV（3.0版）项目中使用神经网络。我已经创建并训练了神经网络，它可以工作，但如果我想从YML文件加载神经网络，则它无法预测。

这是我创建、训练和保存神经网络的代码：

FileStorage fs("nn.yml", FileStorage::WRITE);
int input_neurons = 7;
int hidden_neurons = 100;
int output_neurons = 5;
Ptr<TrainData> train_data = TrainData::loadFromCSV("data.csv", 10, 7, 12);
Ptr<ANN_MLP> neural_network = ANN_MLP::create();
neural_network->setTrainMethod(ANN_MLP::BACKPROP);
neural_network->setBackpropMomentumScale(0.1);
neural_network->setBackpropWeightScale(0.05);
neural_network->setTermCriteria(TermCriteria(TermCriteria::MAX_ITER, (int)10000, 1e-6));
Mat layers = Mat(3, 1, CV_32SC1);
layers.row(0) = Scalar(input_neurons);
layers.row(1) = Scalar(hidden_neurons);
layers.row(2) = Scalar(output_neurons);
neural_network->setLayerSizes(layers);
neural_network->setActivationFunction(ANN_MLP::SIGMOID_SYM, 1, 1); 
neural_network->train(train_data);
if (neural_network->isTrained()) { 
    neural_network->write(fs);
    cout << "It's OK!" << endl; 
}

但下次，如果我想从YML文件加载它：

Ptr<ANN_MLP> neural_network = Algorithm::load<ANN_MLP>("nn.yml", "neural_network");

我得到输出：

[-1.#IND，-1.#IND

[-1.#IND，1.0263158，1.02631581.0263158]

[1.0263158，1.0263158，102063158，10263158，102.63158]

[-1.#IND，-1.#IND

Ptr<ANN_MLP> neural_network = Algorithm::load<ANN_MLP>("nn.yml");

这一行导致我得到一个错误：

OpenCV错误：未指定的错误（节点既不是映射也不是空集合on），文件C:\builds\master_PackSlave-win64-vc12-shared\opencv\modules\core\src\persistence.cpp，第739行

我做错了什么？问题出在哪里？

Answer 1

您可以使用save和load，或者write和read，但不应将它们混合使用。

所以你要么需要做：

// Save
neural_network->save("nn.yml");
// Load
Ptr<ANN_MLP> nn = Algorithm::load<ANN_MLP>("nn.yml");

或：

// Write
neural_network->write(fs);
// Read
FileStorage ffs("nn.yml", FileStorage::READ);
Ptr<ANN_MLP> nn = Algorithm::read<ANN_MLP>(ffs.root());