可视化参考模型，卷积核可视化

import os
import torch
import torch.nn as nn
from PIL import Image
import torchvision.transforms as transforms
from torch.utils.tensorboard import SummaryWriter
import torchvision.utils as vutils
import torchvision.models as models
BASE_DIR = os.path.dirname(os.path.abspath(__file__))#获取根目录，里面abspath为了获取文件整个给目录
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")


if __name__ == "__main__":

    log_dir = os.path.join(BASE_DIR, "..", "results")#相对位置存放的地址
    # ----------------------------------- kernel visualization -----------------------------------
    writer = SummaryWriter(log_dir=log_dir, filename_suffix="_kernel")#写这些数据放置的位置

    # m1
    # alexnet = models.alexnet(pretrained=True)

    # m2
    path_state_dict =</