Add LFW

cvat/apps/dataset_manager/formats/lfw.py

@@ -0,0 +1,29 @@
+from tempfile import TemporaryDirectory
+
+from datumaro.components.dataset import Dataset
+from pyunpack import Archive
+
+from cvat.apps.dataset_manager.bindings import (GetCVATDataExtractor,
+                                                import_dm_annotations)
+from cvat.apps.dataset_manager.util import make_zip_archive
+
+from .registry import dm_env, exporter, importer
+
+
+@importer(name='LFW', ext='ZIP', version='1.0')
+def _import(src_file, instance_data):
+    with TemporaryDirectory() as tmp_dir:
+        Archive(src_file.name).extractall(tmp_dir)
+
+        dataset = Dataset.import_from(tmp_dir, 'lfw')
+
+        import_dm_annotations(dataset, instance_data)
+
+@exporter(name='LFW', ext='ZIP', version='1.0')
+def _exporter(dst_file, instance_data, save_images=False):
+    dataset = Dataset.from_extractors(GetCVATDataExtractor(instance_data,
+        include_images=save_images), env=dm_env)
+    with TemporaryDirectory() as tmp_dir:
+        dataset.export(tmp_dir, format='lfw', save_images=save_images)
+
+        make_zip_archive(tmp_dir, dst_file)

cvat/apps/dataset_manager/formats/registry.py

@@ -109,4 +109,4 @@ def make_exporter(name):
 import cvat.apps.dataset_manager.formats.icdar
 import cvat.apps.dataset_manager.formats.velodynepoint
 import cvat.apps.dataset_manager.formats.pointcloud
-
+import cvat.apps.dataset_manager.formats.lfw

cvat/apps/dataset_manager/formats/utils.py

@@ -5,7 +5,7 @@
 import os.path as osp
 from hashlib import blake2s
 
-from datumaro.cli.util import make_file_name
+from datumaro.util.os_util import make_file_name
 
 
 def get_color_from_index(index):

cvat/apps/dataset_manager/views.py

@@ -8,7 +8,7 @@
 from datetime import timedelta
 
 import django_rq
-from datumaro.cli.util import make_file_name
+from datumaro.util.os_util import make_file_name
 from datumaro.util import to_snake_case
 from django.utils import timezone