由于pytorch中没有tensorboard的安装包需要我们自行安装
pip install tensorboard
安装后 from torch.utils.tensorboard import SummaryWriter此时还会报错的话需要安装tensorboardX
这里使用清华镜像安装tensorboardX的包
pip install -i https://mirrors.aliyun.com/pypi/simple/ tensorboardX
from tensorboardX import SummaryWriter
这样导入tensorboard便不会报错啦
from tensorboardX import SummaryWriter
import cv2 as cv
str = 'D:/dataset/train/ants/0013035.jpg'
img = cv.imread(str, 1)
write = SummaryWriter("logs") # 在绘制tensorboard时会将所有log文件下的内容均进行绘制 处理方法 1、删除文件内容2、重新指定文件家
write.add_image("test", img, 1, dataformats='HWC')
for i in range(100):
write.add_scalar("y = 2x", 2*i, i) # 标签 y轴x轴
write.close()
此时便会在相对路径下创建一个logs的文件
Terminal下输入tensorboard --logdir logs
点击这个网站即可