溫馨提示×

您好,登錄后才能下訂單哦!

密碼登錄×
登錄注冊(cè)×
其他方式登錄
點(diǎn)擊 登錄注冊(cè) 即表示同意《億速云用戶服務(wù)條款》

pytorch讀取圖像數(shù)據(jù)的方法

發(fā)布時(shí)間:2021-07-27 15:48:47 來源:億速云 閱讀:322 作者:chen 欄目:大數(shù)據(jù)

本篇內(nèi)容主要講解“pytorch讀取圖像數(shù)據(jù)的方法”,感興趣的朋友不妨來看看。本文介紹的方法操作簡(jiǎn)單快捷,實(shí)用性強(qiáng)。下面就讓小編來帶大家學(xué)習(xí)“pytorch讀取圖像數(shù)據(jù)的方法”吧!

pytoech讀取圖像數(shù)據(jù)的方法和步驟是固定的,一般是第一步先設(shè)置預(yù)處理方法,第二步是設(shè)置數(shù)據(jù)類,也就是確定要讀取的數(shù)據(jù)是什么,第三步是讀取數(shù)據(jù),就是按照設(shè)置的要求把數(shù)據(jù)讀取出來。

舉個(gè)常見例子:

# 第一步:預(yù)處理
transform = transforms.Compose([transforms.ToTensor(), transforms.Normalize(mean=[0.5], std=[0.5])])

# 第二步:設(shè)置數(shù)據(jù)類
train_dataset = datasets.MNIST(root='D:\code\mnist', train=True, download=False, transform=transform)
test_dataset = datasets.MNIST(root='D:\code\mnist', train=False, download=False, transform=transform)

# 第三步:讀取數(shù)據(jù)
train_loader = DataLoader(train_dataset, shuffle=True, batch_size=batch_size)
test_loader = DataLoader(test_dataset, shuffle=False, batch_size=batch_size)

到此,相信大家對(duì)“pytorch讀取圖像數(shù)據(jù)的方法”有了更深的了解,不妨來實(shí)際操作一番吧!這里是億速云網(wǎng)站,更多相關(guān)內(nèi)容可以進(jìn)入相關(guān)頻道進(jìn)行查詢,關(guān)注我們,繼續(xù)學(xué)習(xí)!

向AI問一下細(xì)節(jié)

免責(zé)聲明:本站發(fā)布的內(nèi)容(圖片、視頻和文字)以原創(chuàng)、轉(zhuǎn)載和分享為主,文章觀點(diǎn)不代表本網(wǎng)站立場(chǎng),如果涉及侵權(quán)請(qǐng)聯(lián)系站長(zhǎng)郵箱:is@yisu.com進(jìn)行舉報(bào),并提供相關(guān)證據(jù),一經(jīng)查實(shí),將立刻刪除涉嫌侵權(quán)內(nèi)容。

AI