🩹 [Fix] config for auto download dataset
Browse files
yolo/tools/data_loader.py
CHANGED
@@ -213,7 +213,7 @@ def create_dataloader(data_cfg: DataConfig, dataset_cfg: DatasetConfig, task: st
|
|
213 |
if task == "inference":
|
214 |
return StreamDataLoader(data_cfg)
|
215 |
|
216 |
-
if dataset_cfg
|
217 |
prepare_dataset(dataset_cfg, task)
|
218 |
dataset = YoloDataset(data_cfg, dataset_cfg, task)
|
219 |
|
|
|
213 |
if task == "inference":
|
214 |
return StreamDataLoader(data_cfg)
|
215 |
|
216 |
+
if getattr(dataset_cfg, "auto_download", False):
|
217 |
prepare_dataset(dataset_cfg, task)
|
218 |
dataset = YoloDataset(data_cfg, dataset_cfg, task)
|
219 |
|