Add group field

Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
2020-05-30 19:03:29 +02:00
parent e5e6654b1d
commit d45947eec5
1 changed files with 2 additions and 0 deletions
@@ -101,6 +101,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)