AttributeError: объект ConvertModel не имеет атрибута seek

Я попытался преобразовать модель MATLAB в PyTorch с помощью ONNX, как это было предложено здесь Эндрю Нагибом:

Как импортировать модели глубокого обучения из MATLAB в 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?


python pytorch attributeerror onnx seek
person Oubi    schedule 21.02.2021    source источник

Ответы (1)


arrow_upward
0
arrow_downward

Предполагая, что 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, и я не знаю, что он делает. Но если в двоичном файле возвращается файловый объект, то приведенное выше может вам помочь.

person joao    schedule 21.02.2021
comment
Работает спасибо! - person Oubi; 23.02.2021