Pytorch中的Tensorboard与Transforms怎么搭配使用
导读:本文共889.5字符,通常情况下阅读需要3分钟。同时您也可以点击右侧朗读,来听本文内容。按键盘←(左) →(右) 方向键可以翻页。
摘要: 直接上代码:fromPILimportImagefromtorch.utils.tensorboardimportSummaryWriterfromtorchvisionimporttransformsimportosroot_path="D:\\data\\basic\\Image"lable_path="aligne... ...
音频解说
目录
(为您整理了一些要点),点击可以直达。直接上代码:
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结果:
在命令行打开tensorboard:
但是在打开tensorboard
之前,要打开正确的位置,不然会报错,以此程序为例,需要在找到logs的目录。
在浏览器显示的结果:
</div> <div class="zixun-tj-product adv-bottom"></div> </div> </div> <div class="prve-next-news">
本文:
Pytorch中的Tensorboard与Transforms怎么搭配使用的详细内容,希望对您有所帮助,信息来源于网络。