|
|
|
@ -87,15 +87,18 @@ jobs:
|
|
|
|
|
yolo task=detect mode=train model=yolov8n.yaml data=coco128.yaml epochs=1 imgsz=64
|
|
|
|
|
yolo task=detect mode=val model=runs/detect/train/weights/last.pt imgsz=64
|
|
|
|
|
yolo task=detect mode=predict model=runs/detect/train/weights/last.pt imgsz=64 source=ultralytics/assets/bus.jpg
|
|
|
|
|
yolo mode=export model=runs/detect/train/weights/last.pt imgsz=64 format=torchscript
|
|
|
|
|
- name: Test segmentation
|
|
|
|
|
shell: bash # for Windows compatibility
|
|
|
|
|
run: |
|
|
|
|
|
yolo task=segment mode=train model=yolov8n-seg.yaml data=coco128-seg.yaml epochs=1 imgsz=64
|
|
|
|
|
yolo task=segment mode=val model=runs/segment/train/weights/last.pt data=coco128-seg.yaml imgsz=64
|
|
|
|
|
yolo task=segment mode=predict model=runs/segment/train/weights/last.pt imgsz=64 source=ultralytics/assets/bus.jpg
|
|
|
|
|
yolo mode=export model=runs/segment/train/weights/last.pt imgsz=64 format=torchscript
|
|
|
|
|
- name: Test classification
|
|
|
|
|
shell: bash # for Windows compatibility
|
|
|
|
|
run: |
|
|
|
|
|
yolo task=classify mode=train model=yolov8n-cls.yaml data=mnist160 epochs=1 imgsz=32
|
|
|
|
|
yolo task=classify mode=val model=runs/classify/train/weights/last.pt data=mnist160 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
|
|
|
|
|