Я конвертирую контрольную точку тензорного потока, обученную с помощью Lan enet https://github.com/MaybeShewill-CV/lanenet-lane-detection, в UFF, чтобы я мог использовать TensorRT для выполнения вывода.
Имеется выходной узел в исходной модели тензорного потока lanenet_model / vgg_backend / binary_seg / ArgMax, который был [1, 256, 512] с типом int64. Однако после того, как я преобразовал контрольную точку тензорного потока в UFF и загрузил с помощью кода C ++ tensorrt, узел стал [1, 512] с типом float32.
Я не в чем дело. Кто-нибудь может помочь?