From 754c0b3601d53a7f883aedecfd6fa4ebfbf339b2 Mon Sep 17 00:00:00 2001 From: Won-Kyu Park Date: Fri, 5 May 2023 00:05:08 +0900 Subject: [PATCH] update num_things_classes=80 for the new model file --- config/mask2former_r50_8xb2-lsj-50e_coco-panoptic.py | 2 +- config/mmdet_dd-person_mask2former-v3.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/config/mask2former_r50_8xb2-lsj-50e_coco-panoptic.py b/config/mask2former_r50_8xb2-lsj-50e_coco-panoptic.py index b67d9b0..882ef7b 100644 --- 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 index 375d45d..ab1cfc3 100644 --- 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)