Custom UNet segmentation model uses only 2 colors for output map

[1529418187]

@fanzh

there will be multiple classes, please refer to Gst-nvsegvisual — DeepStream 6.1.1 Release documentation , nvsegvisual will use different colors if there are different classe

There are 22 class, but there is only one. I replace my model with the official model at /opt/nvidia/deepstream/deepstream/sources/apps/sample_apps/deepstream-segmentation-test. The number of classes is consistent with the official model