PRIS-CV / AP-CNN_Pytorch-master

Weakly Supervised Attention Pyramid Convolutional Neural Network for Fine-Grained Visual Classification (TIP2021)
MIT License
47 stars 4 forks source link

关于可视化 #4

Closed MercuialC closed 2 years ago

MercuialC commented 2 years ago

感谢您优秀的工作! 想请教一下对于论文中各层特征的黑白可视化图是用什么工具实现,能否提供可视化的思路或者代码。 非常感谢!

dingyf05 commented 2 years ago

论文中我们对各层特征的spatial attention mask进行了可视化,将attention mask上各点的响应值由0~1映射到0~255转化为灰度图即可,https://github.com/PRIS-CV/AP-CNN_Pytorch-master/blob/master/utils/utils.py中已给出相关代码实现