| | |
| | | return float_to_image(w,h,c,layer.output); |
| | | } |
| | | |
| | | crop_layer *make_crop_layer(int batch, int h, int w, int c, int crop_height, int crop_width, int flip) |
| | | crop_layer *make_crop_layer(int batch, int h, int w, int c, int crop_height, int crop_width, int flip, float angle) |
| | | { |
| | | fprintf(stderr, "Crop Layer: %d x %d -> %d x %d x %d image\n", h,w,crop_height,crop_width,c); |
| | | crop_layer *layer = calloc(1, sizeof(crop_layer)); |
| | |
| | | layer->w = w; |
| | | layer->c = c; |
| | | layer->flip = flip; |
| | | layer->angle = angle; |
| | | layer->crop_width = crop_width; |
| | | layer->crop_height = crop_height; |
| | | layer->output = calloc(crop_width*crop_height * c*batch, sizeof(float)); |