12.13 pytorch学习(tensorboard的使用)
12.13 pytorch学习(tensorboard的使用)
代码:from torch.utils.tensorboard import SummaryWriter#SummaryWriter文件可视化import numpy as npfrom PIL import Image#文件夹为logswriter = SummaryWriter("logs")Image_path = "dataset\\train\\ants\\0013035.jpg"img_PIL = Image.open(Image_path)img_array = np.ar


