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