xdobro23/YOLOv8-16bit
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Detection support (#60)

Browse Source 
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: Laughing-q <1185102784@qq.com>
This commit is contained in:
Ayush Chaurasia
2022-12-03 21:01:23 +05:30
committed by GitHub
parent 5a52e7663a
commit 7ec7cf3aef
9 changed files with 545 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
.github/workflows/ci.yaml vendored
View File

@ -90,15 +90,16 @@ jobs:
- name: Test detection
shell: bash # for Windows compatibility
run: |
echo "TODO"
yolo task=detect mode=train model=yolov5n.yaml data=coco128.yaml epochs=1 img_size=64
yolo task=detect mode=val model=runs/exp/weights/last.pt img_size=64
- name: Test segmentation
shell: bash # for Windows compatibility
# TODO: redo val test without hardcoded weights
run: |
yolo task=segment mode=train model=yolov5n-seg.yaml data=coco128-seg.yaml epochs=1 img_size=64
yolo task=segment mode=val model=runs/exp/weights/last.pt data=coco128-seg.yaml img_size=64
yolo task=segment mode=val model=runs/exp2/weights/last.pt data=coco128-seg.yaml img_size=64
- name: Test classification
shell: bash # for Windows compatibility
run: |
yolo task=classify mode=train model=resnet18 data=mnist160 epochs=1 img_size=32
yolo task=classify mode=val model=runs/exp2/weights/last.pt data=mnist160
yolo task=classify mode=val model=runs/exp3/weights/last.pt data=mnist160