yolo working w/ regions

Joseph Redmon

2015-09-16 c53e03348c65462bcba33f6352087dd6b9268e8f

 src/network_kernels.cu

@@ -134,6 +134,7 @@
    network_state state;
    int x_size = get_network_input_size(net)*net.batch;
    int y_size = get_network_output_size(net)*net.batch;
    if(net.layers[net.n-1].type == REGION) y_size = net.layers[net.n-1].truths*net.batch;
    if(!*net.input_gpu){
        *net.input_gpu = cuda_make_array(x, x_size);
        *net.truth_gpu = cuda_make_array(y, y_size);