显示train loss 的方式
image.png
tensorboard —logdir=log
tensorboard —logdir=logs —port=6007
一般默认是6006,但是可能会有多个项目访问,所以可以手动指定打开端口。

.add_scalar()

画 y=x:
image.png
image.png

y=3x
image.png


如何向里面写图像

.add_image()

image.png
要求img_tensor的类型必须是 torch.Tensor 或者 numpy
image.png
可见,目前img的类型不满足要求:PIL.JpegImage
|
|
|
用opencv,或者numpy——>读取的类型为numpy
image.png
注意:图片的格式是(H,W,C),add_image()函数有默认图片格式要求,需要加上 dataformats=’HWC’


totensor

image.png
image.png


normalize

image.png
image.png