Merge pull request #3 from v7labs/rle

rle support

README.md

@@ -61,3 +61,9 @@ Similar to OpenCV's `cv2.findContours` but lacking hierarchies. Also similar to
 
 Note that currently the input mask to find_contour needs to be uint8.
 
+## rle_encode
+`rle_encode(mask: array[:,:]) -> list`
+Takes a 2-dim binary mask and generates a run length encoding according to the coco specs
+
+~ 15 times faster than written in plain python
+

setup.py

@@ -11,6 +11,7 @@
     setuptools.Extension("draw_polygon", ["upolygon/draw_polygon" + ext], extra_compile_args=["-O3", "-Wall"]),
     setuptools.Extension("find_contours", ["upolygon/find_contours" + ext], extra_compile_args=["-O3", "-Wall"]),
     setuptools.Extension("simplify_polygon", ["upolygon/simplify_polygon" + ext], extra_compile_args=["-O3", "-Wall"]),
+    setuptools.Extension("rle", ["upolygon/rle" + ext], extra_compile_args=["-O3", "-Wall"]),
 ]
 
 if USE_CYTHON:
@@ -20,7 +21,7 @@
 
 setuptools.setup(
     name="upolygon",
-    version="0.1.4",
+    version="0.1.5",
     author="V7",
     author_email="simon@v7labs.com",
     description="Collection of fast polygon operations for DL",

upolygon/__init__.py

@@ -1,3 +1,4 @@
 from draw_polygon import draw_polygon  # noqa
 from find_contours import find_contours  # noqa
 from simplify_polygon import simplify_polygon  # noqa
+from rle import rle_decode, rle_encode # noqa