加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
文件
克隆/下载
pth2onnx.py 460 Bytes
一键复制 编辑 原始数据 按行查看 历史
luoxl 提交于 2024-07-02 09:53 . 增加openvino和onnxruntime加速推理
'''
Description:
version:
Author: aps.auto
Date: 2024-06-26 11:39:40
LastEditors: Luoxl
LastEditTime: 2024-07-02 08:39:50
'''
from ultralytics import YOLO
import ultralytics
import torch
import numpy as np
# Load a model
model = ultralytics.YOLO("yolov8n-seg.pt") # load an official model
model = ultralytics.YOLO("D:/BaiduSyncdisk/Projects/yolov8-segment/yolov8n-seg-AGI.pt") # load a custom model
result = model.export(format='onnx') #yolov8原生转换
马建仓 AI 助手
尝试更多
代码解读
代码找茬
代码优化