opencv3 ann 人工神经网络使用方法

//创建ANN
Ptr<ANN_MLP> bp = ANN_MLP::create();

// (a) 3层，输入层神经元个数为 4，隐层的为 6，输出层的为 4
Mat layers_size = (Mat_<int>(1,3) << 4,6,4);

// (b) 4层，输入层神经元个数为 4，第一个隐层的为 6，第二个隐层的为 5，输出层的为 4
Mat layers_size = (Mat_<int>(1,4) << 4,6,5,4);

//设置层数
 Mat layerSizes = (Mat_<int>(1, 4) << image_rows*image_cols, int(image_rows*image_cols / 2), int(image_rows*image_cols / 2), class_num);
 bp->setLayerSizes(layerSizes);

设置层数的时候第一项为特征的维数，输出层神经元的个数为种类的个数。

下面设置各种参数：

bp->setActivationFunction(ANN_MLP::SIGMOID_SYM, 1, 1);
void setActivationFunction(int _activ_func, double _f_param1, double _f_param2 );

//终止训练的条件
bp->setTermCriteria(TermCriteria(TermCriteria::MAX_ITER + TermCriteria::EPS, 300, FLT_EPSILON));
bp->setTrainMethod(ANN_MLP::BACKPROP, 0.001);

Ptr<TrainData> tData = TrainData::create(DataMat, ROW_SAMPLE, labelsMat);

bp->train(tData);

float response = ann->predict(testMat);

//保存模型
bp->save("bp_param");

参考资料：https://www.cnblogs.com/xinxue/archive/2017/06/27/5789421.html