pytorch学习

tensorboard学习

from torch.utils.tensorboard import SummaryWriter
#Tensorboaed用来展示图像
import numpy as np
from PIL import Image

writer = SummaryWriter("logs")
#writer = SummaryWriter(log_dir='logs',flush_secs=60)
#log_dir：tensorboard文件的存放路径
#flush_secs：表示写入tensorboard文件的时间间隔
image_path = "D:\\pythonProject1\\数据集\\dataset\\train\\ants\\0013035.jpg"
img_PIL = Image.open(image_path)
img_array = np.array(img_PIL)

print(type(img_array))
print(img_array.shape)
#writer.add_image = ("test",img_array,1, dataformats='HWC')

for i in range(100):
    writer.add_scalar("y=x",i,i)
writer.close()

常见的Transforms