uploaded the data, reformatted datamodule

InnopolisUni · Aug 9, 2024 · 110c5a8 · 110c5a8
1 parent 3320230
commit 110c5a8
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 5 deletions.
diff --git a/config/datasets/detection-3d/lep_3d_dataset.yaml b/config/datasets/detection-3d/lep_3d_dataset.yaml
@@ -2,17 +2,22 @@ task:
   - 3d-detection
 
 name: lep_3d
-description: Набор данных содержит видео облетов c дронов над территорией линии электропередач
+description: Набор данных содержит видео облетов c бпла над территорией линии электропередач
 
 markup_info: Набор данных содержит разметку bounding box, под формат детекции столбов ЛЭП.
 date_time: 10.06.2024
 
 _target_: innofw.core.integrations.mmdetection.datamodule.Mmdetection3DDataModuleAdapter
 
-data:
-  source: /home/karim/workdir/innofw/data/lep3d # machine .79
+train:
+  source: https://api.blackhole.ai.innopolis.university/public-datasets/lep_3d_detection/train.zip
   target: /home/karim/workdir/innofw/data/lep3d
 
+test:
+  source: https://api.blackhole.ai.innopolis.university/public-datasets/lep_3d_detection/test.zip
+  target: /home/karim/workdir/innofw/data/lep3d
+
+
 num_workers: 8
 
 val_size: 0.2

diff --git a/innofw/core/integrations/mmdetection/datamodule.py b/innofw/core/integrations/mmdetection/datamodule.py
@@ -47,15 +47,16 @@ class Mmdetection3DDataModuleAdapter(BaseDataModule, ABC):
 
     def __init__(
             self,
-            data: Optional[str],
+            train: Optional[str],
+            test: Optional[str],
             num_classes: int,
             val_size: float = 0.2,
             stage=False,
             random_state: int = 0,
             *args,
             **kwargs,
     ):
-        super().__init__(train=data, test=data, infer=data, stage=stage, *args, **kwargs)
+        super().__init__(train=train, test=test, infer=test, stage=stage, *args, **kwargs)
         if self.train:
             self.train_source = Path(self.train)
         if self.test: