HongChuang/ultralytics-ascend
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 3

COCO8 and COCO8-seg Pytest and CI updates (#307)

Browse source 
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: RangiLyu <lyuchqi@gmail.com>
This commit is contained in:
Glenn Jocher 2023-01-13 14:34:51 +01:00 • committed by GitHub
parent 2bc36d97ce
commit 70427579b8
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
13 changed files with 254 additions and 59 deletions
Download patch file Download diff file Expand all files Collapse all files

8
tests/test_python.py
View file

@ -37,18 +37,18 @@ def test_predict_dir():
def test_val():
model = YOLO(MODEL)
model.val(data="coco128.yaml", imgsz=32)
model.val(data="coco8.yaml", imgsz=32)
def test_train_scratch():
model = YOLO(CFG)
model.train(data="coco128.yaml", epochs=1, imgsz=32)
model.train(data="coco8.yaml", epochs=1, imgsz=32)
model(SOURCE)
def test_train_pretrained():
model = YOLO(MODEL)
model.train(data="coco128.yaml", epochs=1, imgsz=32)
model.train(data="coco8.yaml", epochs=1, imgsz=32)
model(SOURCE)
@ -102,7 +102,7 @@ def test_all_model_yamls():
def test_workflow():
model = YOLO(MODEL)
model.train(data="coco128.yaml", epochs=1, imgsz=32)
model.train(data="coco8.yaml", epochs=1, imgsz=32)
model.val()
model.predict(SOURCE)
model.export(format="onnx", opset=12) # export a model to ONNX format