π [Merge] branch 'TEST'
Browse files
yolo/tools/data_loader.py
CHANGED
@@ -46,7 +46,7 @@ class YoloDataset(Dataset):
|
|
46 |
Returns:
|
47 |
dict: The loaded data from the cache for the specified phase.
|
48 |
"""
|
49 |
-
cache_path = dataset_path / f"{phase_name}.
|
50 |
|
51 |
if not cache_path.exists():
|
52 |
logger.info(f":factory: Generating {phase_name} cache")
|
|
|
46 |
Returns:
|
47 |
dict: The loaded data from the cache for the specified phase.
|
48 |
"""
|
49 |
+
cache_path = dataset_path / f"{phase_name}.cache"
|
50 |
|
51 |
if not cache_path.exists():
|
52 |
logger.info(f":factory: Generating {phase_name} cache")
|