溫馨提示×

溫馨提示×

您好，登錄后才能下訂單哦！

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

pytorch 彩色圖像轉(zhuǎn)灰度圖像實例

發(fā)布時間：2020-10-24 17:11:00 來源：腳本之家 閱讀：145 作者：zzw小凡 欄目：開發(fā)技術(shù)

pytorch 庫

pytorch 本身具有載入cifar10等數(shù)據(jù)集的函數(shù)，但是載入的是3*200*200的張量，當(dāng)碰到要使用灰度圖像時，可以使用他本身的函數(shù)進(jìn)行修改，以較快速的完成彩色圖像轉(zhuǎn)灰度圖像

pytorch函數(shù)

dataset = dset.CIFAR10(root='../train/data', download=True,transform=transforms.Compose([
        transforms.Scale(200),
        transforms.ToTensor(),
        transforms.Normalize((0.5, 0.5, 0.5), (0.5, 0.5, 0.5)),
       ]))

這里包含了對圖像進(jìn)行的一些處理，尺寸的控制，Normalize等

修改pytorch 庫函數(shù)

找到transforms 的庫函數(shù)位置，在后面添加 類 Gray（）

class Gray(object)

 def __call__(self, tensor):
  # TODO: make efficient
  R = tensor[0]
  G = tensor[1]
  B = tensor[2]
  tensor[0]=0.299*R+0.587*G+0.114*B
  tensor = tensor[0]
  tensor = tensor.view(1,200,200)
  return tensor

tensor.view 這里使用是為了將tensor 做成 1*200*200的大小，否則出來是200*200

以上這篇pytorch 彩色圖像轉(zhuǎn)灰度圖像實例就是小編分享給大家的全部內(nèi)容了，希望能給大家一個參考，也希望大家多多支持億速云。

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

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

AI
助
手