Как настроить вывод модели onnx при преобразовании из tenorflow в onnx?

Question

Я пытаюсь преобразовать сохраненную модель (файл тензорного потока .pb) yolov4 в onnx для вывода tenorrt. Мне удалось установить форму ввода перед преобразованием в onnx(как показано в документации с помощью параметра входной сигнатуры). Однако я не могу найти нигде, чтобы установить выходную сигнатуру. Я хочу установить это, потому что форма выходного слоя в соответствии с кратким описанием модели. который при выполнении вывода модели в тензорном потоке возвращается (значение 1-го и 3-го индексов фиксировано, а 2-й элемент меняется). Я попытался преобразовать его в onnx, не задав форму вывода, а затем попытался выполнить вывод tenorrt. Однако выходная форма тензора всегда будет который никогда не может быть преобразован в . Можно ли решить эту проблему, задав форму вывода модели onnx, или мне нужно выполнить какую-либо постобработку вывода tenorrt?