Realization of feature map visualization (heat map) based on torch

the code

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

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