ZHAGNQ的博客

读SPED_900数据集下各文件夹图片并显示，以选取用于测试的图片import osimport pdbimport matplotlib.pyplot as pltimport matplotlib.image as mpimg# from PIL import Imagepath = r'D:\SPED_900'files = os.listdir(path)plt....

在网络的完成后，由于画图需要，我们可能需要可视化中间特征图，可使用以下代码import osfrom torch.utils.data import Dataset, DataLoaderfrom PIL import Imagefrom torchvision import transformspath = 'path'def default_loader(img_pat...