1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
| #ifndef CROP_LAYER_H
| #define CROP_LAYER_H
|
| #include "image.h"
|
| typedef struct {
| int batch;
| int h,w,c;
| int crop_width;
| int crop_height;
| int flip;
| float *output;
| #ifdef GPU
| float *output_gpu;
| #endif
| } crop_layer;
|
| image get_crop_image(crop_layer layer);
| crop_layer *make_crop_layer(int batch, int h, int w, int c, int crop_height, int crop_width, int flip);
| void forward_crop_layer(const crop_layer layer, int train, float *input);
|
| #ifdef GPU
| void forward_crop_layer_gpu(crop_layer layer, int train, float *input);
| #endif
|
| #endif
|
|
