这章是结合之前学习的Tensforboard与Transforms的一个练习。

直接上代码：

from PIL import Image
from torch.utils.tensorboard import SummaryWriter
from torchvision import transforms
import os

root_path = "D:\\data\\basic\\Image"
lable_path = "aligned"
img_dir = os.path.join(root_path, lable_path)
img_list = os.listdir(img_dir)
img_path = img_list[0]
# 加载img_path路径
img_path = os.path.join(img_dir, img_path)

# 将img以PIL格式打开
img = Image.open(img_path)

# 创建具体的SummaryWriter
writer = SummaryWriter("logs")

# 创建具体的ToTensor方法
to_tensor = transforms.ToTensor()

# 将img以PIL格式转化成Tensor格式
tensor_img = to_tensor(img)

# 使用writer.add_image方法
writer.add_image("tensor_img", tensor_img)

writer.close()

代码run结果:

在命令行打开tensorboard：

但是在打开tensorboard之前，要打开正确的位置，不然会报错，以此程序为例，需要在找到logs的目录。

在浏览器显示的结果：

免责声明：本站发布的内容（图片、视频和文字）以原创、转载和分享为主，文章观点不代表本网站立场，如果涉及侵权请联系站长邮箱：mmqy2019@163.com进行举报，并提供相关证据，查实之后，将立刻删除涉嫌侵权内容。

猜你喜欢

• Python中怎样用dict获取关键字和值呢

  这篇文章主要介绍了python中dict获取关键字与值的实现方式，具有很好的参考价值，希望对大家有所帮助。如有错误或未考虑完全的地方，望不吝赐教

• python绘制m1-m2数据怎样做，具体步骤是什么

  这篇文章主要介绍了Python数据分析之绘制m1-m2数据，文章基于python的相关资料展开详细的内容介绍，具有一定的参考价值，需要的小伙伴可以参考一下

• Python实现PyEMD两种方法及不同是什么

  两种实现形式区别前言PyEMD是经验模态分解 (EMD)及其变体的Python实现，EMD最流行的扩展之一是集成经验模态分解 (EEMD)，它利用了噪声辅助执行的集成。顾名思义，这个包

• tensorflow中slim是什么，常用函数有哪些

  这篇文章主要为大家介绍了python神经网络使用slim函数进行模型的训练及保存模型示例详解，有需要的朋友可以借鉴参考下，希望能够有所帮助，祝大家多多进步，早日升职加薪

• python中xlwt模块如何使用？

  这篇文章给大家分享是关于python xlwt模块的使用内容，包含基础类介绍，xlwt高级应用等等，下文有具有实例，感兴趣的朋友可以参考学习，接下来就跟随小编一起来看看xlwt模块的使用吧。