Merge branch 'master' of https://github.com/superannotateai/superannotate-python-sdk

Author: agyurjin

Dockerfile

@@ -17,6 +17,7 @@ RUN update-alternatives --install /usr/bin/python python /usr/bin/python3 1
 RUN update-alternatives --install /usr/local/bin/pip pip /usr/bin/pip3 1
 
 RUN apt-get install -y build-essential && \
+    pip install --no-cache-dir shapely && \
     pip install --no-cache-dir ${PIP_FLAGS} superannotate && \
     apt-get remove -y build-essential && \
     apt-get -y autoremove && \

MANIFEST.in

@@ -1,5 +1,3 @@
 include README.md
 include requirements.txt
-include requirements_coco.txt
 include LICENSE
-include superannotate/pycocotools_sa/maskApi.h

pyproject.toml

@@ -14,6 +14,5 @@ packaging>=20.4
 pandas>=1.1.2
 plotly>=4.1.0
 ffmpeg-python>=0.2.0
-Shapely>=1.7.1
 google-cloud-storage>=1.33.0
 azure-storage-blob>=12.6.0

requirements.txt

@@ -1,5 +1,4 @@
-import numpy as np
-from setuptools import Extension, find_packages, setup
+from setuptools import find_packages, setup
 
 with open('requirements.txt') as f:
     requirements = f.read()
@@ -18,18 +17,6 @@
 Version = Version.rstrip()
 Version = Version[15:-1]
 
-ext_modules = [
-    Extension(
-        'superannotate.pycocotools_sa._mask',
-        sources=[
-            'superannotate/pycocotools_sa/_mask.pyx',
-            'superannotate/pycocotools_sa/maskApi.c'
-        ],
-        include_dirs=[np.get_include(), 'superannotate/pycocotools_sa'],
-        extra_compile_args=[],
-    )
-]
-
 setup(
     name='superannotate',
     version=Version,
@@ -46,6 +33,5 @@
     entry_points={
         'console_scripts': ['superannotatecli = superannotate.__main__:main']
     },
-    python_requires='>=3.6',
-    ext_modules=ext_modules
+    python_requires='>=3.6'
 )

setup.py

@@ -17,8 +17,13 @@
 )
 from .api import API as _API
 from .common import image_path_to_annotation_paths
-from .consensus_benchmark.benchmark import benchmark
-from .consensus_benchmark.consensus import consensus
+
+try:
+    from .consensus_benchmark.benchmark import benchmark
+    from .consensus_benchmark.consensus import consensus
+except:
+    pass
+
 from .dataframe_filtering import (
     filter_annotation_instances, filter_images_by_comments,
     filter_images_by_tags

superannotate/__init__.py

@@ -1,10 +1,8 @@
-from shapely.geometry import Polygon, box, Point
-import plotly.express as px
-from plotly.subplots import make_subplots
-import plotly.graph_objects as go
-import plotly.colors as col
 import logging
-import numpy as np
+
+import plotly.express as px
+from shapely.geometry import Point, Polygon, box
+
 logger = logging.getLogger("superannotate-python-sdk")
 
 
@@ -185,4 +183,4 @@ def consensus_plot(consensus_df, projects):
     )
     fig.for_each_annotation(lambda a: a.update(text=a.text.split("=")[-1]))
     fig.for_each_trace(lambda t: t.update(name=t.name.split("=")[-1]))
-    fig.show()
+    fig.show()

superannotate/consensus_benchmark/helpers.py

@@ -12,6 +12,7 @@
 
 logger = logging.getLogger("superannotate-python-sdk")
 
+
 def annot_to_bitmask(annot):
     if isinstance(annot['counts'], list):
         bitmask = _maskfrRLE(annot)
@@ -20,6 +21,7 @@ def annot_to_bitmask(annot):
 
     return bitmask
 
+
 def coco_panoptic_segmentation_to_sa_pixel(coco_path, images_path):
     coco_json = json.load(open(coco_path))
     hex_colors = blue_color_generator(len(coco_json["categories"]))