pytorch之TensorBoard的简单使用方法

        这部分主要记录两个重要函数 add_image 和 add_scalar 的使用：

from torch.utils.tensorboard import SummaryWriter
import cv2 as cv
#实例化
writer=SummaryWriter(log_dir="logs")
#绘制图片
image_path1="flower_data/train/1/image_06737.jpg"
image_path2="flower_data/train/6/image_07162.jpg"
img1=cv.imread(image_path1)
img2=cv.imread(image_path2)
print(img1.shape)
writer.add_image("train",img1,1,dataformats='HWC')
writer.add_image("train",img2,2,dataformats='HWC')
#绘制曲线
for i in range(100):
    writer.add_scalar(tag="y=2x",scalar_value=2*i,global_step=i)
writer.close()