PyTorch 解決Dataset和Dataloader遇到的問題
今天在使用PyTorch中Dataset遇到了一個(gè)問題。先看代碼
class psDataset(Dataset):
def __init__(self, x, y, transforms = None):
super(Dataset, self).__init__()
self.x = x
self.y = y
if transforms == None:
self.transforms = Compose([Resize((224, 224)), ToTensor()])
else:
self.transforms = transforms
def __len__(self):
return len(self.x)
def __getitem__(self, idx):
img = Image.open(self.x[idx])
img = self.transforms(img)
return img, torch.tensor([[self.y[idx]]])
結(jié)果運(yùn)行時(shí)報(bào)錯(cuò)：RuntimeError: invalid argument 0: Sizes of tensors must match except in dimension 0. Got 3 and 1 in dimension 1 at /opt/conda/conda-bld/pytorch_1522182087074/work/torch/lib/TH/generic/THTensorMath.c:2897
Google了一下發(fā)現(xiàn)是這樣的：讀入的圖片有些是灰度圖（1個(gè)通道），絕大多數(shù)是RGB圖片（3通道），也有些是帶透明度的（4通道）
。這導(dǎo)致在讀入后最后一個(gè)維度（通道數(shù)）不一致（可能是1、3或者4）。
Dataloader在制作batch data時(shí)，tensor的shape必須一樣，就報(bào)了這個(gè)錯(cuò)誤。解決的方法是：img = img.convert(“RGB”)。完
整代碼如下：
class psDataset(Dataset):
def __init__(self, x, y, transforms = None):
super(Dataset, self).__init__()
self.x = x
self.y = y
if transforms == None:
self.transforms = Compose([Resize((224, 224)), ToTensor()])
else:
self.transforms = transforms
def __len__(self):
return len(self.x)
def __getitem__(self, idx):
img = Image.open(self.x[idx])
img = img.convert("RGB")
img = self.transforms(img)
return img, torch.tensor([[self.y[idx]]])
以上這篇PyTorch 解決Dataset和Dataloader遇到的問題就是小編分享給大家的全部?jī)?nèi)容了，希望能給大家一個(gè)參考，也希望大家多多支持腳本之家。
相關(guān)文章
Selenium+BeautifulSoup+json獲取Script標(biāo)簽內(nèi)的json數(shù)據(jù)
這篇文章主要介紹了Selenium+BeautifulSoup+json獲取Script標(biāo)簽內(nèi)的json數(shù)據(jù)，文中通過示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值，需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧2020-12-12
這篇文章主要介紹了如何手動(dòng)安裝python3.6,本文給大家?guī)砹税惭b步驟，非常不錯(cuò)，具有一定的參考借鑒價(jià)值，需要的朋友可以參考下2020-01-01
反射就是通過字符串的形式，導(dǎo)入模塊；通過字符串的形式，去模塊尋找指定函數(shù)，并執(zhí)行。利用字符串的形式去對(duì)象（模塊）中操作（查找/獲取/刪除/添加）成員，一種基于字符串的事件驅(qū)動(dòng)2022-09-09
Python 抓取微信公眾號(hào)賬號(hào)信息的方法
搜狗微信搜索提供兩種類型的關(guān)鍵詞搜索，一種是搜索公眾號(hào)文章內(nèi)容，另一種是直接搜索微信公眾號(hào)。這篇文章主要介紹了Python 抓取微信公眾號(hào)賬號(hào)信息,需要的朋友可以參考下2019-06-06
用pyqt5 給按鈕設(shè)置圖標(biāo)和css樣式的方法
今天小編就為大家分享一篇用pyqt5 給按鈕設(shè)置圖標(biāo)和css樣式的方法，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過來看看吧2019-06-06
python利用socketserver實(shí)現(xiàn)并發(fā)套接字功能
這篇文章主要為大家詳細(xì)介紹了python利用socketserver實(shí)現(xiàn)并發(fā)套接字功能，文中示例代碼介紹的非常詳細(xì)，具有一定的參考價(jià)值，感興趣的小伙伴們可以參考一下2018-01-01
Python3中的map函數(shù)調(diào)用后內(nèi)存釋放問題
這篇文章主要介紹了Python3中的map函數(shù)調(diào)用后內(nèi)存釋放問題,具有很好的參考價(jià)值,希望對(duì)大家有所幫助,如有錯(cuò)誤或未考慮完全的地方,望不吝賜教2024-02-02
Python實(shí)現(xiàn)一個(gè)優(yōu)先級(jí)隊(duì)列的方法
這篇文章主要介紹了Python實(shí)現(xiàn)一個(gè)優(yōu)先級(jí)隊(duì)列的方法，文中講解非常細(xì)致，代碼幫助大家更好的理解和學(xué)習(xí)，感興趣的朋友可以了解下2020-07-07
Jmeter并發(fā)執(zhí)行Python 腳本的完整流程
這篇文章主要介紹了Jmeter并發(fā)執(zhí)行 Python 腳本的問題詳解,本文給大家介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或工作具有一定的參考借鑒價(jià)值，需要的朋友可以參考下2021-09-09
利用python實(shí)現(xiàn)在微信群刷屏的方法
今天小編就為大家分享一篇利用python實(shí)現(xiàn)在微信群刷屏的方法，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過來看看吧2019-02-02