π¨ [Add] Some icon in logger, add figure in README
Browse files- README.md +2 -0
- utils/dataloader.py +1 -1
- utils/get_dataset.py +1 -1
README.md
CHANGED
@@ -13,7 +13,9 @@ While the project's structure is still being finalized, we ask that potential co
|
|
13 |
|
14 |
If you are interested in contributing, please keep an eye on project updates or contact us directly at [[email protected]](mailto:[email protected]) for more information.
|
15 |
|
|
|
16 |
|
|
|
17 |
|
18 |
## To-Do Lists
|
19 |
- [ ] Project Setup
|
|
|
13 |
|
14 |
If you are interested in contributing, please keep an eye on project updates or contact us directly at [[email protected]](mailto:[email protected]) for more information.
|
15 |
|
16 |
+
## Star History
|
17 |
|
18 |
+
[](https://star-history.com/#WongKinYiu/yolov9mit&Date)
|
19 |
|
20 |
## To-Do Lists
|
21 |
- [ ] Project Setup
|
utils/dataloader.py
CHANGED
@@ -49,7 +49,7 @@ class YoloDataset(Dataset):
|
|
49 |
cache[phase_name] = data
|
50 |
|
51 |
cache.close()
|
52 |
-
logger.info("Loaded {} cache", phase_name)
|
53 |
data = cache[phase_name]
|
54 |
return data
|
55 |
|
|
|
49 |
cache[phase_name] = data
|
50 |
|
51 |
cache.close()
|
52 |
+
logger.info("π¦ Loaded {} cache", phase_name)
|
53 |
data = cache[phase_name]
|
54 |
return data
|
55 |
|
utils/get_dataset.py
CHANGED
@@ -60,7 +60,7 @@ def prepare_dataset(download_cfg):
|
|
60 |
|
61 |
# Check if the correct number of files exists
|
62 |
if check_files(extract_to, expected_files):
|
63 |
-
logger.info(f"Dataset {dataset_type} already verified.")
|
64 |
continue
|
65 |
|
66 |
if os.path.exists(local_zip_path):
|
|
|
60 |
|
61 |
# Check if the correct number of files exists
|
62 |
if check_files(extract_to, expected_files):
|
63 |
+
logger.info(f"β
Dataset {dataset_type: >4} already verified.")
|
64 |
continue
|
65 |
|
66 |
if os.path.exists(local_zip_path):
|