【深度学习目标检测】Pytorch学习（2）tensorboard的使用

from torch.utils.tensorboard import SummaryWriter
from PIL import Image
import numpy as np

write = SummaryWriter("logs")
img_path = "DataSet/Data/train/ants_image/7759525_1363d24e88.jpg"
img_PIL = Image.open(img_path)
img_array = np.array(img_PIL)
write.add_image("test", img_array, 2, dataformats='HWC')
write.add_image("train", img_array, 1, dataformats='HWC')
# write.add_image()
# y=x
for i in range(100):
    write.add_scalar("y=2x", 2 * i, i)
write.close()