Convert tensorflow model to UFF, output node size wrong

I am converting the tensorflow checkpoint trained with Lanenet https://github.com/MaybeShewill-CV/lanenet-lane-detection to UFF so I can use TensorRT to run inference.

There is a output node in the orignal tensorflow model lanenet_model/vgg_backend/binary_seg/ArgMax, which was [1, 256, 512] with type int64. However, after I convert the tensorflow checkpoint to UFF and load with C++ tensorrt code, the node became [1, 512] with type float32.

I do not what’s wrong. Anyone can help?