手写数字识别——深度学习（C语言实现，pytorch训练）

用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);

下载源码