MinaGhadimiAtigh / HyperbolicImageSegmentation

Hyperbolic Image Segmentation, CVPR 2022
https://minaghadimi.github.io/papers/HIS/index.html
72 stars 7 forks source link

How to get a Hyperbolic Uncertainty map? #1

Closed shuaige-inthewind closed 2 years ago

shuaige-inthewind commented 2 years ago

How do I get the Hyperbolic Uncertainty map of a picture?

MinaGhadimiAtigh commented 2 years ago

The code in generate_confidence_map.py is to generate the confidence map (with has an inverse relation with uncertainty) given input image.

First, you should train a segmentation model. Then, the code will load the model and generate and show the confidence map.