From bd765e96495d80d02932471c5950d792b3557c8e Mon Sep 17 00:00:00 2001 From: Takehiro Ohtani Date: Fri, 25 Oct 2019 17:04:11 +0900 Subject: [PATCH] fix separate use list, set --- lmnet/lmnet/datasets/mscoco.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/lmnet/lmnet/datasets/mscoco.py b/lmnet/lmnet/datasets/mscoco.py index fd4b6559d..fa79be244 100644 --- a/lmnet/lmnet/datasets/mscoco.py +++ b/lmnet/lmnet/datasets/mscoco.py @@ -198,7 +198,7 @@ def coco(self): def _image_ids(self): """Return all files and gt_boxes list.""" classes = [class_name for class_name in self.classes if class_name is not "__background__"] - target_class_ids = self.coco.getCatIds(catNms=classes) + target_class_ids = set(self.coco.getCatIds(catNms=classes)) image_ids = [] for target_class_id in target_class_ids: target_image_ids = self.coco.getImgIds(catIds=[target_class_id]) @@ -225,7 +225,7 @@ def coco_category_id_to_lmnet_class_id(self, cat_id): @functools.lru_cache(maxsize=None) def _gt_boxes_from_image_id(self, image_id): """Return gt boxes list ([[x, y, w, h, class_id]]) of a image.""" - classes = set(class_name for class_name in self.classes if class_name is not "__background__") + classes = [class_name for class_name in self.classes if class_name is not "__background__"] class_ids = set(self.coco.getCatIds(catNms=classes)) boxes = []