src/crop_layer.c
@@ -33,7 +33,7 @@ l.output = calloc(crop_width*crop_height * c*batch, sizeof(float)); #ifdef GPU l.output_gpu = cuda_make_array(l.output, crop_width*crop_height*c*batch); l.rand_gpu = cuda_make_array(0, l.batch*8); l.rand_gpu = cuda_make_array(0, l.batch*8); #endif return l; } @@ -48,6 +48,10 @@ int dw = rand()%(l.w - l.crop_width + 1); float scale = 2; float trans = -1; if(l.noadjust){ scale = 1; trans = 0; } if(!state.train){ flip = 0; dh = (l.h - l.crop_height)/2;