From 041cffd9b87d39229b90be7c6882cefbb8e4b674 Mon Sep 17 00:00:00 2001 From: AlexeyAB Date: Fri, 11 Jan 2019 01:05:44 +0300 Subject: [PATCH] Compile fix --- include/darknet.h | 3 ++- src/detector.c | 1 + 2 files changed, 3 insertions(+), 1 deletion(-) diff --git a/include/darknet.h b/include/darknet.h index 626e5277..c312be9c 100644 --- a/include/darknet.h +++ b/include/darknet.h @@ -753,7 +753,8 @@ LIB_API void reset_rnn(network *net); LIB_API float *network_predict_image(network *net, image im); LIB_API float validate_detector_map(char *datacfg, char *cfgfile, char *weightfile, float thresh_calc_avg_iou, const float iou_thresh, network *existing_net); LIB_API void train_detector(char *datacfg, char *cfgfile, char *weightfile, int *gpus, int ngpus, int clear, int dont_show, int calc_map); -LIB_API void test_detector(char *datacfg, char *cfgfile, char *weightfile, char *filename, float thresh, int ext_output); +LIB_API void test_detector(char *datacfg, char *cfgfile, char *weightfile, char *filename, float thresh, + float hier_thresh, int dont_show, int ext_output, int save_labels, char *outfile); LIB_API int network_width(network *net); LIB_API int network_height(network *net); LIB_API void optimize_picture(network *net, image orig, int max_layer, float scale, float rate, float thresh, int norm); diff --git a/src/detector.c b/src/detector.c index 2ae92ea7..455366e3 100644 --- a/src/detector.c +++ b/src/detector.c @@ -1,3 +1,4 @@ +#include "darknet.h" #include "network.h" #include "region_layer.h" #include "cost_layer.h"