packages install fix

Author: yharby

deep-learning-datasets-maker/utils/COCO/pycococreatortools/pycococreatortools.py

@@ -5,15 +5,17 @@
 import datetime
 import numpy as np
 from itertools import groupby
-from skimage import measure
 from PIL import Image
 import subprocess
 try:
     from pycocotools import mask
+    from skimage import measure
 except ImportError:
-    print('pycocotools package not installed. Installing ...')
-    subprocess.check_call(["python", '-m', 'pip', 'install', 'pycocotools'])
+    print('some packages not installed. Installing ...')
+    subprocess.check_call(
+        ["python3", '-m', 'pip', 'install', 'pycocotools', 'scikit-image'])
     from pycocotools import mask
+    from skimage import measure
 
 
 def convert(text): return int(text) if text.isdigit() else text.lower()

deep-learning-datasets-maker/utils/COCO/visualize_coco.py

@@ -1,9 +1,17 @@
-from pycocotools.coco import COCO
 import numpy as np
-import skimage.io as io
 import matplotlib.pyplot as plt
 import pylab
 import os
+import subprocess
+try:
+    from pycocotools.coco import COCO
+    import skimage.io as io
+except ImportError:
+    print('some packages not installed. Installing ...')
+    subprocess.check_call(
+        ["python3", '-m', 'pip', 'install', 'pycocotools', 'scikit-image'])
+    from pycocotools.coco import COCO
+    import skimage.io as io
 
 ROOT_DIR = r'./example_data/original_data/dataset/eval'
 image_directory = os.path.join(ROOT_DIR, "image")
@@ -16,9 +24,11 @@
 image_data = example_coco.loadImgs(image_ids[0])[0]
 
 image = io.imread(image_directory + '/' + image_data['file_name'])
-plt.imshow(image); plt.axis('off')
+plt.imshow(image)
+plt.axis('off')
 pylab.rcParams['figure.figsize'] = (8.0, 10.0)
-annotation_ids = example_coco.getAnnIds(imgIds=image_data['id'], catIds=category_ids, iscrowd=None)
+annotation_ids = example_coco.getAnnIds(
+    imgIds=image_data['id'], catIds=category_ids, iscrowd=None)
 annotations = example_coco.loadAnns(annotation_ids)
 example_coco.showAnns(annotations)
-plt.show()
+plt.show()