У меня есть модель CNN, которая была обучена в Pytorch на основе формата данных N (пакет) x C (канал) x H (высота) x W (ширина). Я сохранил предварительно обученную модель как model.pth. После этого я преобразовал предварительно обученную модель из model.pth -> model.onnx с помощью существующей функции:
torch.onnx.export(model, dummy_input, «model.onnx»)
Затем я преобразовал этот model.onnx -> model.pb с помощью модуля ниже:
import onnx from onnx_tf.backend import prepare model_onnx = onnx.load(‘model.onnx’) tf_rep = prepare(model_onnx) tf_rep.export_graph(‘model.pb’)
Проблема в том, что я хочу использовать этот model.pb на устройстве с ЦП, которому требуется формат данных NHWC. Однако моя модель основана на формате данных NCHW. Есть ли какой-либо метод, который может преобразовать формат данных этого model.pb из NCHW в NHWC?
Уточните свой вопрос, см. Как задать вопрос и предоставьте минимальный воспроизводимый пример. — person MY C schedule 04.02.2020
Извини за это. Я обновил свой вопрос. Это лучше? Спасибо:) — person MY C schedule 04.02.2020
Источник: