feature map熱力圖和聚合掩碼可視化

這段代碼能夠實現features map的熱力圖可視化，而且有兩種保存方式。一種是保存爲tensorboard的形式，另外一種是保存爲.png圖片形式。其中還穿插了一部分用來生成和驗證聚合掩碼的代碼。web import torch import torch.nn as nn import torch.nn.functional as F from torchvision import mode

>>阅读原文<<