溫馨提示×
您好，登錄后才能下訂單哦！
點擊 登錄注冊 即表示同意《億速云用戶服務條款》
您好，登錄后才能下訂單哦！
最近在做一個項目，用雙通道神經網絡，每個通道輸入不同數據訓練，具有相同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 # 我的例子中label是一樣的，如果你的不同，再增加一個即可
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) # 我的例子中len(self.data1) = len(self.data2)
以上這篇Pytorch 定義MyDatasets實現多通道分別輸入不同數據方式就是小編分享給大家的全部內容了，希望能給大家一個參考，也希望大家多多支持億速云。
免責聲明：本站發布的內容（圖片、視頻和文字）以原創、轉載和分享為主，文章觀點不代表本網站立場，如果涉及侵權請聯系站長郵箱：is@yisu.com進行舉報，并提供相關證據，一經查實，將立刻刪除涉嫌侵權內容。
