diff --git a/tests/test_ncnn.py b/tests/test_ncnn.py index 0c9c2d7f4..8aa33b8b8 100644 --- a/tests/test_ncnn.py +++ b/tests/test_ncnn.py @@ -4,7 +4,7 @@ from ultralytics import YOLO def test_ncnn(): # export ncnn and do inference model = YOLO("yolo11n.pt") - model.export(imgsz=160, format="ncnn", half=False, int8=False, device="cpu", verbose=False) + filename = model.export(imgsz=160, format="ncnn", half=False, int8=False, device="cpu", verbose=False) # exported_model = YOLO(filename, task=model.task) # exported_model.val( # data="coco8.yaml", @@ -17,5 +17,7 @@ def test_ncnn(): # verbose=False, # ) # then export onnx + +def test_onnx(): model = YOLO("yolo11n.pt") model.export(imgsz=160, format="onnx", half=False, int8=False, device="cpu", verbose=False)