Merge pull request #4 from deepbands/develop

update new version v0.2.1

Author: yharby

deep-learning-datasets-maker/metadata.txt

@@ -6,7 +6,7 @@
 name=Deep Learning Datasets Maker
 qgisMinimumVersion=3.0
 description=tools to handle raster and vector data to split it into small pieces equaled in size for machine learning datasets
-version=0.2
+version=0.2.1
 author=deepbands (Youssef Harby and Yizhou Chen)
 email=youssef_harby@yahoo.com
 
@@ -37,7 +37,7 @@ deprecated=False
 # Since QGIS 3.8, a comma separated list of plugins to be installed
 # (or upgraded) can be specified.
 # Check the documentation for more information.
-plugin_dependencies=scikit-image, pycocotools
+plugin_dependencies=Cython, scikit-image, pycocotools, Pillow
 
 Category of the plugin: Raster, Vector, Database or Web
 # category=

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

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

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

@@ -5,7 +5,13 @@
 import os
 import re
 import fnmatch
-from PIL import Image
+try:
+    from PIL import Image
+except ImportError:
+    print('some packages not installed. Installing ...')
+    subprocess.check_call(
+        ["python3", '-m', 'pip', 'install', 'Pillow'])
+    from PIL import Image
 import numpy as np
 from .pycococreatortools import *
 from .tif_process import *

deep-learning-datasets-maker/utils/__init__.py

@@ -1,5 +1,4 @@
 from .rasterize import rasterize
 from .splitting import splitting
-from .semantic2instance import segMaskB2I
 from .paddlepaddle_split_dataset_list import generate_list
 from .intSegGDAL import rasterizeInsSeg

deep-learning-datasets-maker/utils/semantic2instance.py

@@ -1,2 +1,4 @@
+Cython
 scikit-image
-pycocotools
+pycocotools
+Pillow