Assignment visualization

[TeodorChiaburu]

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!