Реализация визуализации карты признаков (тепловой карты) на основе факела

код

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

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