update coco utils (#40)

* update coco utils * update to v0.3.3
obss · Apr 11, 2021 · 3ad54d6 · 3ad54d6
1 parent e734ae8
commit 3ad54d6
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 2 deletions.
diff --git a/sahi/__init__.py b/sahi/__init__.py
@@ -1 +1 @@
-__version__ = "0.3.2"
+__version__ = "0.3.3"
diff --git a/sahi/utils/coco.py b/sahi/utils/coco.py
@@ -774,6 +774,18 @@ def add_category(self, category):
 
         self.categories.append(category)
 
+    def add_image(self, image):
+        """
+        Adds image to this Coco instance
+
+        Args:
+            image: CocoImage
+        """
+
+        assert type(image) == CocoImage, "image must be a CocoImage instance"
+
+        self.images.append(image)
+
     @classmethod
     def from_coco_dict_or_path(cls, coco_dict_or_path):
         """
@@ -1566,7 +1578,7 @@ def add_video(self, video):
             video: CocoVideo
         """
 
-        assert isinstance(video, CocoVideo), "video must be a CocoVideo instance"
+        assert type(video) == CocoVideo, "video must be a CocoVideo instance"
 
         self.videos.append(video)