diff --git a/configs/yolov7/yolov7_l_syncbn_fast_8x16b-300e_coco.py b/configs/yolov7/yolov7_l_syncbn_fast_8x16b-300e_coco.py index 5c4e5141..659f3a0f 100644 --- 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),