ultralytics 8.0.65 YOLOv8 Pose models (#1347)
Signed-off-by: dependabot[bot] <support@github.com> Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com> Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com> Co-authored-by: Mert Can Demir <validatedev@gmail.com> Co-authored-by: Laughing <61612323+Laughing-q@users.noreply.github.com> Co-authored-by: Fabian Greavu <fabiangreavu@gmail.com> Co-authored-by: Yonghye Kwon <developer.0hye@gmail.com> Co-authored-by: Eric Pedley <ericpedley@gmail.com> Co-authored-by: JustasBart <40023722+JustasBart@users.noreply.github.com> Co-authored-by: dependabot[bot] <49699333+dependabot[bot]@users.noreply.github.com> Co-authored-by: Aarni Koskela <akx@iki.fi> Co-authored-by: Sergio Sanchez <sergio.ssm.97@gmail.com> Co-authored-by: Bogdan Gheorghe <112427971+bogdan-galileo@users.noreply.github.com> Co-authored-by: Jaap van de Loosdrecht <jaap@vdlmv.nl> Co-authored-by: Noobtoss <96134731+Noobtoss@users.noreply.github.com> Co-authored-by: nerdyespresso <106761627+nerdyespresso@users.noreply.github.com> Co-authored-by: Farid Inawan <frdteknikelektro@gmail.com> Co-authored-by: Laughing-q <1185102784@qq.com> Co-authored-by: Alexander Duda <Alexander.Duda@me.com> Co-authored-by: Mehran Ghandehari <mehran.maps@gmail.com> Co-authored-by: Snyk bot <snyk-bot@snyk.io> Co-authored-by: majid nasiri <majnasai@gmail.com>
This commit is contained in:
Ayush Chaurasia
GitHub
parent
9af3e69b1a
commit
1cb92d7f42
57 changed files with 1578 additions and 489 deletions
95
.github/workflows/ci.yaml
vendored
95
.github/workflows/ci.yaml
vendored
|
|
@ -43,21 +43,7 @@ jobs:
|
|||
python --version
|
||||
pip --version
|
||||
pip list
|
||||
- name: Test HUB training (Python Usage 1)
|
||||
shell: python
|
||||
env:
|
||||
APIKEY: ${{ secrets.ULTRALYTICS_HUB_APIKEY }}
|
||||
run: |
|
||||
import os
|
||||
from pathlib import Path
|
||||
from ultralytics import YOLO, hub
|
||||
from ultralytics.yolo.utils import USER_CONFIG_DIR
|
||||
Path(USER_CONFIG_DIR / 'settings.yaml').unlink()
|
||||
key = os.environ['APIKEY']
|
||||
hub.reset_model(key)
|
||||
model = YOLO('https://hub.ultralytics.com/models/' + key)
|
||||
model.train()
|
||||
- name: Test HUB training (Python Usage 2)
|
||||
- name: Test HUB training
|
||||
shell: python
|
||||
env:
|
||||
APIKEY: ${{ secrets.ULTRALYTICS_HUB_APIKEY }}
|
||||
|
|
@ -73,36 +59,6 @@ jobs:
|
|||
hub.login(key)
|
||||
model = YOLO('https://hub.ultralytics.com/models/' + model_id)
|
||||
model.train()
|
||||
- name: Test HUB training (Python Usage 3)
|
||||
shell: python
|
||||
env:
|
||||
APIKEY: ${{ secrets.ULTRALYTICS_HUB_APIKEY }}
|
||||
run: |
|
||||
import os
|
||||
from pathlib import Path
|
||||
from ultralytics import YOLO, hub
|
||||
from ultralytics.yolo.utils import USER_CONFIG_DIR
|
||||
Path(USER_CONFIG_DIR / 'settings.yaml').unlink()
|
||||
key = os.environ['APIKEY']
|
||||
hub.reset_model(key)
|
||||
model = YOLO(key)
|
||||
model.train()
|
||||
- name: Test HUB training (Python Usage 4)
|
||||
shell: python
|
||||
env:
|
||||
APIKEY: ${{ secrets.ULTRALYTICS_HUB_APIKEY }}
|
||||
run: |
|
||||
import os
|
||||
from pathlib import Path
|
||||
from ultralytics import YOLO, hub
|
||||
from ultralytics.yolo.utils import USER_CONFIG_DIR
|
||||
Path(USER_CONFIG_DIR / 'settings.yaml').unlink()
|
||||
key = os.environ['APIKEY']
|
||||
hub.reset_model(key)
|
||||
key, model_id = key.split('_')
|
||||
hub.login(key)
|
||||
model = YOLO(model_id)
|
||||
model.train()
|
||||
|
||||
Benchmarks:
|
||||
runs-on: ${{ matrix.os }}
|
||||
|
|
@ -154,6 +110,11 @@ jobs:
|
|||
run: |
|
||||
from ultralytics.yolo.utils.benchmarks import benchmark
|
||||
benchmark(model='${{ matrix.model }}-cls.pt', imgsz=160, half=False, hard_fail=0.61)
|
||||
- name: Benchmark PoseModel
|
||||
shell: python
|
||||
run: |
|
||||
from ultralytics.yolo.utils.benchmarks import benchmark
|
||||
benchmark(model='${{ matrix.model }}-pose.pt', imgsz=160, half=False, hard_fail=0.0)
|
||||
- name: Benchmark Summary
|
||||
run: |
|
||||
cat benchmarks.log
|
||||
|
|
@ -200,30 +161,38 @@ jobs:
|
|||
python --version
|
||||
pip --version
|
||||
pip list
|
||||
- name: Test detection
|
||||
- name: Test Detect
|
||||
shell: bash # for Windows compatibility
|
||||
run: |
|
||||
yolo task=detect mode=train data=coco8.yaml model=yolov8n.yaml epochs=1 imgsz=32
|
||||
yolo task=detect mode=train data=coco8.yaml model=yolov8n.pt epochs=1 imgsz=32
|
||||
yolo task=detect mode=val data=coco8.yaml model=runs/detect/train/weights/last.pt imgsz=32
|
||||
yolo task=detect mode=predict model=runs/detect/train/weights/last.pt imgsz=32 source=ultralytics/assets/bus.jpg
|
||||
yolo mode=export model=runs/detect/train/weights/last.pt imgsz=32 format=torchscript
|
||||
- name: Test segmentation
|
||||
yolo detect train data=coco8.yaml model=yolov8n.yaml epochs=1 imgsz=32
|
||||
yolo detect train data=coco8.yaml model=yolov8n.pt epochs=1 imgsz=32
|
||||
yolo detect val data=coco8.yaml model=runs/detect/train/weights/last.pt imgsz=32
|
||||
yolo detect predict model=runs/detect/train/weights/last.pt imgsz=32 source=ultralytics/assets/bus.jpg
|
||||
yolo export model=runs/detect/train/weights/last.pt imgsz=32 format=torchscript
|
||||
- name: Test Segment
|
||||
shell: bash # for Windows compatibility
|
||||
run: |
|
||||
yolo task=segment mode=train data=coco8-seg.yaml model=yolov8n-seg.yaml epochs=1 imgsz=32
|
||||
yolo task=segment mode=train data=coco8-seg.yaml model=yolov8n-seg.pt epochs=1 imgsz=32
|
||||
yolo task=segment mode=val data=coco8-seg.yaml model=runs/segment/train/weights/last.pt imgsz=32
|
||||
yolo task=segment mode=predict model=runs/segment/train/weights/last.pt imgsz=32 source=ultralytics/assets/bus.jpg
|
||||
yolo mode=export model=runs/segment/train/weights/last.pt imgsz=32 format=torchscript
|
||||
- name: Test classification
|
||||
yolo segment train data=coco8-seg.yaml model=yolov8n-seg.yaml epochs=1 imgsz=32
|
||||
yolo segment train data=coco8-seg.yaml model=yolov8n-seg.pt epochs=1 imgsz=32
|
||||
yolo segment val data=coco8-seg.yaml model=runs/segment/train/weights/last.pt imgsz=32
|
||||
yolo segment predict model=runs/segment/train/weights/last.pt imgsz=32 source=ultralytics/assets/bus.jpg
|
||||
yolo export model=runs/segment/train/weights/last.pt imgsz=32 format=torchscript
|
||||
- name: Test Classify
|
||||
shell: bash # for Windows compatibility
|
||||
run: |
|
||||
yolo task=classify mode=train data=imagenet10 model=yolov8n-cls.yaml epochs=1 imgsz=32
|
||||
yolo task=classify mode=train data=imagenet10 model=yolov8n-cls.pt epochs=1 imgsz=32
|
||||
yolo task=classify mode=val data=imagenet10 model=runs/classify/train/weights/last.pt imgsz=32
|
||||
yolo task=classify mode=predict model=runs/classify/train/weights/last.pt imgsz=32 source=ultralytics/assets/bus.jpg
|
||||
yolo mode=export model=runs/classify/train/weights/last.pt imgsz=32 format=torchscript
|
||||
yolo classify train data=imagenet10 model=yolov8n-cls.yaml epochs=1 imgsz=32
|
||||
yolo classify train data=imagenet10 model=yolov8n-cls.pt epochs=1 imgsz=32
|
||||
yolo classify val data=imagenet10 model=runs/classify/train/weights/last.pt imgsz=32
|
||||
yolo classify predict model=runs/classify/train/weights/last.pt imgsz=32 source=ultralytics/assets/bus.jpg
|
||||
yolo export model=runs/classify/train/weights/last.pt imgsz=32 format=torchscript
|
||||
- name: Test Pose
|
||||
shell: bash # for Windows compatibility
|
||||
run: |
|
||||
yolo pose train data=coco8-pose.yaml model=yolov8n-pose.yaml epochs=1 imgsz=32
|
||||
yolo pose train data=coco8-pose.yaml model=yolov8n-pose.pt epochs=1 imgsz=32
|
||||
yolo pose val data=coco8-pose.yaml model=runs/pose/train/weights/last.pt imgsz=32
|
||||
yolo pose predict model=runs/pose/train/weights/last.pt imgsz=32 source=ultralytics/assets/bus.jpg
|
||||
yolo export model=runs/pose/train/weights/last.pt imgsz=32 format=torchscript
|
||||
- name: Pytest tests
|
||||
shell: bash # for Windows compatibility
|
||||
run: pytest tests
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue