Using maskensemble to the shape of the tensor [B, H, W]

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

Using maskensemble to the shape of the tensor [B, H, W] #11

Open diogofreitasribeiro opened 2 years ago

diogofreitasribeiro commented 2 years ago

Hi,

I have read the paper about masksembles, congratulations for the article. I'm trying to adapt to my problem. In my research, I don't work with images and the shape of the tensor is [B, H, W].

How can I adapt the maskensemble to the shape of my problem?