CoreML NMS and half fixes (#143)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
This commit is contained in:
parent
55bdca6768
commit
172cef2d20
18 changed files with 95 additions and 414 deletions
63
tests/test_cli.py
Normal file
63
tests/test_cli.py
Normal file
|
|
@ -0,0 +1,63 @@
|
|||
import os
|
||||
from pathlib import Path
|
||||
|
||||
from ultralytics.yolo.utils import ROOT, SETTINGS
|
||||
|
||||
MODEL = Path(SETTINGS['weights_dir']) / 'yolov8n.pt'
|
||||
CFG = 'yolov8n.yaml'
|
||||
|
||||
|
||||
def test_checks():
|
||||
os.system('yolo mode=checks')
|
||||
|
||||
|
||||
# Train checks ---------------------------------------------------------------------------------------------------------
|
||||
def test_train_detect():
|
||||
os.system(f'yolo mode=train task=detect model={MODEL} data=coco128.yaml imgsz=32 epochs=1')
|
||||
|
||||
|
||||
def test_train_segment():
|
||||
os.system('yolo mode=train task=segment model=yolov8n-seg.yaml data=coco128-seg.yaml imgsz=32 epochs=1')
|
||||
|
||||
|
||||
def test_train_classify():
|
||||
pass
|
||||
|
||||
|
||||
# Val checks -----------------------------------------------------------------------------------------------------------
|
||||
def test_val_detect():
|
||||
os.system(f'yolo mode=val task=detect model={MODEL} data=coco128.yaml imgsz=32 epochs=1')
|
||||
|
||||
|
||||
def test_val_segment():
|
||||
pass
|
||||
|
||||
|
||||
def test_val_classify():
|
||||
pass
|
||||
|
||||
|
||||
# Predict checks -------------------------------------------------------------------------------------------------------
|
||||
def test_predict_detect():
|
||||
os.system(f"yolo mode=predict model={MODEL} source={ROOT / 'assets'}")
|
||||
|
||||
|
||||
def test_predict_segment():
|
||||
pass
|
||||
|
||||
|
||||
def test_predict_classify():
|
||||
pass
|
||||
|
||||
|
||||
# Export checks --------------------------------------------------------------------------------------------------------
|
||||
def test_export_detect_torchscript():
|
||||
os.system(f'yolo mode=export model={MODEL} format=torchscript')
|
||||
|
||||
|
||||
def test_export_segment_torchscript():
|
||||
pass
|
||||
|
||||
|
||||
def test_export_classify_torchscript():
|
||||
pass
|
||||
Loading…
Add table
Add a link
Reference in a new issue