用pytorch训练手写识别的网络模型出来,导出训练好的参数,再用C语言编写深度神经网络的代码,将训练好的参数导入进去,对图片进行识别。
// 卷积
void conv2d_f32(float* dout,
float* din,
int din_hgt,
int din_wid,
const float* ker,
const float* bias,
const int* shape);
// 全连接
void fc_32(float* dout,
float* din,
const float* weight,
const float* bias,
const int* shape);
// ReLU激活
void relu(float* dout,
float* din,
int size);
// 最大池化
void maxpool2d(float* dout,
float* din,
int din_hgt,
int din_wid,
int num_c,
int ksize);
// 展平为一维
void flatten(float* dout,
float* din,
int out_len,
int in_size,
int num_c);