深度学习之卷积操作中的im2col-代码篇

最新推荐文章于 2025-06-25 19:25:42 发布

大腿壮

最新推荐文章于 2025-06-25 19:25:42 发布

阅读量1.4k

点赞数 1

CC 4.0 BY-SA版权

分类专栏：深度学习

本文链接：https://blog.csdn.net/qq_41092190/article/details/105749622

本文详细解读了深度学习中用于卷积运算的img2col函数，通过代码展示了如何将3D输入矩阵转换为2D矩阵，然后进行卷积计算。文章深入探讨了函数内部的三个for循环的逻辑，解释了各变量的意义，帮助理解卷积过程中的数据映射关系。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

前一篇博客
中介绍了img2col的原理好理论实现方式，本篇结合代码来剖析该方法。
img2col函数的代码如下所示，下面来分析代码。

#include "im2col.h"
#include <stdio.h>
float im2col_get_pixel(float *im, int height, int width, int channels,
                        int row, int col, int channel, int pad)
{
   
   
    row -= pad;
    col -= pad;

    if (row < 0 || col < 0 ||
        row >= height || col >= width) return 0;
    return im[col + width*(row + height*channel)];
}

//From Berkeley Vision's Caffe!
//https://github.com/BVLC/caffe/blob/master/LICENSE
void im2col_cpu(float* data_im,
     int channels,  int height,