How would I use the output of any of the segnet programs? If I just wanted to know, for each pixel, what class was the highest probability, is there any way to do that? The colorful mask is not too easy to use because of all the blending it does, I tried setting alpha to 0 and 255, it did not really help. Is there some other form of output it can generate that I just didnt see?
Hi @duner, you can get the classID mask by passing in a uint8 image to the
For example, see this code from the Python segnet example:
This code gets the class ID mask, and then computes a histogram of all the class ID’s (showing the distribution of classes in the image)
There is also a C++ API for getting the class ID mask, see here: https://github.com/dusty-nv/jetson-inference/blob/022739abc7631fa853ae63550ec091cd838b2105/c/segNet.h#L256