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

下载源码

更多推荐