基于 torch 的特征图可视化（热图）的实现

代码

import torch
from torchvision import models
from torchvision import transforms
from PIL import Image
import matplotlib.pyplot as plt

# 加载预训练模型
model = models.resnet50(pretrained=True