Move some variable (#19)

* Move some variable * bump 12.0.3
open-mmlab · Sep 20, 2020 · 01f805c · 01f805c
1 parent 558d9e5
commit 01f805c
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 5 deletions.
diff --git a/pycocotools/pycocotools/coco.py b/pycocotools/pycocotools/coco.py
@@ -78,8 +78,6 @@ def __init__(self, annotation_file=None):
         self.dataset, self.anns, self.cats, self.imgs = dict(), dict(), dict(
         ), dict()
         self.imgToAnns, self.catToImgs = defaultdict(list), defaultdict(list)
-        self.img_ann_map = self.imgToAnns
-        self.cat_img_map = self.catToImgs
         if annotation_file is not None:
             print('loading annotations into memory...')
             tic = time.time()
@@ -92,6 +90,8 @@ def __init__(self, annotation_file=None):
             print('Done (t={:0.2f}s)'.format(time.time() - tic))
             self.dataset = dataset
             self.createIndex()
+        self.img_ann_map = self.imgToAnns
+        self.cat_img_map = self.catToImgs
 
     def createIndex(self):
         # create index
@@ -123,8 +123,6 @@ def createIndex(self):
         self.catToImgs = catToImgs
         self.imgs = imgs
         self.cats = cats
-        self.img_ann_map = self.imgToAnns
-        self.cat_img_map = self.catToImgs
 
     def info(self):
         """

diff --git a/pycocotools/setup.py b/pycocotools/setup.py
@@ -24,5 +24,5 @@
       install_requires=[
           'setuptools>=18.0', 'cython>=0.27.3', 'matplotlib>=2.1.0'
       ],
-      version='12.0.2',
+      version='12.0.3',
       ext_modules=ext_modules)