src/detection_layer.h
@@ -11,6 +11,8 @@ int background; int rescore; int nuisance; int does_cost; float *cost; float *output; float *delta; #ifdef GPU @@ -23,6 +25,7 @@ void forward_detection_layer(const detection_layer layer, network_state state); void backward_detection_layer(const detection_layer layer, network_state state); int get_detection_layer_output_size(detection_layer layer); int get_detection_layer_locations(detection_layer layer); #ifdef GPU void forward_detection_layer_gpu(const detection_layer layer, network_state state);