最近在做一個(gè)項(xiàng)目,用雙通道神經(jīng)網(wǎng)絡(luò),每個(gè)通道輸入不同數(shù)據(jù)訓(xùn)練,具有相同label。開(kāi)始沒(méi)想到如何實(shí)現(xiàn),網(wǎng)上很多例子都是單通道,即便找到雙通道的例子,兩個(gè)通道的輸入也相同。

最后,終于想到了一個(gè)辦法。多輸入和單輸入其實(shí)是一樣的,只需要重新改寫(xiě)torch.utils.data.Datasets. 需要改寫(xiě)class Dataset里面的init、len和getitem
一個(gè)例子:
class MyDataset(data.Dataset):
def __init__(self, data1,data2, labels):
self.data1= data1
self.data2= data2
self.labels = labels # 我的例子中l(wèi)abel是一樣的,如果你的不同,再增加一個(gè)即可
def __getitem__(self, index):
img1,img2, target = self.data1[index], self.data2[index], self.labels[index]
return img1,img2, target
def __len__(self):
return len(self.data1) # 我的例子中l(wèi)en(self.data1) = len(self.data2)
另外有需要云服務(wù)器可以了解下創(chuàng)新互聯(lián)scvps.cn,海內(nèi)外云服務(wù)器15元起步,三天無(wú)理由+7*72小時(shí)售后在線(xiàn),公司持有idc許可證,提供“云服務(wù)器、裸金屬服務(wù)器、高防服務(wù)器、香港服務(wù)器、美國(guó)服務(wù)器、虛擬主機(jī)、免備案服務(wù)器”等云主機(jī)租用服務(wù)以及企業(yè)上云的綜合解決方案,具有“安全穩(wěn)定、簡(jiǎn)單易用、服務(wù)可用性高、性?xún)r(jià)比高”等特點(diǎn)與優(yōu)勢(shì),專(zhuān)為企業(yè)上云打造定制,能夠滿(mǎn)足用戶(hù)豐富、多元化的應(yīng)用場(chǎng)景需求。
網(wǎng)站欄目:Pytorch定義MyDatasets實(shí)現(xiàn)多通道分別輸入不同數(shù)據(jù)方式-創(chuàng)新互聯(lián)
轉(zhuǎn)載來(lái)源:http://chinadenli.net/article28/cojscp.html
成都網(wǎng)站建設(shè)公司_創(chuàng)新互聯(lián),為您提供軟件開(kāi)發(fā)、面包屑導(dǎo)航、響應(yīng)式網(wǎng)站、網(wǎng)站制作、Google、微信公眾號(hào)
聲明:本網(wǎng)站發(fā)布的內(nèi)容(圖片、視頻和文字)以用戶(hù)投稿、用戶(hù)轉(zhuǎn)載內(nèi)容為主,如果涉及侵權(quán)請(qǐng)盡快告知,我們將會(huì)在第一時(shí)間刪除。文章觀(guān)點(diǎn)不代表本網(wǎng)站立場(chǎng),如需處理請(qǐng)聯(lián)系客服。電話(huà):028-86922220;郵箱:631063699@qq.com。內(nèi)容未經(jīng)允許不得轉(zhuǎn)載,或轉(zhuǎn)載時(shí)需注明來(lái)源: 創(chuàng)新互聯(lián)
猜你還喜歡下面的內(nèi)容