[GSoC2024] Fix missing related images 3d dataset exports

cvat/apps/dataset_manager/bindings.py

@@ -1534,9 +1534,13 @@ def __init__(
                         attributes["labels"].append({"label_id": idx, "name": label["name"], "color": label["color"], "type": label["type"]})
                         attributes["track_id"] = -1
 
+                extra_images_paths = dm_image[1]
+                extra_images_objects = [dm.Image(path=image_path) for image_path in extra_images_paths]
+
                 dm_item = dm.DatasetItem(
                     id=osp.splitext(osp.split(frame_data.name)[-1])[0],
-                    annotations=dm_anno, media=PointCloud(dm_image[0]), related_images=dm_image[1],
+                    annotations=dm_anno, 
+                    media=PointCloud(dm_image[0], extra_images=extra_images_objects),
                     attributes=attributes
                 )
 
@@ -1624,9 +1628,13 @@ def __init__(
                         attributes["labels"].append({"label_id": idx, "name": label["name"], "color": label["color"], "type": label["type"]})
                         attributes["track_id"] = -1
 
+                extra_images_paths = dm_image[1]
+                extra_images_objects = [dm.Image(path=image_path) for image_path in extra_images_paths]
+
                 dm_item = dm.DatasetItem(
                     id=osp.splitext(osp.split(frame_data.name)[-1])[0],
-                    annotations=dm_anno, media=PointCloud(dm_image[0]), related_images=dm_image[1],
+                    annotations=dm_anno, 
+                    media=PointCloud(dm_image[0], extra_images=extra_images_objects),
                     attributes=attributes, subset=frame_data.subset
                 )
             dm_items.append(dm_item)

cvat/apps/engine/tests/test_rest_api_3D.py

@@ -263,13 +263,35 @@ def etree_to_dict(t):
                 jsons = glob(osp.join(tmp_dir, '**', '*.json'), recursive=True)
                 self.assertTrue(jsons)
                 self.assertTrue(set(checking_files).issubset(set(jsons)))
+        elif format_name == "Datumaro 3D 1.0":
+            with tempfile.TemporaryDirectory() as tmp_dir:
+                zipfile.ZipFile(content).extractall(tmp_dir)
+                extracted_files = set(
+                    os.path.relpath(os.path.join(root, file), tmp_dir)
+                    for root, dirs, files in os.walk(tmp_dir)
+                    for file in files
+                )
+                expected_files = set([
+                    'point_clouds/default/000001.pcd',
+                    'point_clouds/default/000002.pcd',
+                    'point_clouds/default/000003.pcd',
+                    'annotations/default.json'
+                ])
+                if related_files:
+                    expected_files.update([
+                        'related_images/default/000001/image_0.png',
+                        'related_images/default/000002/image_0.png',
+                        'related_images/default/000003/image_0.png'
+                    ])
+                self.assertEqual(extracted_files, expected_files)
+
 
 
 class Task3DTest(_DbTestBase):
     @classmethod
     def setUpClass(cls):
         super().setUpClass()
-        cls.format_names = ["Sly Point Cloud Format 1.0", "Kitti Raw Format 1.0"]
+        cls.format_names = ["Sly Point Cloud Format 1.0", "Kitti Raw Format 1.0", "Datumaro 3D 1.0"]
         cls._image_sizes = {}
         cls.pointcloud_pcd_filename = "test_canvas3d.zip"
         cls.pointcloud_pcd_path = osp.join(os.path.dirname(__file__), 'assets', cls.pointcloud_pcd_filename)
@@ -761,5 +783,5 @@ def test_api_v2_export_dataset(self):
                             with open(file_name, "wb") as f:
                                 f.write(content.getvalue())
                         self.assertEqual(osp.exists(file_name), edata['file_exists'])
-                        self._check_dump_content(content, task_ann_prev.data, format_name,related_files=False)
+                        self._check_dump_content(content, task_ann_prev.data, format_name, related_files=True)