diff --git a/config/datasets/detection-3d/lep_3d_dataset.yaml b/config/datasets/detection-3d/lep_3d_dataset.yaml index d1806c80..d3188a92 100644 --- 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 index 925c8940..93acf0e1 100644 --- a/innofw/core/integrations/mmdetection/datamodule.py +++ b/innofw/core/integrations/mmdetection/datamodule.py @@ -47,7 +47,8 @@ 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, @@ -55,7 +56,7 @@ def __init__( *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: