Custom UNet segmentation model uses only 2 colors for output map

yes, please refer to peopleSemSegNet TAO segmentation model in
GitHub - NVIDIA-AI-IOT/deepstream_tao_apps: Sample apps to demonstrate how to deploy models trained with TAO on DeepStream,

Yes, I did refer to this project, which leads to this result

Please try to git clone the github(GitHub - NVIDIA-AI-IOT/deepstream_tao_apps: Sample apps to demonstrate how to deploy models trained with TAO on DeepStream) again because there are some changes for Unet about 24 days ago.