mapillary_vistas_format: fix image import (#546)

openvinotoolkit · Nov 10, 2021 · 2b93ba1 · 2b93ba1
1 parent fe0b13a
commit 2b93ba1
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/datumaro/plugins/mapillary_vistas_format/extractor.py b/datumaro/plugins/mapillary_vistas_format/extractor.py
@@ -14,7 +14,8 @@
     Polygon,
 )
 from datumaro.components.extractor import DatasetItem, SourceExtractor
-from datumaro.util.image import Image, find_images, lazy_image, load_image
+from datumaro.components.media import Image
+from datumaro.util.image import find_images, lazy_image, load_image
 from datumaro.util.mask_tools import bgr2index
 
 from .format import (