|
|
@ -193,14 +193,14 @@ def test_export_paddle(): |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
@pytest.mark.slow |
|
|
|
@pytest.mark.slow |
|
|
|
def test_export_ncnn(): |
|
|
|
def test_export_mnn(): |
|
|
|
"""Test YOLO exports to NCNN format.""" |
|
|
|
"""Test YOLO exports to MNN format (WARNING: MNN test must precede NCNN test or CI error on Windows).""" |
|
|
|
file = YOLO(MODEL).export(format="ncnn", imgsz=32) |
|
|
|
file = YOLO(MODEL).export(format="mnn", imgsz=32) |
|
|
|
YOLO(file)(SOURCE, imgsz=32) # exported model inference |
|
|
|
YOLO(file)(SOURCE, imgsz=32) # exported model inference |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
@pytest.mark.slow |
|
|
|
@pytest.mark.slow |
|
|
|
def test_export_mnn(): |
|
|
|
def test_export_ncnn(): |
|
|
|
"""Test YOLO exports to MNN format.""" |
|
|
|
"""Test YOLO exports to NCNN format.""" |
|
|
|
file = YOLO(MODEL).export(format="mnn", imgsz=32) |
|
|
|
file = YOLO(MODEL).export(format="ncnn", imgsz=32) |
|
|
|
YOLO(file)(SOURCE, imgsz=32) # exported model inference |
|
|
|
YOLO(file)(SOURCE, imgsz=32) # exported model inference |
|
|
|