confidence level of segmentation

[DoDucNhan]

Hello everyone. I use the BEiT network for semantic segmentation and have trouble determining if my model segments the object correctly. In this case, there won't be any ground truth to validate. The result from "inference_segmentor" is only one array with the class id.

Since I try to extract a specific object (e.g. barrel) from a list of images, I want to apply a threshold to remove bad segmented objects. Is there any way I can obtain a confidence level that tells how confident a model is about that segmentation?

[changwsh12]

Please refer to
#3528 (comment)