From c35fc135a198264d9c2c7935cf206955a511f1b8 Mon Sep 17 00:00:00 2001 From: Hossin Asaadi Date: Thu, 9 Jan 2025 08:08:22 -0500 Subject: [PATCH 1/2] Update coco.py --- mmdet/datasets/coco.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/mmdet/datasets/coco.py b/mmdet/datasets/coco.py index 1cf21c4e667..2a6d01db757 100644 --- a/mmdet/datasets/coco.py +++ b/mmdet/datasets/coco.py @@ -69,6 +69,10 @@ def load_data_list(self) -> List[dict]: # change with the order of the `classes` self.cat_ids = self.coco.get_cat_ids( cat_names=self.metainfo['classes']) + + if len(self.cat_ids) == 0: + raise Exception(f"Can't find classes in annotation file {self.metainfo['classes']}") + self.cat2label = {cat_id: i for i, cat_id in enumerate(self.cat_ids)} self.cat_img_map = copy.deepcopy(self.coco.cat_img_map) From 5cd69652bb8f2088dd575be5d9a4f440efb915a7 Mon Sep 17 00:00:00 2001 From: Hossin Asaadi Date: Thu, 9 Jan 2025 17:29:26 +0400 Subject: [PATCH 2/2] Update coco.py --- mmdet/datasets/coco.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/mmdet/datasets/coco.py b/mmdet/datasets/coco.py index 2a6d01db757..ebea7673e72 100644 --- a/mmdet/datasets/coco.py +++ b/mmdet/datasets/coco.py @@ -71,7 +71,8 @@ def load_data_list(self) -> List[dict]: cat_names=self.metainfo['classes']) if len(self.cat_ids) == 0: - raise Exception(f"Can't find classes in annotation file {self.metainfo['classes']}") + raise Exception(f"Can't find classes in \ + annotation file {self.metainfo['classes']}") self.cat2label = {cat_id: i for i, cat_id in enumerate(self.cat_ids)} self.cat_img_map = copy.deepcopy(self.coco.cat_img_map)