update num_things_classes=80 for the new model file

dustysys · May 4, 2023 · 754c0b3 · 754c0b3
1 parent afeb6aa
commit 754c0b3
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/config/mask2former_r50_8xb2-lsj-50e_coco-panoptic.py b/config/mask2former_r50_8xb2-lsj-50e_coco-panoptic.py
@@ -24,7 +24,7 @@
     batch_augments=batch_augments,
 )
 
-num_things_classes = 1
+num_things_classes = 80
 num_stuff_classes = 0
 num_classes = num_things_classes + num_stuff_classes
 model = dict(

diff --git a/config/mmdet_dd-person_mask2former-v3.py b/config/mmdet_dd-person_mask2former-v3.py
@@ -1,6 +1,6 @@
 _base_ = ['./mask2former_r50_8xb2-lsj-50e_coco-panoptic.py']
 
-num_things_classes = 1
+num_things_classes = 80
 num_stuff_classes = 0
 num_classes = num_things_classes + num_stuff_classes
 image_size = (1024, 1024)