mirror of
https://github.com/pjreddie/darknet.git
synced 2023-08-10 21:13:14 +03:00
YOLOv3!!!! WOOOO!!!!
This commit is contained in:
commit
d3828827e7
@ -260,9 +260,6 @@ void forward_detection_layer_gpu(const detection_layer l, network net)
|
|||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
|
|
||||||
//float *in_cpu = calloc(l.batch*l.inputs, sizeof(float));
|
|
||||||
//float *truth_cpu = 0;
|
|
||||||
|
|
||||||
cuda_pull_array(net.input_gpu, net.input, l.batch*l.inputs);
|
cuda_pull_array(net.input_gpu, net.input, l.batch*l.inputs);
|
||||||
forward_detection_layer(l, net);
|
forward_detection_layer(l, net);
|
||||||
cuda_push_array(l.output_gpu, l.output, l.batch*l.outputs);
|
cuda_push_array(l.output_gpu, l.output, l.batch*l.outputs);
|
||||||
|
Loading…
Reference in New Issue
Block a user