Detection good, split up col images

Joseph Redmon

2014-12-04 1edcf73a73d2007afc61289245763f5cf0c29e10

 src/im2col.c

@@ -1,23 +1,23 @@
#include "mini_blas.h"
#include <stdio.h>
inline float im2col_get_pixel(float *im, int height, int width, int channels,
                        int b, int row, int col, int channel, int pad)
                        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+b*channels))];
    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 batch,
void im2col_cpu(float* data_im,
     int channels,  int height,  int width,
     int ksize,  int stride, int pad, float* data_col) 
{
    int c,h,w,b;
    int c,h,w;
    int height_col = (height - ksize) / stride + 1;
    int width_col = (width - ksize) / stride + 1;
    if (pad){
@@ -26,8 +26,6 @@
        pad = ksize/2;
    }
    int channels_col = channels * ksize * ksize;
    int col_size = height_col*width_col*channels_col;
    for (b = 0; b < batch; ++b) {
        for (c = 0; c < channels_col; ++c) {
            int w_offset = c % ksize;
            int h_offset = (c / ksize) % ksize;
@@ -36,10 +34,9 @@
                for (w = 0; w < width_col; ++w) {
                    int im_row = h_offset + h * stride;
                    int im_col = w_offset + w * stride;
                    int col_index = (c * height_col + h) * width_col + w + b*col_size;
                int col_index = (c * height_col + h) * width_col + w;
                    data_col[col_index] = im2col_get_pixel(data_im, height, width, channels,
                            b, im_row, im_col, c_im, pad);
                }
                        im_row, im_col, c_im, pad);
            }
        }
    }
@@ -74,7 +71,7 @@
}


void im2col_ongpu(cl_mem data_im,  int batch,
void im2col_ongpu(cl_mem data_im, int offset,
         int channels,  int height,  int width,
         int ksize,  int stride,  int pad, cl_mem data_col)
{
@@ -95,7 +92,7 @@

    cl_uint i = 0;
    cl.error = clSetKernelArg(kernel, i++, sizeof(data_im), (void*) &data_im);
    cl.error = clSetKernelArg(kernel, i++, sizeof(batch), (void*) &batch);
    cl.error = clSetKernelArg(kernel, i++, sizeof(offset), (void*) &offset);
    cl.error = clSetKernelArg(kernel, i++, sizeof(channels), (void*) &channels);
    cl.error = clSetKernelArg(kernel, i++, sizeof(height), (void*) &height);
    cl.error = clSetKernelArg(kernel, i++, sizeof(width), (void*) &width);
@@ -104,14 +101,15 @@
    cl.error = clSetKernelArg(kernel, i++, sizeof(data_col), (void*) &data_col);
    check_error(cl);

    size_t global_size = batch*channels_col*height_col*width_col;
    size_t global_size = channels_col*height_col*width_col;

    cl.error = clEnqueueNDRangeKernel(queue, kernel, 1, 0,
            &global_size, 0, 0, 0, 0);
    check_error(cl);
}

void im2col_gpu(float *data_im,  int batch,
/*
   void im2col_gpu(float *data_im,
         int channels,  int height,  int width,
         int ksize,  int stride,  int pad, float *data_col) 
{
@@ -144,5 +142,6 @@
    clReleaseMemObject(col_gpu);
    clReleaseMemObject(im_gpu);
}
 */

#endif