🔀 [Merge] branch 'SETUP' into TRAIN

README.md

@@ -13,7 +13,9 @@ While the project's structure is still being finalized, we ask that potential co
 
 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.
 
+## Star History
 
+[![Star History Chart](https://api.star-history.com/svg?repos=WongKinYiu/yolov9mit&type=Date)](https://star-history.com/#WongKinYiu/yolov9mit&Date)
 
 ## To-Do Lists
 - [ ] Project Setup

utils/dataloader.py

@@ -50,7 +50,7 @@ class YoloDataset(Dataset):
             cache[phase_name] = data
 
         cache.close()
-        logger.info("Loaded {} cache", phase_name)
+        logger.info("📦 Loaded {} cache", phase_name)
         data = cache[phase_name]
         return data
 

utils/get_dataset.py

@@ -60,7 +60,7 @@ def prepare_dataset(download_cfg):
 
         # Check if the correct number of files exists
         if check_files(extract_to, expected_files):
-            logger.info(f"Dataset {dataset_type} already verified.")
+            logger.info(f"✅ Dataset {dataset_type: >4} already verified.")
             continue
 
         if os.path.exists(local_zip_path):