У меня есть модель 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