Save full image paths in coco (#1381)

6 years ago · afeab69aa3
parent e84f482600
commit afeab69aa3
1 changed files with 2 additions and 2 deletions
--- a/datumaro/datumaro/plugins/coco_format/converter.py
+++ b/datumaro/datumaro/plugins/coco_format/converter.py
@ -514,7 +514,7 @@ class _Converter:
        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 @@ class _Converter:
            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)