Pytorch 定義MyDatasets實現多通道分別輸入不同數據方式
更新時間：2020年01月15日 09:59:40 作者：Devin-LQP
今天小編就為大家分享一篇Pytorch 定義MyDatasets實現多通道分別輸入不同數據方式，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧
最近在做一個項目，用雙通道神經網絡，每個通道輸入不同數據訓練，具有相同label。開始沒想到如何實現，網上很多例子都是單通道，即便找到雙通道的例子，兩個通道的輸入也相同。
最后，終于想到了一個辦法。多輸入和單輸入其實是一樣的，只需要重新改寫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實現多通道分別輸入不同數據方式就是小編分享給大家的全部內容了，希望能給大家一個參考，也希望大家多多支持腳本之家。
相關文章
Pythonr基于selenium如何實現不同商城的商品價格差異分析系統(tǒng)
這篇文章主要給大家介紹了關于Pythonr基于selenium如何實現不同商城的商品價格差異分析系統(tǒng)的相關資料,文中通過示例代碼介紹的非常詳細,對大家的學習或者工作具有一定的參考學習價值,需要的朋友可以參考下2022-03-03
python類:class創(chuàng)建、數據方法屬性及訪問控制詳解
下面小編就為大家?guī)硪黄猵ython類:class創(chuàng)建、數據方法屬性及訪問控制詳解。小編覺得挺不錯的，現在就分享給大家，也給大家做個參考。一起跟隨小編過來看看吧2016-07-07
這篇文章主要為大家詳細介紹了python實現浪漫的煙花秀，具有一定的參考價值，感興趣的小伙伴們可以參考一下2019-01-01
當我們在上網的時候，通常輸入的是網址，其實這就是一個域名，而我們計算機網絡上的計算機彼此之間只能用IP地址才能相互識別2014-06-06
這篇文章主要介紹了Python?字典(Dictionary)詳細，字典是另一種可變容器模型，且可存儲任意類型對象。下面和小編一起進入文章學習新內容吧,需要的朋友可以參考一下2022-02-02
這篇文章主要介紹了Django之無名分組和有名分組,小編覺得挺不錯的，現在分享給大家，也給大家做個參考。一起跟隨小編過來看看吧2019-04-04
我們此前花了很大功夫寫了一個極其簡單的web框架myWeb,想要給別人用的時候，需要讓別人拷貝源代碼才行，這太low了，所以本篇文章會介紹如何將自己寫的模塊打包上傳至pypi，以便讓需要的人通過pip進行安裝，感興趣的同學可以參考閱讀2023-05-05
這篇文章主要介紹了python Polars庫的使用簡介，幫助大家更好的理解和學習使用python，感興趣的朋友可以了解下2021-04-04
這篇文章主要是想和大家一起來討論下去Python中的命名空間、作用域和LEGB規(guī)則等不同概念,文中的示例代碼簡潔易懂,快跟隨小編一起來學習一下吧2023-08-08
這篇文章主要介紹了Python通過字典映射函數實現switch,文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值,需要的朋友可以參考下2020-11-11