| | |
|---|
| | | #include "mini_blas.h" |
|---|
| | | |
|---|
| | | inline 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 + channel*height)]; |
|---|
| | | } |
|---|
| | | |
|---|
| | | //From Berkeley Vision's Caffe! |
|---|
| | | //https://github.com/BVLC/caffe/blob/master/LICENSE |
|---|
| | | void im2col_cpu(float* data_im, |
|---|
| | | const int batch, const int channels, const int height, const int width, |
|---|
| | | const int ksize, const int stride, float* data_col) |
|---|
| | | const int ksize, const int stride, int pad, float* data_col) |
|---|
| | | { |
|---|
| | | int c,h,w,b; |
|---|
| | | int height_col = (height - ksize) / stride + 1; |
|---|
| | | int width_col = (width - ksize) / stride + 1; |
|---|
| | | if (pad){ |
|---|
| | | height_col = 1 + (height-1) / stride; |
|---|
| | | width_col = 1 + (width-1) / stride; |
|---|
| | | pad = ksize/2; |
|---|
| | | } |
|---|
| | | int channels_col = channels * ksize * ksize; |
|---|
| | | int im_size = height*width*channels; |
|---|
| | | int col_size = height_col*width_col*channels_col; |
|---|
| | |
|---|
| | | int c_im = c / ksize / ksize; |
|---|
| | | for ( h = 0; h < height_col; ++h) { |
|---|
| | | for ( w = 0; w < width_col; ++w) { |
|---|
| | | int im_row = h_offset + h * stride; |
|---|
| | | int im_col = w_offset + w * stride; |
|---|
| | | data_col[(c * height_col + h) * width_col + w] = |
|---|
| | | data_im[(c_im * height + h * stride + h_offset) * width |
|---|
| | | + w * stride + w_offset]; |
|---|
| | | im2col_get_pixel(data_im, height, width, channels, |
|---|
| | | im_row, im_col, c_im, pad); |
|---|
| | | } |
|---|
| | | } |
|---|
| | | } |
|---|
