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)

# 創(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é)果:

在命令行打開tensorboard：

但是在打開tensorboard之前，要打開正確的位置，不然會報錯，以此程序為例，需要在找到logs的目錄。

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

到此這篇關(guān)于Pytorch中的Tensorboard與Transforms搭配使用的文章就介紹到這了,更多相關(guān)Tensorboard與Transforms搭配使用內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！

您可能感興趣的文章:

相關(guān)文章

Python利用pip安裝tar.gz格式的離線資源包
這篇文章主要給大家介紹了關(guān)于Python利用pip安裝tar.gz格式的離線資源包的相關(guān)資料，文中通過示例代碼介紹的非常詳細，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值，需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧
2020-09-09
詳談python http長連接客戶端
下面小編就為大家?guī)硪黄斦刾ython http長連接客戶端。小編覺得挺不錯的，現(xiàn)在就分享給大家，也給大家做個參考。一起跟隨小編過來看看吧
2017-06-06
python實現(xiàn)二維碼掃碼自動登錄淘寶
最近做項目，需要用到自動登錄淘寶，正好在學(xué)習(xí)python,整網(wǎng)絡(luò)爬蟲,所以就嘗試著寫一個腳本,自動解決。有相同需求的小伙伴可以參考下
2016-12-12
Python3實現(xiàn)獲取圖片文字里中文的方法分析
這篇文章主要介紹了Python3實現(xiàn)獲取圖片文字里中文的方法,結(jié)合實例形式分析了Python基于文字識別引擎tesseract-ocr針對圖片里中文識別的相關(guān)操作技巧與操作注意事項,需要的朋友可以參考下
2018-12-12
這篇文章主要介紹了pytest自動化測試中fixture的作用域、實例化順序及可用性的詳解示例有需要的朋友可以借鑒參考下，希望能夠有所幫助
2021-10-10