idk, probably something changed

Joseph Redmon

2015-01-31 0f1a31648c5292fa49b35eac90a2ee676d6c13e6

 src/crop_layer_kernels.cu

@@ -24,11 +24,16 @@
    output[count] = input[index];
}

extern "C" void forward_crop_layer_gpu(crop_layer layer, float *input)
extern "C" void forward_crop_layer_gpu(crop_layer layer, int train, float *input)
{
    int flip = (layer.flip && rand()%2);
    int dh = rand()%(layer.h - layer.crop_height);
    int dw = rand()%(layer.w - layer.crop_width);
    int dh = rand()%(layer.h - layer.crop_height + 1);
    int dw = rand()%(layer.w - layer.crop_width + 1);
    if(!train){
        flip = 0;
        dh = (layer.h - layer.crop_height)/2;
        dw = (layer.w - layer.crop_width)/2;
    }
    int size = layer.batch*layer.c*layer.crop_width*layer.crop_height;

    dim3 dimBlock(BLOCK, 1, 1);