Custom UNet segmentation model uses only 2 colors for output map

[1529418187]

@1529418187

deepstream-segment-test

You are right, but I use TAO convert to verify that the deployment is successful。

Is your official deep stream segment test model trained with TAO? It seems to be trained with caffe. Have you tested the segmentation model trained with TAO and successfully deployed it to deep stream?