henry000 commited on
Commit
c134940
Β·
1 Parent(s): 2ee7407

πŸ‘· [Add] more workflows, deploy & developer

Browse files
.github/workflows/deploy.yaml ADDED
@@ -0,0 +1,68 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ name: Deploy Mode Validation & Inference
2
+
3
+ on:
4
+ push:
5
+ branches: [main]
6
+ pull_request:
7
+ branches: [main]
8
+
9
+ jobs:
10
+ deploy:
11
+ runs-on: ${{ matrix.operating-system }}
12
+
13
+ strategy:
14
+ matrix:
15
+ operating-system: [ubuntu-latest, windows-latest, macos-latest]
16
+ python-version: [3.8, 3.9, '3.10']
17
+ fail-fast: false
18
+
19
+ steps:
20
+ - name: Checkout repository
21
+ uses: actions/checkout@v2
22
+
23
+ - name: Set up Python
24
+ uses: actions/setup-python@v2
25
+ with:
26
+ python-version: ${{ matrix.python-version }}
27
+
28
+ - name: Cache pip dependencies
29
+ uses: actions/cache@v2
30
+ with:
31
+ path: ~/.cache/pip
32
+ key: ${{ runner.os }}-pip-${{ hashFiles('**/requirements.txt') }}-${{ matrix.python-version }}
33
+ restore-keys: |
34
+ ${{ runner.os }}-pip-${{ matrix.python-version }}
35
+ ${{ runner.os }}-pip-
36
+
37
+ - name: Install dependencies
38
+ run: |
39
+ python -m pip install --upgrade pip
40
+ pip install -r requirements.txt
41
+
42
+ - name: Install YOLO package
43
+ run: pip install -e .
44
+
45
+ - name: Cache model weights
46
+ id: cache-weights
47
+ uses: actions/cache@v2
48
+ with:
49
+ path: weights
50
+ key: ${{ runner.os }}-weights
51
+ restore-keys: |
52
+ ${{ runner.os }}-weights
53
+
54
+ - name: Run Validation
55
+ run: |
56
+ python yolo/lazy.py task=validation dataset=mock
57
+ python yolo/lazy.py task=validation dataset=mock model=v9-s
58
+ python yolo/lazy.py task=validation dataset=mock name=AnyNameYouWant
59
+
60
+ - name: Run Inference
61
+ run: |
62
+ python yolo/lazy.py task=inference
63
+ python yolo/lazy.py task=inference +quite=True
64
+ python yolo/lazy.py task=inference name=AnyNameYouWant
65
+ python yolo/lazy.py task=inference image_size=\[480,640]
66
+ python yolo/lazy.py task=inference task.nms.min_confidence=0.1
67
+ python yolo/lazy.py task=inference task.fast_inference=deploy
68
+ python yolo/lazy.py task=inference task.data.source=tests/data/images/val
.github/workflows/{main.yaml β†’ develop.yaml} RENAMED
@@ -1,26 +1,34 @@
1
- name: YOLOv9 - Model test and Code Style Check
2
 
3
  on:
4
  push:
5
- branches: [ main ]
6
  pull_request:
7
- branches: [ main ]
8
 
9
  jobs:
10
  build:
11
  runs-on: ubuntu-latest
12
 
13
  steps:
14
- - uses: actions/checkout@v2
15
- - name: Set up Python 3.8
 
 
16
  uses: actions/setup-python@v2
17
  with:
18
- python-version: 3.8
 
 
 
 
 
 
 
 
19
 
20
  - name: Install dependencies
21
  run: |
22
- sudo apt-get update
23
- sudo apt-get install -y libgl1-mesa-glx
24
  python -m pip install --upgrade pip
25
  pip install -r requirements.txt
26
 
@@ -35,8 +43,10 @@ jobs:
35
  restore-keys: |
36
  ${{ runner.os }}-precommit-
37
 
38
- - name: Run pre-commit (black and isort)
39
  run: pre-commit run --all-files
40
 
41
  - name: Test with pytest
42
- run: pytest
 
 
 
1
+ name: Developer Mode Build & Test
2
 
3
  on:
4
  push:
5
+ branches: [main]
6
  pull_request:
7
+ branches: [main]
8
 
9
  jobs:
10
  build:
11
  runs-on: ubuntu-latest
12
 
13
  steps:
14
+ - name: Checkout repository
15
+ uses: actions/checkout@v2
16
+
17
+ - name: Set up Python 3.10
18
  uses: actions/setup-python@v2
19
  with:
20
+ python-version: 3.10
21
+
22
+ - name: Cache pip dependencies
23
+ uses: actions/cache@v2
24
+ with:
25
+ path: ~/.cache/pip
26
+ key: ${{ runner.os }}-pip-${{ hashFiles('**/requirements.txt') }}
27
+ restore-keys: |
28
+ ${{ runner.os }}-pip-
29
 
30
  - name: Install dependencies
31
  run: |
 
 
32
  python -m pip install --upgrade pip
33
  pip install -r requirements.txt
34
 
 
43
  restore-keys: |
44
  ${{ runner.os }}-precommit-
45
 
46
+ - name: Run pre-commit hooks
47
  run: pre-commit run --all-files
48
 
49
  - name: Test with pytest
50
+ run: |
51
+ pip install pytest pytest-cov
52
+ pytest --cov=yolo
README.md CHANGED
@@ -1,8 +1,14 @@
1
  # YOLO: Official Implementation of YOLOv9, YOLOv7
2
 
3
  ![GitHub License](https://img.shields.io/github/license/WongKinYiu/YOLO)
4
- [![Hugging Face Spaces](https://img.shields.io/badge/%F0%9F%A4%97%20Hugging%20Face-Spaces-blue)](https://huggingface.co/spaces/henry000/YOLO)
5
  ![WIP](https://img.shields.io/badge/status-WIP-orange)
 
 
 
 
 
 
 
6
  <!-- > [!IMPORTANT]
7
  > This project is currently a Work In Progress and may undergo significant changes. It is not recommended for use in production environments until further notice. Please check back regularly for updates.
8
  >
 
1
  # YOLO: Official Implementation of YOLOv9, YOLOv7
2
 
3
  ![GitHub License](https://img.shields.io/github/license/WongKinYiu/YOLO)
 
4
  ![WIP](https://img.shields.io/badge/status-WIP-orange)
5
+ ![GitHub Actions Workflow Status](https://img.shields.io/github/actions/workflow/status/WongKinYiu/YOLO/main.yaml)
6
+
7
+ [![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)
8
+
9
+ [![Open In Colab](https://colab.research.google.com/assets/colab-badge.svg)]()
10
+ [![Hugging Face Spaces](https://img.shields.io/badge/%F0%9F%A4%97%20Hugging%20Face-Spaces-green)](https://huggingface.co/spaces/henry000/YOLO)
11
+
12
  <!-- > [!IMPORTANT]
13
  > This project is currently a Work In Progress and may undergo significant changes. It is not recommended for use in production environments until further notice. Please check back regularly for updates.
14
  >