Joseph Redmon
2015-03-30 81751b47dd5d2e63f571f048bdd0a6a2a45617b0
src/crop_layer.h
@@ -2,6 +2,7 @@
#define CROP_LAYER_H
#include "image.h"
#include "params.h"
typedef struct {
    int batch;
@@ -17,10 +18,10 @@
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);
void forward_crop_layer(const crop_layer layer, network_state state);
#ifdef GPU
void forward_crop_layer_gpu(crop_layer layer, int train, float *input);
void forward_crop_layer_gpu(crop_layer layer, network_state state);
#endif
#endif