diff --git a/include/tkDNN/DarknetParser.h b/include/tkDNN/DarknetParser.h index d92f671..9a1a500 100644 --- a/include/tkDNN/DarknetParser.h +++ b/include/tkDNN/DarknetParser.h @@ -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)