Pytorch中的Tensorboard与Transforms怎么搭配使用(pytorch,tensorboard,transforms,开发技术)

时间:2024-05-02 13:57:53 作者 : 石家庄SEO 分类 : 开发技术
  • TAG :

直接上代码:

fromPILimportImagefromtorch.utils.tensorboardimportSummaryWriterfromtorchvisionimporttransformsimportosroot_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)#创建具体的SummaryWriterwriter=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结果:

Pytorch中的Tensorboard与Transforms怎么搭配使用

在命令行打开tensorboard:

Pytorch中的Tensorboard与Transforms怎么搭配使用

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

在浏览器显示的结果:

Pytorch中的Tensorboard与Transforms怎么搭配使用

 </div> <div class="zixun-tj-product adv-bottom"></div> </div> </div> <div class="prve-next-news">
本文:Pytorch中的Tensorboard与Transforms怎么搭配使用的详细内容,希望对您有所帮助,信息来源于网络。
上一篇:Java如何解实现一元稀疏多项式计算器下一篇:

11 人围观 / 0 条评论 ↓快速评论↓

(必须)

(必须,保密)

阿狸1 阿狸2 阿狸3 阿狸4 阿狸5 阿狸6 阿狸7 阿狸8 阿狸9 阿狸10 阿狸11 阿狸12 阿狸13 阿狸14 阿狸15 阿狸16 阿狸17 阿狸18