导包
from pytorch.utils.tensorboard import SummaryWriter
实例化
writer = SummaryWriter("log文件名")
1.绘曲线.add_scalar(如)
python
for i in range(100):
writer.add_scalar("y=x",i,i)
writer.close()
2.绘图像.add_image(如)
python
img = cv.imread("***/**/***.jpg")
writer.add_image("test",img[:,:,::-1],1(step),dataformats="HWC")
writer.close()