文章作者:激萌の小宅
促销:¥0
价格:¥0
配送方式:
购买后立即生效(如购买异常,请联系站长)
付款之后一定要等待自动跳转结束,否则购买可能会失败
-
0 天
有效期
-
0
总销量
-
0
累计评价
YOLOv8 pt转onnx文件与测试
pt转onnx
新建个py文件,然后执行下面代码即可
from ultralytics import YOLO if __name__ == '__main__': model = YOLO('./ultralytics/models/v8/yolov8.yaml') model = YOLO('./runs/detect/train10/weights/best.pt') # 注意设置opset=11 success = model.export(format='onnx', opset=11)
然后会在pt文件目录下生成一个onnx文件
onnx文件测试
测试文件:ultralytics-main\examples\YOLOv8-OpenCV-ONNX-Python\main.py
需要将这个main.py文件拷贝到ultralytics-main目录下,执行才行,我把他重命令为test_onnx.py文件了,执行命令如下:
python test_onnx.py --model ./runs/detect/train10/weights/best.onnx --img datasets/coco128/images/train2017/000027.jpg