| | |
| | | #ifndef CROP_LAYER_H |
| | | #define CROP_LAYER_H |
| | | |
| | | #include "opencl.h" |
| | | #include "image.h" |
| | | #include "layer.h" |
| | | #include "network.h" |
| | | |
| | | typedef struct { |
| | | int batch; |
| | | int h,w,c; |
| | | int crop_width; |
| | | int crop_height; |
| | | int flip; |
| | | float *output; |
| | | #ifdef GPU |
| | | cl_mem output_cl; |
| | | #endif |
| | | } crop_layer; |
| | | typedef layer 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, float *input); |
| | | image get_crop_image(crop_layer l); |
| | | crop_layer make_crop_layer(int batch, int h, int w, int c, int crop_height, int crop_width, int flip, float angle, float saturation, float exposure); |
| | | void forward_crop_layer(const crop_layer l, network_state state); |
| | | void resize_crop_layer(layer *l, int w, int h); |
| | | |
| | | #ifdef GPU |
| | | void forward_crop_layer_gpu(crop_layer layer, cl_mem input); |
| | | void forward_crop_layer_gpu(crop_layer l, network_state state); |
| | | #endif |
| | | |
| | | #endif |