From 538ec219c68bc0a9b7ce994d5f2f06c9b76f7786 Mon Sep 17 00:00:00 2001 From: akalankag Date: Thu, 6 Oct 2022 11:48:16 -0600 Subject: [PATCH] dict comprehension is faster than invoking a dict constructor --- lib/dataset/COCODataset.py | 9 +++------ lib/dataset/CrowdPoseDataset.py | 8 +++----- 2 files changed, 6 insertions(+), 11 deletions(-) diff --git a/lib/dataset/COCODataset.py b/lib/dataset/COCODataset.py index 265c54a..f653851 100644 --- a/lib/dataset/COCODataset.py +++ b/lib/dataset/COCODataset.py @@ -58,13 +58,10 @@ def __init__(self, root, dataset, data_format, transform=None, self.num_classes = len(self.classes) self._class_to_ind = dict(zip(self.classes, range(self.num_classes))) self._class_to_coco_ind = dict(zip(cats, self.coco.getCatIds())) - self._coco_ind_to_class_ind = dict( - [ - (self._class_to_coco_ind[cls], self._class_to_ind[cls]) + self._coco_ind_to_class_ind = { + self._class_to_coco_ind[cls]: self._class_to_ind[cls] for cls in self.classes[1:] - ] - ) - + } def _get_anno_file_name(self): # example: root/annotations/person_keypoints_tran2017.json # image_info_test-dev2017.json diff --git a/lib/dataset/CrowdPoseDataset.py b/lib/dataset/CrowdPoseDataset.py index 3329aaf..13627fb 100644 --- a/lib/dataset/CrowdPoseDataset.py +++ b/lib/dataset/CrowdPoseDataset.py @@ -57,12 +57,10 @@ def __init__(self, root, dataset, data_format, transform=None, self.num_classes = len(self.classes) self._class_to_ind = dict(zip(self.classes, range(self.num_classes))) self._class_to_coco_ind = dict(zip(cats, self.coco.getCatIds())) - self._coco_ind_to_class_ind = dict( - [ - (self._class_to_coco_ind[cls], self._class_to_ind[cls]) + self._coco_ind_to_class_ind = { + self._class_to_coco_ind[cls]: self._class_to_ind[cls] for cls in self.classes[1:] - ] - ) + } def _get_anno_file_name(self): # example: root/json/crowdpose_{train,val,test}.json