Merge pull request #25 from pylabel-project/dev

To deploy version v0.1.31
pylabel-project · Feb 11, 2022 · d18d52a · d18d52a
2 parents 9402b15 + 6433501
commit d18d52a
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 1 deletion.
diff --git a/pylabel/importer.py b/pylabel/importer.py
@@ -79,8 +79,15 @@ def ImportCoco(path, path_to_images=None, name=None):
 
     categories = pd.json_normalize(annotations_json["categories"])
     categories.columns = "cat_" + categories.columns
+
+    # Converting this to string resolves issue #23
+    categories.cat_id = categories.cat_id.astype(str)
 
     df = annotations
+
+    # Converting this to string resolves issue #23
+    df.ann_category_id = df.ann_category_id.astype(str)
+
     df[
         ["ann_bbox_xmin", "ann_bbox_ymin", "ann_bbox_width", "ann_bbox_height"]
     ] = pd.DataFrame(df.ann_bbox.tolist(), index=df.index)

diff --git a/setup.py b/setup.py
@@ -8,7 +8,7 @@
 setup(
     name="pylabel",
     packages=["pylabel"],
-    version="0.1.30",
+    version="0.1.31",
     description="Transform, analyze, and visualize computer vision annotations.",
     long_description=long_description,
     long_description_content_type="text/markdown",