yolact icon indicating copy to clipboard operation
yolact copied to clipboard

Published 20 hours ago • verified publisher icon dbolya

Probability Maps

Open MiguelAngeloMartinsRibeiro opened this issue 3 years ago • 2 comments

Hi @dbolya Is it possible to get the probability maps, i.e, instead of binary masks get a matrix with the probability of each pixel?

Thank you for your time

MiguelAngeloMartinsRibeiro avatar Dec 13 '21 12:12 MiguelAngeloMartinsRibeiro

The softmax output would be useful as well

MiguelAngeloMartinsRibeiro avatar Jan 11 '22 23:01 MiguelAngeloMartinsRibeiro

You are right the probability map Output would really be nice for Heatmap Visualisation of a specific class. Convert the Modell to ONNX and visualize it with https://netron.app/ So you can find the specific layers you need. You can also use pre converted ONNX yolact/yolactEdge Modells from https://github.com/PINTO0309/PINTO_model_zoo

If you use Python then you can use this as starting point for Postprocessing Layers: https://github.com/openvinotoolkit/open_model_zoo/tree/master/demos/instance_segmentation_demo/python

For c++ use this: https://github.com/PINTO0309/PINTO_model_zoo/blob/main/085_Yolact_Edge/30_Full_Converted_mbnv2_550x550/yolact_test.cpp

telemetrieTP23 avatar Jan 14 '22 15:01 telemetrieTP23