Pytorch 定義MyDatasets實現(xiàn)多通道分別輸入不同數(shù)據(jù)方式
更新時間：2020年01月15日 09:59:40 作者：Devin-LQP
今天小編就為大家分享一篇Pytorch 定義MyDatasets實現(xiàn)多通道分別輸入不同數(shù)據(jù)方式，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧
最近在做一個項目，用雙通道神經(jīng)網(wǎng)絡(luò)，每個通道輸入不同數(shù)據(jù)訓(xùn)練，具有相同label。開始沒想到如何實現(xiàn)，網(wǎng)上很多例子都是單通道，即便找到雙通道的例子，兩個通道的輸入也相同。
最后，終于想到了一個辦法。多輸入和單輸入其實是一樣的，只需要重新改寫torch.utils.data.Datasets. 需要改寫class Dataset里面的init、len和getitem
一個例子：
class MyDataset(data.Dataset):
def __init__(self, data1,data2, labels):
self.data1= data1
self.data2= data2
self.labels = labels # 我的例子中l(wèi)abel是一樣的，如果你的不同，再增加一個即可
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)
以上這篇Pytorch 定義MyDatasets實現(xiàn)多通道分別輸入不同數(shù)據(jù)方式就是小編分享給大家的全部內(nèi)容了，希望能給大家一個參考，也希望大家多多支持腳本之家。
相關(guān)文章
這篇文章主要為大家詳細介紹了如何基于Python開發(fā)一個文件快速搜索工具,可以實現(xiàn)多條件文件搜索并實時搜索狀態(tài)反饋,需要的可以參考一下2025-03-03
基于keras 模型、結(jié)構(gòu)、權(quán)重保存的實現(xiàn)
今天小編就為大家分享一篇基于keras 模型、結(jié)構(gòu)、權(quán)重保存的實現(xiàn)，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2020-01-01
本文主要介紹了OpenCV仿射變換的示例代碼，文中通過示例代碼介紹的非常詳細，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值，需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧2022-08-08
Pytorch中的backward()多個loss函數(shù)用法
這篇文章主要介紹了Pytorch中的backward()多個loss函數(shù)用法，具有很好的參考價值，希望對大家有所幫助。如有錯誤或未考慮完全的地方，望不吝賜教2021-05-05
python 標(biāo)準(zhǔn)庫原理與用法詳解之os.path篇
os.path模塊主要用于文件的屬性獲取，在編程中經(jīng)常用到，本文將帶你熟悉這個模塊并掌握它的用法，感興趣的朋友跟小編來看看吧2021-10-10
Python實現(xiàn)讀取Linux系統(tǒng)的CPU以及內(nèi)存占用
這篇文章主要為大家詳細介紹了如何利用Python語言實現(xiàn)Linux系統(tǒng)的CPU以及內(nèi)存占用，文中的示例代碼講解詳細，具有一定的學(xué)習(xí)價值，需要的可以收藏一下2023-05-05
使用pyecharts生成Echarts網(wǎng)頁的實例
今天小編就為大家分享一篇使用pyecharts生成Echarts網(wǎng)頁的實例，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2019-08-08
這篇文章主要為大家詳細介紹了python多進程下實現(xiàn)日志記錄按時間分割，具有一定的參考價值，感興趣的小伙伴們可以參考一下2019-07-07
python利用re,bs4,requests模塊獲取股票數(shù)據(jù)
這篇文章主要介紹了python利用re,bs4,requests模塊獲取股票數(shù)據(jù),文中通過示例代碼介紹的非常詳細，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值,需要的朋友可以參考下2019-07-07
這篇文章主要介紹了關(guān)于Python中的空值問題及解決方案,具有很好的參考價值,希望對大家有所幫助,如有錯誤或未考慮完全的地方,望不吝賜教2023-11-11