CNN代码系列之训练初始化（三）

本博客为CNN卷积代码系列之训练初始化。

注意：本博客是系列博客，请链接上一博客http://blog.csdn.net/samylee/article/details/69471988

CNN.hpp定义网络参数：

#ifndef _CNN_HPP_
#define _CNN_HPP_

#include <vector>

namespace ANN {

#define width_image_input_CNN		32 //归一化图像宽
#define height_image_input_CNN		32 //归一化图像高
#define width_image_C1_CNN		28
#define height_image_C1_CNN		28
#define width_image_S2_CNN		14
#define height_image_S2_CNN		14
#define width_image_C3_CNN		10
#define height_image_C3_CNN		10
#define width_image_S4_CNN		5
#define height_image_S4_CNN		5
#define width_image_C5_CNN		1
#define height_image_C5_CNN		1
#define width_image_output_CNN		1
#define height_image_output_CNN		1

#define width_kernel_conv_CNN		5 //卷积核大小
#define height_kernel_conv_CNN		5
#define width_kernel_pooling_CNN	2
#define height_kernel_pooling_CNN	2
#define size_pooling_CNN		2

#define num_map_input_CNN		1 //输入层map个数
#define num_map_C1_CNN			6 //C1层map个数
#define num_map_S2_CNN			6 //S2层map个数
#define num_map_C3_CNN			16 //C3层map个数
#define num_map_S4_CNN			16 //S4层map个数
#define num_map_C5_CNN			120 //C5层map个数
#define num_map_output_CNN		10 //输出层map个数

#define num_patterns_train_CNN		60000 //训练模式对数(总数)
#define num_patterns_test_CNN		10000 //测试模式对数(总数)
#define num_epochs_CNN			100 //最大迭代次数
#define accuracy_rate_CNN		0.985 //要求达到的准确率
#define learning_rate_CNN		0.01 //学习率
#define eps_CNN				1e-8

#define len_weight_C1_CNN		150 //C1层权值数，(5*5*1)*6=150
#define len_bias_C1_CNN			6 //C1层阈值数，6
#define len_weight_S2_CNN		6 //S2层权值数,1*6=6
#define len_bias_S2_CNN			6 //S2层阈值数,6
#define len_weight_C3_CNN		2400 //C3层权值数，(5*5*6)*16=2400
#define len_bias_C3_CNN			16 //C3层阈值数,16
#define len_weight_S4_CNN		16 //S4层权值数，1*16=16
#define len_bias_S4_CNN			16 //S4层阈值数，16
#define len_weight_C5_CNN		48000 //C5层权值数，(5*5*16)*120=48000
#define len_bias_C5_CNN			120 //C5层阈值数，120
#define len_weight_output_CNN		1200 //输出层权值数，(1*120)*10=1200
#define len_bias_output_CNN		10 //输出层阈值数，10

#define num_neuron_input_CNN		1024 //输入层神经元数，(32*32)*1=1024
#define num_neuron_C1_CNN		4704 //C1层神经元数，(28*28)*6=4704
#define num_neuron_S2_CNN		1176 //S2层神经元数，(14*14)*6=1176
#define num_neuron_C3_CNN		1600 //C3层神经元数，(10*10)*16=1600
#define num_neuron_S4_CNN		400 //S4层神经元数，(5*5)*16=400
#define num_neuron_C5_CNN		120 //C5层神经元数，(1*1)*120=120
#define num_neuron_output_CNN		10 //输出层神经元数，(1*1)*10=10

class CNN {
public:
CNN();
~CNN();

void init(); //初始化，分配空间

protected:

double E_weight_C1[len_weight_C1_CNN];
double E_bias_C1[len_bias_C1_CNN];
double E_weight_S2[len_weight_S2_CNN];
double E_bias_S2[len_bias_S2_CNN];
double E_weight_C3[len_weight_C3_CNN];
double E_bias_C3[len_bias_C3_CNN];
double E_weight_S4[len_weight_S4_CNN];
double E_bias_S4[len_bias_S4_CNN];
double* E_weight_C5;
double* E_bias_C5;
double* E_weight_output;
double* E_bias_output;
};

}

#endif //_CNN_HPP_

funset.cpp中的cnn1.init()定义

void CNN::init()
{
int len1 = width_image_input_CNN * height_image_input_CNN * num_patterns_train_CNN;//训练集输入
data_input_train = new double[len1];
init_variable(data_input_train, -1.0, len1);//初始化为-1

int len2 = num_map_output_CNN * num_patterns_train_CNN;//训练集输出
data_output_train = new double[len2];
init_variable(data_output_train, -0.8, len2);//初始化-0.8

int len3 = width_image_input_CNN * height_image_input_CNN * num_patterns_test_CNN;//测试集输入
data_input_test = new double[len3];
init_variable(data_input_test, -1.0, len3);//初始化为-1

int len4 = num_map_output_CNN * num_patterns_test_CNN;//测试集输出
data_output_test = new double[len4];
init_variable(data_output_test, -0.8, len4);//初始化-0.8

std::fill(E_weight_C1, E_weight_C1 + len_weight_C1_CNN, 0.0);//初始化0.0
std::fill(E_bias_C1, E_bias_C1 + len_bias_C1_CNN, 0.0);
std::fill(E_weight_S2, E_weight_S2 + len_weight_S2_CNN, 0.0);
std::fill(E_bias_S2, E_bias_S2 + len_bias_S2_CNN, 0.0);
std::fill(E_weight_C3, E_weight_C3 + len_weight_C3_CNN, 0.0);
std::fill(E_bias_C3, E_bias_C3 + len_bias_C3_CNN, 0.0);
std::fill(E_weight_S4, E_weight_S4 + len_weight_S4_CNN, 0.0);
std::fill(E_bias_S4, E_bias_S4 + len_bias_S4_CNN, 0.0);
E_weight_C5 = new double[len_weight_C5_CNN];
std::fill(E_weight_C5, E_weight_C5 + len_weight_C5_CNN, 0.0);
E_bias_C5 = new double[len_bias_C5_CNN];
std::fill(E_bias_C5, E_bias_C5 + len_bias_C5_CNN, 0.0);
E_weight_output = new double[len_weight_output_CNN];
std::fill(E_weight_output, E_weight_output + len_weight_output_CNN, 0.0);
E_bias_output = new double[len_bias_output_CNN];
std::fill(E_bias_output, E_bias_output + len_bias_output_CNN, 0.0);

initWeightThreshold();//初始化权重
getSrcData();//载入数据
}

init_variable函数定义：

void CNN::init_variable(double* val, double c, int len)
{
//printf("%d\n", &val);
for (int i = 0; i < len; i++) {
val[i] = c;
}
}

未完待续。。。

任何问题请加唯一QQ2258205918（名称samylee）！