classifier.c - set batch=1 in "predict", "demo" and "try" mode.

vinjn

2018-05-26 6b6f0f0ca3193ff6707dd0acac8ccd3e3e3b3873

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
#ifndef IM2COL_H
#define IM2COL_H
 
void im2col_cpu(float* data_im,
        int channels, int height, int width,
        int ksize, int stride, int pad, float* data_col);
 
#ifdef GPU
 
void im2col_ongpu(float *im,
         int channels, int height, int width,
         int ksize, int stride, int pad,float *data_col);
 
#endif
#endif