欧美bbbwbbbw肥妇,免费乱码人妻系列日韩,一级黄片

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

 更新時(shí)間：2020年01月13日 08:36:44   作者：zzw小凡  
今天小編就為大家分享一篇pytorch 彩色圖像轉(zhuǎn)灰度圖像實(shí)例，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧

pytorch 庫(kù)

pytorch 本身具有載入cifar10等數(shù)據(jù)集的函數(shù)，但是載入的是3*200*200的張量，當(dāng)碰到要使用灰度圖像時(shí)，可以使用他本身的函數(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)),
       ]))

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

修改pytorch 庫(kù)函數(shù)

找到transforms 的庫(kù)函數(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的大小，否則出來(lái)是200*200

以上這篇pytorch 彩色圖像轉(zhuǎn)灰度圖像實(shí)例就是小編分享給大家的全部?jī)?nèi)容了，希望能給大家一個(gè)參考，也希望大家多多支持腳本之家。

相關(guān)文章

最新評(píng)論