文章讲述了如何从PyTorch模型加载并解决报错问题，通过torch和ultralytics库将yolov8n.pt转换为ONNX格式，以及在AcuityToolkit中遇到的导入和量化模型的问题及解决方法。
直接运行
import torch
model = torch.load('models/yolov8n.pt')
报错
解决
pip install ultralytics
加了个
model.eval()
又报错了
原因是因为pt文件里面不包含网络结构，只有参数权重。
根据yolov8官网来做是这样
from ultralytics import YOLO
# Load a model
model = YOLO('models/yolov8n.pt') # load an official model
success = model.export(format="onnx", simplify=
最低0.47元/天 解锁文章
扫一扫
5万+
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
