zxhuang1698 / interpretability-by-parts

Code repository for "Interpretable and Accurate Fine-grained Recognition via Region Grouping", CVPR 2020 (Oral)
https://www.biostat.wisc.edu/~yli/cvpr2020-interp/
128 stars 27 forks source link

Assignment visualization #11

Open TeodorChiaburu opened 2 years ago

TeodorChiaburu commented 2 years ago

Hello! I have retrained a ResNet on CUB, but when I run the visualization script, every pixel in the images is assigned a color mask, even those not belonging to any body part. In your demo.jpg, I see that your model only masks real relevant pixels for the learned parts (and maybe some of the neighboring pixels as well at most), while leaving non-relevant pixels unmasked. Any idea what may be causing this? Thank you for your great work!