如何解决AttributeError: 'ConvertModel' 对象没有属性 'seek'
我尝试使用 ONNX 将 MATLAB 模型转换为 PyTorch,就像 Andrew Naguib 在此处提出的那样:
How to import deep learning models from MATLAB to PyTorch?
我尝试使用以下代码运行模型:
import onnx
from onnx2pytorch import ConvertModel
import torch
onnx_model = onnx.load ('resnet50.onnx')
pytorch_model = ConvertModel(onnx_model)
model = torch.load(pytorch_model)
但是我收到了这个错误:
AttributeError: 'ConvertModel' 对象没有属性 'seek'。你可以 只有来自可查找文件的 torch.load。请预先加载数据 进入像 io.BytesIO 这样的缓冲区,然后尝试从它加载。
请问我该如何解决?关于如何“将数据预加载到像 io.BytesIO 这样的缓冲区中”的任何想法?
解决方法
假设 my_data.dat
是一个包含二进制数据的文件,以下代码将其加载到可查找的 ioBytesIO 缓冲区中:
import io
with open('my_data.dat','rb') as f:
buf = io.BytesIO(f.read())
您现在可以编写诸如
buf.seek(4)
和
x = buf.read(1)
当然,就您而言,您正在使用 onnx.load
方法,我不知道它的作用是什么。但是如果确实返回一个文件对象,在一个二进制文件上,那么上面的内容可能对你有帮助。
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。