可视化网络中间图特征文件

 在网络的完成后，由于画图需要，我们可能需要可视化中间特征图，可使用以下代码

import os
from torch.utils.data import Dataset, DataLoader
from PIL import Image
from torchvision import transforms

path = 'path'


def default_loader(img_path):
    return Image.open(img_path).convert('RGB')

transform = transforms.Compose([
    transforms.Resize((320, 240)),  # 缩放到 320 * 240 大小
    # transforms.Resize((240, 320)),  # 缩放到 320 * 240 大小
    transforms.ToTensor(),
    transforms.Normalize(mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225])
])

def visualize():
    path_list = ['SPED_900/220/20140201_093920.jpg', 'SPED_900/1/20140217_185540.jpg', 'SPED_900/1/20140216_165546.jpg', 'SPED_900/1/20140217_182550.jpg']
    img_test  = []
    for name in path_list:
        img1_path = path + name
        img1 = default_loader(img1_path)
        img1 = transform(img1)
        img_test.append(img1)