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

這篇文章主要介紹了Pytorch中的Tensorboard與Transforms怎么搭配使用，具有一定借鑒價(jià)值，感興趣的朋友可以參考下，希望大家閱讀完這篇文章之后大有收獲，下面讓小編帶著大家一起了解一下。

直接上代碼：

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格式打開(kāi)
img = Image.open(img_path)

# 創(chuàng)建具體的SummaryWriter
writer = SummaryWriter("logs")

# 創(chuàng)建具體的ToTensor方法
to_tensor = transforms.ToTensor()

# 將img以PIL格式轉(zhuǎn)化成Tensor格式
tensor_img = to_tensor(img)

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

writer.close()

代碼run結(jié)果:

在命令行打開(kāi)tensorboard：

但是在打開(kāi)tensorboard之前，要打開(kāi)正確的位置，不然會(huì)報(bào)錯(cuò)，以此程序?yàn)槔枰谡业絣ogs的目錄。

在瀏覽器顯示的結(jié)果：

感謝你能夠認(rèn)真閱讀完這篇文章，希望小編分享的“Pytorch中的Tensorboard與Transforms怎么搭配使用”這篇文章對(duì)大家有幫助，同時(shí)也希望大家多多支持億速云，關(guān)注億速云行業(yè)資訊頻道，更多相關(guān)知識(shí)等著你來(lái)學(xué)習(xí)!