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

@ -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);
}