我将使用Lanenet https://github.com/MaybeShewill-CV/lanenet-lane-detection训练的tensorflow检查点转换为UFF，以便可以使用TensorRT进行推理。
原始张量流模型lanenet_model / vgg_backend / binary_seg / ArgMax中有一个输出节点，该节点为[1，256，512]，类型为int64。但是，在将tensorflow检查点转换为UFF并加载C ++ tensorrt代码后，该节点的类型为float32成为[1，512]。
我没有什么问题。有人可以帮忙吗？