Additional fixes

This commit is contained in:
AlexeyAB 2017-10-29 18:58:07 +03:00
parent 84cdbaa1f1
commit 6ccb41808c
4 changed files with 6 additions and 4 deletions

View File

@ -1,6 +1,8 @@
darknet.exe detector test data/voc.data yolo-voc.cfg yolo-voc.weights 009460.jpg
darknet.exe detector test data/voc.data yolo-voc.cfg yolo-voc.weights -i 0 -thresh 0.2
darknet.exe detector test data/voc.data yolo-voc.cfg yolo-voc.weights -i 0 -thresh 0.1 dogr.jpg
pause

View File

@ -1,6 +1,6 @@
./darknet detector test ./cfg/voc.data ./cfg/yolo-voc.cfg ./yolo-voc.weights dog.jpg -i 0 -thresh 0.2
./darknet detector test ./cfg/voc.data ./cfg/yolo-voc.cfg ./yolo-voc.weights data/dog.jpg -i 0 -thresh 0.2

View File

@ -85,7 +85,7 @@ int main(int argc, char *argv[])
std::string filename;
if (argc > 1) filename = argv[1];
Detector detector("yolo-voc.cfg", "yolo-voc.weights");
Detector detector("cfg/yolo-voc.cfg", "yolo-voc.weights");
auto obj_names = objects_names_from_file("data/voc.names");
std::string out_videofile = "result.avi";

View File

@ -54,7 +54,7 @@ YOLODLL_API Detector::Detector(std::string cfg_filename, std::string weight_file
char *cfgfile = const_cast<char *>(cfg_filename.data());
char *weightfile = const_cast<char *>(weight_filename.data());
net = parse_network_cfg(cfgfile);
net = parse_network_cfg_custom(cfgfile, 1);
if (weightfile) {
load_weights(&net, weightfile);
}