yolov8
  KYTr22BsbLeM 2023年11月02日 37 0

https://github.com/ultralytics/ultralytics/tree/main/ultralytics/cfg/models

https://docs.ultralytics.com/models/yolov8/#supported-tasks

https://docs.ultralytics.com/models/yolov8/#supported-modes


root@c6f94564c001:/home/xxx/demo_3.0/yolov8#pip install ultralytics
root@c6f94564c001:/home/xxx/demo_3.0/yolov8# python generat_onnx.py
Ultralytics YOLOv8.0.163 🚀 Python-3.8.10 torch-2.0.1+cu117 CPU (Intel Core(TM) i7-7800X 3.50GHz)
YOLOv8n summary (fused): 168 layers, 3151904 parameters, 0 gradients

PyTorch: starting from 'yolov8n.pt' with input shape (1, 3, 640, 640) BCHW and output shape(s) (1, 84, 8400) (6.2 MB)

ONNX: starting export with onnx 1.14.0 opset 17...
============= Diagnostic Run torch.onnx.export version 2.0.1+cu117 =============
verbose: False, log level: Level.ERROR
======================= 0 NONE 0 NOTE 0 WARNING 0 ERROR ========================

ONNX: export success ✅ 0.5s, saved as 'yolov8n.onnx' (12.2 MB)

Export complete (3.2s)
Results saved to /home/xxx/demo_3.0/yolov8
Predict:         yolo predict task=detect model=yolov8n.onnx imgsz=640
Validate:        yolo val task=detect model=yolov8n.onnx imgsz=640 data=coco.yaml
Visualize:       https://netron.app
root@c6f94564c001:/home/xxx/demo_3.0/yolov8# ls
generat_onnx.py  yolov8n.onnx  yolov8n.pt
root@c6f94564c001:/home/xxx/demo_3.0/yolov8# cat generat_onnx.py
from ultralytics import YOLO
model = YOLO("./yolov8n.pt")
success = model.export(format="onnx")


* batch = 1 is important here,
    even export support dynamic axis using dynamic=True,
    sometimes it just fail to export
model.export(format='onnx', simplify=True, imgsz=[640,640], batch=1)

https://medium.com/mlearning-ai/stitching-non-max-suppression-nms-to-yolov8n-on-exported-onnx-model-1c625021b22* model is the YOLOv8n trained (YOLO class)

https://aitechtogether.com/python/132556.html

import os
from ultralytics import YOLO
model = YOLO("yolov8s.yaml")
model=YOLO("../pretrained_model/yolov8s.pt")
#success=model.export(format="onnx")
success=model.export(format="onnx", half=False, dynamic=True, opset=17)
#success=model.export(format="onnx", dynamic=True)

print("demo")

https://blog.csdn.net/u013302570/article/details/130525295

【版权声明】本文内容来自摩杜云社区用户原创、第三方投稿、转载,内容版权归原作者所有。本网站的目的在于传递更多信息,不拥有版权,亦不承担相应法律责任。如果您发现本社区中有涉嫌抄袭的内容,欢迎发送邮件进行举报,并提供相关证据,一经查实,本社区将立刻删除涉嫌侵权内容,举报邮箱: cloudbbs@moduyun.com

  1. 分享:
最后一次编辑于 2023年11月08日 0

暂无评论

推荐阅读