henry000 commited on
Commit
ebb5315
Β·
1 Parent(s): 95ca62f

🎨 [Add] Some icon in logger, add figure in README

Browse files
Files changed (3) hide show
  1. README.md +2 -0
  2. utils/dataloader.py +1 -1
  3. 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
+ [![Star History Chart](https://api.star-history.com/svg?repos=WongKinYiu/yolov9mit&type=Date)](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):