feature map热力图和聚合掩码可视化

这段代码能够实现features map的热力图可视化，而且有两种保存方式。一种是保存为tensorboard的形式，另外一种是保存为.png图片形式。其中还穿插了一部分用来生成和验证聚合掩码的代码。web import torch import torch.nn as nn import torch.nn.functional as F from torchvision import mode

>>阅读原文<<