Update image.py

jsbroks · Mar 14, 2019 · 40a4d90 · 40a4d90
1 parent e2f73fd
commit 40a4d90
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/imantics/image.py b/imantics/image.py
@@ -133,16 +133,16 @@ def add(self, annotation, category=None):
 
             height, width = annotation.array.shape[:2]
             if width == self.width and height == self.height:
-                annotation = Annotation.from_mask(annotation, category, annotation)
+                annotation = Annotation.from_mask(annotation, image=self, category=category)
             else:
                 raise ValueError('Cannot add annotaiton of size {} to image of size {}'\
                                  .format(annotation.array.shape, (self.height, self.width)))
 
         if isinstance(annotation, BBox):
-            annotation = Annotation.from_bbox(annotation, category, annotation)
+            annotation = Annotation.from_bbox(annotation, image=self, category=category)
 
         if isinstance(annotation, Polygons):
-            annotation = Annotation.from_polygons(annotation, category, annotation)
+            annotation = Annotation.from_polygons(annotation, image=self, category=category)
 
         annotation.set_image(self)
         annotation.index(self)