Save full image paths in coco (#1381)

cvat-ai · Apr 8, 2020 · afeab69 · afeab69
1 parent e84f482
commit afeab69
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/datumaro/datumaro/plugins/coco_format/converter.py b/datumaro/datumaro/plugins/coco_format/converter.py
@@ -514,7 +514,7 @@ def _save_image(self, item):
         filename += CocoPath.IMAGE_EXT
         path = osp.join(self._images_dir, filename)
         save_image(path, image)
-        return filename
+        return path
 
     def convert(self):
         self._make_dirs()
@@ -536,7 +536,7 @@ def convert(self):
             for item in subset:
                 filename = ''
                 if item.has_image:
-                    filename = item.image.filename
+                    filename = item.image.path
                 if self._save_images:
                     if item.has_image:
                         filename = self._save_image(item)