📝 [Update] README Link & requirement

.github/workflows/deploy.yaml

@@ -13,7 +13,7 @@ jobs:
     strategy:
       matrix:
         operating-system: [ubuntu-latest, macos-latest]
-        python-version: [3.8, 3.9, '3.10']
+        python-version: [3.8, '3.10', '3.12']
       fail-fast: false
 
     steps:

.github/workflows/develop.yaml

@@ -7,15 +7,15 @@ on:
     branches: [main]
 
 jobs:
-  build:
+  build_and_test:
     runs-on: ubuntu-latest
 
     steps:
     - name: Checkout repository
-      uses: actions/checkout@v2
+      uses: actions/checkout@v3
 
     - name: Set up Python 3.10
-      uses: actions/setup-python@v2
+      uses: actions/setup-python@v4
       with:
         python-version: '3.10'
 
@@ -30,7 +30,7 @@ jobs:
     - name: Install dependencies
       run: |
         python -m pip install --upgrade pip
-        pip install -r requirements.txt
+        pip install -r requirements-dev.txt
 
     - name: Install pre-commit
       run: pip install pre-commit

README.md

@@ -2,7 +2,8 @@
 
 ![GitHub License](https://img.shields.io/github/license/WongKinYiu/YOLO)
 ![WIP](https://img.shields.io/badge/status-WIP-orange)
-![GitHub Actions Workflow Status](https://img.shields.io/github/actions/workflow/status/WongKinYiu/YOLO/main.yaml)
+![](https://img.shields.io/github/actions/workflow/status/WongKinYiu/YOLO/deploy.yaml)
+
 
 [![PWC](https://img.shields.io/endpoint.svg?url=https://paperswithcode.com/badge/yolov9-learning-what-you-want-to-learn-using/real-time-object-detection-on-coco)](https://paperswithcode.com/sota/real-time-object-detection-on-coco)
 

requirements-dev.txt

@@ -0,0 +1,5 @@
+-r requirements.txt
+pytest
+pytest-cov
+pre-commit
+pycocotools

requirements.txt

@@ -6,11 +6,8 @@ numpy
 opencv-python
 Pillow
 pycocotools
-pytest
-pyyaml
 requests
 rich
 torch
 torchvision
-tqdm
 wandb

yolo/tools/data_conversion.py

@@ -2,7 +2,7 @@ import json
 from pathlib import Path
 from typing import Dict, List, Optional
 
-from tqdm import tqdm
+from rich.progress import track
 
 
 def discretize_categories(categories: List[Dict[str, int]]) -> Dict[int, int]:
@@ -23,7 +23,7 @@ def process_annotations(
     """
     Process and save annotations to files, with option to remap category IDs.
     """
-    for image_id, annotations in tqdm(image_annotations.items(), desc="Processing annotations"):
+    for image_id, annotations in track(image_annotations.items(), description="Processing annotations"):
         file_path = output_dir / "{image_id:0>12}.txt"
         if not annotations:
             continue