初識(shí)Pytorch使用transforms的代碼
首先,這次講解的tansforms
功能,通俗地講,類似于在計(jì)算機(jī)視覺(jué)流程里的圖像預(yù)處理部分的數(shù)據(jù)增強(qiáng)。
transforms
的原理:
說(shuō)明:圖片(輸入)通過(guò)工具得到結(jié)果(輸出),這個(gè)工具,就是transforms模板工具,(tool=transforms.ToTensor()具體工具),使用工具result=tool(圖片)
tansforms
的調(diào)用與使用,由下圖可得: 先創(chuàng)建一個(gè)transforms.Tensor()
,使用from torchvision import transforms
調(diào)包transforms
去調(diào)init
函數(shù)init
去調(diào)用真正的transforms
類,里面就有很多的方法(綠色五角星標(biāo)注),例如:resize,ToTensor,CenterCrop
(從這些方法可以看出,許多都是數(shù)據(jù)增強(qiáng)的方法)。
3. 接下來(lái),上代碼:
import os from torchvision import transforms from PIL import Image root_path = "D:\\data\\basic\\Image" label_path = "aligned" # 1.獲取aligned第一張圖的名字 img_dir = os.path.join(root_path, label_path) img_list = os.listdir(img_dir) img_path = img_list[0] # 2.獲取aligned第一張圖的路徑 img = os.path.join(root_path, label_path, img_path) # 3.使用python自帶的PIL獲取圖片 img = Image.open(img) # 4.將PIL利用transforms轉(zhuǎn)換成ToTensor to_tensor = transforms.ToTensor() # 創(chuàng)建totensor () img = to_tensor(img) # 使用to_tensor直接將圖片的PIL轉(zhuǎn)化為tensor print(img) # transforms
代碼結(jié)果:
到此這篇關(guān)于初識(shí)Pytorch使用transforms的文章就介紹到這了,更多相關(guān)Pytorch使用transforms內(nèi)容請(qǐng)搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家!
相關(guān)文章
conda管理Python虛擬環(huán)境的實(shí)現(xiàn)
本文主要介紹了conda管理Python虛擬環(huán)境的實(shí)現(xiàn),主要包括使用conda工具創(chuàng)建、查看和刪除Python虛擬環(huán)境,具有一定的參考價(jià)值,感興趣的可以了解一下2024-01-01django基于存儲(chǔ)在前端的token用戶認(rèn)證解析
這篇文章主要介紹了django基于存儲(chǔ)在前端的token用戶認(rèn)證解析,文中通過(guò)示例代碼介紹的非常詳細(xì),對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值,需要的朋友可以參考下2019-08-08詳解用Python把PDF轉(zhuǎn)為Word方法總結(jié)
這篇文章主要介紹了詳解用Python把PDF轉(zhuǎn)為Word方法總結(jié),文中通過(guò)示例代碼介紹的非常詳細(xì),對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值,需要的朋友們下面隨著小編來(lái)一起學(xué)習(xí)學(xué)習(xí)吧2021-04-04python中讀入二維csv格式的表格方法詳解(以元組/列表形式表示)
這篇文章主要介紹了python中如何讀入二維csv格式的表格(以元組/列表形式表示),本文通過(guò)兩種方法給大家詳細(xì)介紹,通過(guò)實(shí)例代碼給大家介紹的非常詳細(xì),對(duì)大家的學(xué)習(xí)或工作具有一定的參考借鑒價(jià)值,需要的朋友可以參考下2020-04-04python SVD壓縮圖像的實(shí)現(xiàn)代碼
這篇文章主要介紹了python SVD壓縮圖像的實(shí)現(xiàn)代碼,文中通過(guò)示例代碼介紹的非常詳細(xì),對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值,需要的朋友們下面隨著小編來(lái)一起學(xué)習(xí)學(xué)習(xí)吧2019-11-11python中DataFrame常用的描述性統(tǒng)計(jì)分析方法詳解
這篇文章主要介紹了python中DataFrame常用的描述性統(tǒng)計(jì)分析方法詳解,描述性統(tǒng)計(jì)分析是通過(guò)圖表或數(shù)學(xué)方法,對(duì)數(shù)據(jù)資料進(jìn)行整理、分析,并對(duì)數(shù)據(jù)的分布狀態(tài)、數(shù)字特征和隨機(jī)變量之間的關(guān)系進(jìn)行估計(jì)和描述的方法,需要的朋友可以參考下2023-07-07