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

最新推荐文章于 2024-02-21 20:43:52 发布

beyond951

最新推荐文章于 2024-02-21 20:43:52 发布

阅读量340

点赞数

分类专栏：目标检测文章标签：深度学习 pytorch 目标检测

本文链接：https://blog.csdn.net/beyond951/article/details/126309437

版权

目标检测专栏收录该内容

3 篇文章 0 订阅

订阅专栏

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()