Update yolov7_l_syncbn_fast_8x16b-300e_coco.py (#333)

change:model->bbox_head->head_module->dict(num_classes=num_classes)
2022-12-02 16:31:27 +08:00 · 2022-12-02 16:31:27 +08:00 · 9985e0bc08
parent 27487fd587
commit 9985e0bc08
1 changed files with 1 additions and 1 deletions
--- a/configs/yolov7/yolov7_l_syncbn_fast_8x16b-300e_coco.py
+++ b/configs/yolov7/yolov7_l_syncbn_fast_8x16b-300e_coco.py
@ -67,7 +67,7 @@ model = dict(
        type='YOLOv7Head',
        head_module=dict(
            type='YOLOv7HeadModule',
-            num_classes=80,
+            num_classes=num_classes,
            in_channels=[256, 512, 1024],
            featmap_strides=strides,
            num_base_priors=3),