因为要用到tensorboard来可视化图像,但是在过程中遇到了点小问题,先看如下给的代码
from torch.utils.tensorboard import SummaryWriter
from torchvision import transforms
from PIL import Image
path = r'C:\Users\14499\PycharmProjects\vgg实现\data\test\0ae6668fa.png'
img = Image.open(path)
tensor = transforms.ToTensor()
img_tensor = tensor(img)
write = SummaryWriter('logs')
write.add_image("img",img_tensor)
write.close()
这里先测试张图片在tensorboard中的显示情况,这里我将文件存在logs文件夹中,然后在终端输入命令
(c) Microsoft Corporation。保留所有权利。
(venv) C:\Users\14499\PycharmProjects>cd vgg实现\pytorch基础
(venv) C:\Users\14499\PycharmProjects\vgg实现\pytorch基础>tensorboard --logdir=logs
TensorBoard 1.15.0 at http://lyp:6006/ (Press CTRL+C to quit)
然后发现打开网页就显示无法访问的问题,这里直接把我的计算机用户名当成我的主机了,接下来我们指定主机,命令如下
(venv) C:\Users\14499\PycharmProjects\vgg实现\pytorch基础>tensorboard --logdir=logs --host=127.0.0.1
TensorBoard 1.15.0 at http://127.0.0.1:6006/ (Press CTRL+C to quit)
再次打开网页,如图所示成功解决!
如果上述代码帮助您很多,可以打赏下以减少服务器的开支吗,万分感谢!


点击此处登录后即可评论