nikitadurasov / masksembles

Official repository for the paper "Masksembles for Uncertainty Estimation" (CVPR 2021).
https://www.norange.io/projects/masksembles/
MIT License
98 stars 15 forks source link

Mask generation functions *sometimes* fail to produce proper masks #4

Open nikitadurasov opened 3 years ago

nikitadurasov commented 3 years ago

Here is a problem. When result from here fails to produce proper masks -- the number of features in those masks isn't equal to the number of channels required.

The current solution is just to try searching for a new value of scale parameter via grid-search -- trying values from np.linspace. This solution is far from perfect since for some function configurations it fails to find proper scale. Need to implement something more sophisticated like binary search or something so it would work for every possible configuration.

Beichen-Ma commented 11 months ago

Here I change the condition from ">=" to "==", and resolved the problem of fails to produce mask.