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

51
tests/test_engine.py
View file

@ -1,6 +1,5 @@
# Ultralytics YOLO 🚀, GPL-3.0 license
from ultralytics import YOLO
from ultralytics.yolo.configs import get_config
from ultralytics.yolo.utils import DEFAULT_CONFIG, ROOT
from ultralytics.yolo.v8 import classify, detect, segment
@ -13,9 +12,10 @@ SOURCE = ROOT / "assets"
def test_detect():
overrides = {"data": "coco128.yaml", "model": CFG_DET, "imgsz": 32, "epochs": 1, "save": False}
CFG.data = "coco128.yaml"
# trainer
overrides = {"data": "coco8.yaml", "model": CFG_DET, "imgsz": 32, "epochs": 1, "save": False}
CFG.data = "coco8.yaml"
# Trainer
trainer = detect.DetectionTrainer(overrides=overrides)
trainer.train()
trained_model = trainer.best
@ -24,12 +24,10 @@ def test_detect():
val = detect.DetectionValidator(args=CFG)
val(model=trained_model)
# predictor
pred = detect.DetectionPredictor(overrides={"imgsz": [640, 640]})
i = 0
for _ in pred(source=SOURCE, model="yolov8n.pt", return_outputs=True):
i += 1
assert i == 2, "predictor test failed"
# Predictor
pred = detect.DetectionPredictor(overrides={"imgsz": [64, 64]})
result = pred(source=SOURCE, model="yolov8n.pt", return_outputs=True)
assert len(list(result)), "predictor test failed"
overrides["resume"] = trainer.last
trainer = detect.DetectionTrainer(overrides=overrides)
@ -43,11 +41,11 @@ def test_detect():
def test_segment():
overrides = {"data": "coco128-seg.yaml", "model": CFG_SEG, "imgsz": 32, "epochs": 1, "save": False}
CFG.data = "coco128-seg.yaml"
overrides = {"data": "coco8-seg.yaml", "model": CFG_SEG, "imgsz": 32, "epochs": 1, "save": False}
CFG.data = "coco8-seg.yaml"
CFG.v5loader = False
# YOLO(CFG_SEG).train(**overrides) # works
# YOLO(CFG_SEG).train(**overrides) # This works
# trainer
trainer = segment.SegmentationTrainer(overrides=overrides)
trainer.train()
@ -57,14 +55,12 @@ def test_segment():
val = segment.SegmentationValidator(args=CFG)
val(model=trained_model)
# predictor
pred = segment.SegmentationPredictor(overrides={"imgsz": [640, 640]})
i = 0
for _ in pred(source=SOURCE, model="yolov8n-seg.pt", return_outputs=True):
i += 1
assert i == 2, "predictor test failed"
# Predictor
pred = segment.SegmentationPredictor(overrides={"imgsz": [64, 64]})
result = pred(source=SOURCE, model="yolov8n-seg.pt", return_outputs=True)
assert len(list(result)) == 2, "predictor test failed"
# test resume
# Test resume
overrides["resume"] = trainer.last
trainer = segment.SegmentationTrainer(overrides=overrides)
try:
@ -81,8 +77,9 @@ def test_classify():
CFG.data = "mnist160"
CFG.imgsz = 32
CFG.batch = 64
# YOLO(CFG_SEG).train(**overrides) # This works
# trainer
# YOLO(CFG_SEG).train(**overrides) # works
# Trainer
trainer = classify.ClassificationTrainer(overrides=overrides)
trainer.train()
trained_model = trainer.best
@ -91,9 +88,7 @@ def test_classify():
val = classify.ClassificationValidator(args=CFG)
val(model=trained_model)
# predictor
pred = classify.ClassificationPredictor(overrides={"imgsz": [640, 640]})
i = 0
for _ in pred(source=SOURCE, model=trained_model, return_outputs=True):
i += 1
assert i == 2, "predictor test failed"
# Predictor
pred = classify.ClassificationPredictor(overrides={"imgsz": [64, 64]})
result = pred(source=SOURCE, model=trained_model, return_outputs=True)
assert len(list(result)) == 2, "predictor test failed"