數(shù)據(jù)格式轉(zhuǎn)化,PyTorch就是救星!
介紹
Pytorch是一個(gè)深度學(xué)習(xí)框架,廣泛用于圖像分類、分割、目標(biāo)識別等各種任務(wù)。在這種情況下,我們必須處理各種類型的數(shù)據(jù)。很可能在大多數(shù)情況下,數(shù)據(jù)可能不是我們所需要的格式。PyTorch轉(zhuǎn)換就是救星。
torchvision.transforms模塊提供了可以使用的各種圖像轉(zhuǎn)換。我們使用變換對數(shù)據(jù)進(jìn)行一些操作,使其適合于訓(xùn)練torchvision模塊,PyTorch為常見的圖像變換提供變換有關(guān)的函數(shù)。這些變換可以使用Compose鏈接在一起。
讓我們在本文中看看其中的幾個(gè)!準(zhǔn)備好了嗎?
1. ToTensor
這是一個(gè)非常常用的轉(zhuǎn)換。在PyTorch中,我們主要處理張量形式的數(shù)據(jù)。如果輸入數(shù)據(jù)是NumPy數(shù)組或PIL圖像的形式,我們可以使用ToTensor將其轉(zhuǎn)換為張量格式。
最后一個(gè)張量的形式是(C * H * W)。同時(shí),還執(zhí)行從0–255到0–1的范圍內(nèi)的縮放操作。
讓我們用一個(gè)例子來更好地理解它。在這個(gè)博客中,我將使用Ragnar(我最喜歡的虛構(gòu)角色)的圖像來執(zhí)行轉(zhuǎn)換。
2. Normalize
此操作將獲取張量圖像,并使用平均值和標(biāo)準(zhǔn)差對其進(jìn)行歸一化。它有3個(gè)參數(shù):mean, std, inplace。我們需要為3個(gè)通道提供一系列平均值,作為參數(shù)“mean”,“std”類似。如果將“inplace”設(shè)為True,則將計(jì)算得到的值覆蓋之前的值。
torchvision.transforms.Normalize([meanOfChannel1, meanOfChannel2, meanOfChannel3], [stdOfChannel1, stdOfChannel2, stdOfChannel3])
#Example:
transforms.Normalize((0.5, 0.5, 0.5), (0.5, 0.5, 0.5))
3. CenterCrop
這將在中心裁剪給定的張量圖像。你可以以(高度、寬度)的形式向transforms.CenterCrop()提供要裁剪的大小作為輸入。讓我們在圖像上實(shí)現(xiàn)這個(gè)并進(jìn)行檢查。
transform = transforms.Compose([transforms.ToTensor(),transforms.CenterCrop((200,100))])
tensor_img = transform(image)
tensor_img.shape
Output: torch.Size([3, 200, 100])
如果只提供一個(gè)尺寸標(biāo)注而不是兩個(gè)尺寸標(biāo)注,會發(fā)生什么情況?
它將假設(shè)它是一個(gè)正方形,并且將生成一個(gè)(size, size))的裁剪。
如果給定的尺寸比原來的尺寸大呢?
沿著這些邊,圖像將填充0!
4. RandomHorizontalFlip
此變換將以給定的概率水平(隨機(jī))翻轉(zhuǎn)圖像。你可以通過參數(shù)“p”來設(shè)置這個(gè)概率。p的默認(rèn)值為0.5。
檢查我下面的例子來理解。
transform = transforms.Compose([transforms.RandomHorizontalFlip(p=0.9)])
tensor_img = transform(image)
tensor_img
查看原始圖像和翻轉(zhuǎn)的圖像!
請輸入評論內(nèi)容...
請輸入評論/評論長度6~500個(gè)字
最新活動更多
-
即日-10.29立即報(bào)名>> 2024德州儀器嵌入式技術(shù)創(chuàng)新發(fā)展研討會
-
10月31日立即下載>> 【限時(shí)免費(fèi)下載】TE暖通空調(diào)系統(tǒng)高效可靠的組件解決方案
-
即日-11.13立即報(bào)名>>> 【在線會議】多物理場仿真助跑新能源汽車
-
11月14日立即報(bào)名>> 2024工程師系列—工業(yè)電子技術(shù)在線會議
-
12月19日立即報(bào)名>> 【線下會議】OFweek 2024(第九屆)物聯(lián)網(wǎng)產(chǎn)業(yè)大會
-
即日-12.26火熱報(bào)名中>> OFweek2024中國智造CIO在線峰會
推薦專題
- 1 Intel宣布40年來最重大轉(zhuǎn)型:年底前裁員15000人、拋掉2/3房產(chǎn)
- 2 因美封殺TikTok,字節(jié)股價(jià)骨折!估值僅Meta1/5
- 3 宏山激光重磅發(fā)布行業(yè)解決方案,助力智能制造產(chǎn)業(yè)新飛躍
- 4 國產(chǎn)AI芯片公司破產(chǎn)!白菜價(jià)拍賣
- 5 具身智能火了,但規(guī)模落地還需時(shí)間
- 6 三次錯(cuò)失風(fēng)口!OpenAI前員工殺回AI編程賽道,老東家捧金相助
- 7 國產(chǎn)英偉達(dá)們,抓緊沖刺A股
- 8 英偉達(dá)的麻煩在后頭?
- 9 將“網(wǎng)紅”變成“商品”,AI“爆改”實(shí)力拉滿
- 10 智慧貨架管理引領(lǐng)零售新風(fēng)尚
- 高級軟件工程師 廣東省/深圳市
- 自動化高級工程師 廣東省/深圳市
- 光器件研發(fā)工程師 福建省/福州市
- 銷售總監(jiān)(光器件) 北京市/海淀區(qū)
- 激光器高級銷售經(jīng)理 上海市/虹口區(qū)
- 光器件物理工程師 北京市/海淀區(qū)
- 激光研發(fā)工程師 北京市/昌平區(qū)
- 技術(shù)專家 廣東省/江門市
- 封裝工程師 北京市/海淀區(qū)
- 結(jié)構(gòu)工程師 廣東省/深圳市