Spaces:
Sleeping
Sleeping
🩹 [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 |
|