Merge branch 'darknetparser' of https://github.com/ceccocats/tkDNN into darknetparser

2020-05-30 19:38:31 +02:00
parent 0ae96a6bc4 d45947eec5
commit 0621ed223b
1 changed files with 2 additions and 0 deletions
@@ -102,6 +102,8 @@ namespace tk { namespace dnn {
            fields.classes = std::stoi(value);
        else if(name.find("num") !=  std::string::npos)
            fields.num = std::stoi(value);
+        else if(name.find("groups") !=  std::string::npos)
+            fields.groups = std::stoi(value);
        else if(name.find("scale_xy") !=  std::string::npos)
            fields.scale_xy = std::stof(value);
        else if(name.find("from") !=  std::string::npos)