🔀 [Merge] branch 'DATASET' into TEST
Browse files- requirements.txt +1 -4
- yolo/utils/get_dataset.py +1 -1
requirements.txt
CHANGED
|
@@ -10,7 +10,4 @@ requests
|
|
| 10 |
rich
|
| 11 |
torch
|
| 12 |
torchvision
|
| 13 |
-
tqdm
|
| 14 |
-
Pillow
|
| 15 |
-
diskcache
|
| 16 |
-
torchvision
|
|
|
|
| 10 |
rich
|
| 11 |
torch
|
| 12 |
torchvision
|
| 13 |
+
tqdm
|
|
|
|
|
|
|
|
|
yolo/utils/get_dataset.py
CHANGED
|
@@ -63,7 +63,7 @@ def prepare_dataset(cfg):
|
|
| 63 |
|
| 64 |
os.makedirs(extract_to, exist_ok=True)
|
| 65 |
if check_files(final_place, dataset_args.get("file_num")):
|
| 66 |
-
logger.info(f"Dataset {dataset_type} already verified.")
|
| 67 |
continue
|
| 68 |
|
| 69 |
if not os.path.exists(local_zip_path):
|
|
|
|
| 63 |
|
| 64 |
os.makedirs(extract_to, exist_ok=True)
|
| 65 |
if check_files(final_place, dataset_args.get("file_num")):
|
| 66 |
+
logger.info(f"✅ Dataset {dataset_type} already verified.")
|
| 67 |
continue
|
| 68 |
|
| 69 |
if not os.path.exists(local_zip_path):
|