🐛 Fix cfg.model.roi_head.bbox.num_classes on train.py

boostcampaitech4lv23cv3 · Nov 28, 2022 · 4a879fe · 4a879fe
1 parent e75ad1a
commit 4a879fe
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/mmdetection/tools/train.py b/mmdetection/tools/train.py
@@ -205,7 +205,8 @@ def main():
     logger = get_root_logger(log_file=log_file, log_level=cfg.log_level)
 
     # set num_classes
-    cfg.model.roi_head.bbox_head.num_classes = len(cfg.classes)
+    for bbox_head in cfg.model.roi_head.bbox_head:
+        bbox_head.num_classes = len(cfg.classes)
 
     # init the meta dict to record some important information such as
     # environment info and seed, which will be logged