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)文章
Pythonr基于selenium如何實現(xiàn)不同商城的商品價格差異分析系統(tǒng)
這篇文章主要給大家介紹了關(guān)于Pythonr基于selenium如何實現(xiàn)不同商城的商品價格差異分析系統(tǒng)的相關(guān)資料,文中通過示例代碼介紹的非常詳細,對大家的學(xué)習或者工作具有一定的參考學(xué)習價值,需要的朋友可以參考下2022-03-03
python類:class創(chuàng)建、數(shù)據(jù)方法屬性及訪問控制詳解
下面小編就為大家?guī)硪黄猵ython類:class創(chuàng)建、數(shù)據(jù)方法屬性及訪問控制詳解。小編覺得挺不錯的，現(xiàn)在就分享給大家，也給大家做個參考。一起跟隨小編過來看看吧2016-07-07
這篇文章主要為大家詳細介紹了python實現(xiàn)浪漫的煙花秀，具有一定的參考價值，感興趣的小伙伴們可以參考一下2019-01-01
python網(wǎng)絡(luò)編程學(xué)習筆記(四)：域名系統(tǒng)
當我們在上網(wǎng)的時候，通常輸入的是網(wǎng)址，其實這就是一個域名，而我們計算機網(wǎng)絡(luò)上的計算機彼此之間只能用IP地址才能相互識別2014-06-06
這篇文章主要介紹了Python?字典(Dictionary)詳細，字典是另一種可變?nèi)萜髂Ｐ?，且可存儲任意類型對象。下面和小編一起進入文章學(xué)習新內(nèi)容吧,需要的朋友可以參考一下2022-02-02
這篇文章主要介紹了Django之無名分組和有名分組,小編覺得挺不錯的，現(xiàn)在分享給大家，也給大家做個參考。一起跟隨小編過來看看吧2019-04-04
我們此前花了很大功夫?qū)懥艘粋€極其簡單的web框架myWeb,想要給別人用的時候，需要讓別人拷貝源代碼才行，這太low了，所以本篇文章會介紹如何將自己寫的模塊打包上傳至pypi，以便讓需要的人通過pip進行安裝，感興趣的同學(xué)可以參考閱讀2023-05-05
這篇文章主要介紹了python Polars庫的使用簡介，幫助大家更好的理解和學(xué)習使用python，感興趣的朋友可以了解下2021-04-04
這篇文章主要是想和大家一起來討論下去Python中的命名空間、作用域和LEGB規(guī)則等不同概念,文中的示例代碼簡潔易懂,快跟隨小編一起來學(xué)習一下吧2023-08-08
Python通過字典映射函數(shù)實現(xiàn)switch
這篇文章主要介紹了Python通過字典映射函數(shù)實現(xiàn)switch,文中通過示例代碼介紹的非常詳細，對大家的學(xué)習或者工作具有一定的參考學(xué)習價值,需要的朋友可以參考下2020-11-11