Additional fixes

pull/328/head
AlexeyAB 8 years ago
parent 84cdbaa1f1
commit 6ccb41808c
  1. 4
      build/darknet/x64/darknet_voc.cmd
  2. 2
      image_voc.sh
  3. 2
      src/yolo_console_dll.cpp
  4. 2
      src/yolo_v2_class.cpp

@ -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 pause

@ -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

@ -85,7 +85,7 @@ int main(int argc, char *argv[])
std::string filename; std::string filename;
if (argc > 1) filename = argv[1]; 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"); auto obj_names = objects_names_from_file("data/voc.names");
std::string out_videofile = "result.avi"; std::string out_videofile = "result.avi";

@ -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 *cfgfile = const_cast<char *>(cfg_filename.data());
char *weightfile = const_cast<char *>(weight_filename.data()); char *weightfile = const_cast<char *>(weight_filename.data());
net = parse_network_cfg(cfgfile); net = parse_network_cfg_custom(cfgfile, 1);
if (weightfile) { if (weightfile) {
load_weights(&net, weightfile); load_weights(&net, weightfile);
} }

Loading…
Cancel
Save