Custom TAO unet model classifying only two classes on Deepstream!

eenav:

I am running ./apps/tao_segmentation/ds-tao-segmentation with a config file modified from peopleSemSegNet_tao which was the closest segmentation related tao apps config .

The network-mode=1 was at 1 because of a previous test, but I run in all three modes and makes no difference.

But I think the problem is elsewhere, perhaps in tao-converter since using the model in C++ and tensorRT produce an identical result.