mmr/tkDNN
Archived
1
0
Fork 0
Code Issues 46 Pull Requests 12 Releases 4 Activity

Check

Browse Source 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
This commit is contained in:
Micaela Verucchi
2020-06-01 10:39:13 +02:00
parent fe07c8fb26
commit 9ee8c5eb58
1 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
include/tkDNN/DarknetParser.h
+4 -1
View File
@@ -181,6 +181,8 @@ namespace tk { namespace dnn {
std::string wgs = wgs_path + "/g" + std::to_string(netLayers.size()) + ".bin"; std::string wgs = wgs_path + "/g" + std::to_string(netLayers.size()) + ".bin";
printf("%d %d %s %d %f\n", f.classes, f.num/f.n_mask, wgs.c_str(), f.n_mask, f.scale_xy); printf("%d %d %s %d %f\n", f.classes, f.num/f.n_mask, wgs.c_str(), f.n_mask, f.scale_xy);
tk::dnn::Yolo *l = new tk::dnn::Yolo(net, f.classes, f.num/f.n_mask, wgs, f.n_mask, f.scale_xy); tk::dnn::Yolo *l = new tk::dnn::Yolo(net, f.classes, f.num/f.n_mask, wgs, f.n_mask, f.scale_xy);
if(names.size() != f.classes)
FatalError("Mismatch between number of classes and names");
l->classesNames = names; l->classesNames = names;
netLayers.push_back(l); netLayers.push_back(l);
@@ -197,7 +199,8 @@ namespace tk { namespace dnn {
std::vector<std::string> names; std::vector<std::string> names;
std::string line; std::string line;
while(std::getline(if_names, line)) while(std::getline(if_names, line))
names.push_back(line); if(line != "")
names.push_back(line);
if_names.close(); if_names.close();
return names; return names;