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

更新時間：2020年01月13日 08:36:44 作者：zzw小凡

今天小編就為大家分享一篇pytorch 彩色圖像轉(zhuǎn)灰度圖像實例，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧

pytorch 庫

pytorch 本身具有載入cifar10等數(shù)據(jù)集的函數(shù)，但是載入的是3*200*200的張量，當碰到要使用灰度圖像時，可以使用他本身的函數(shù)進行修改，以較快速的完成彩色圖像轉(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)),
       ]))

這里包含了對圖像進行的一些處理，尺寸的控制，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)容了，希望能給大家一個參考，也希望大家多多支持腳本之家。

您可能感興趣的文章:

Python opencv醫(yī)學處理的實現(xiàn)過程
這篇文章主要介紹了Python opencv醫(yī)學處理的實現(xiàn)過程,本文給大家介紹的非常詳細，對大家的學習或工作具有一定的參考借鑒價值，需要的朋友可以參考下
2021-05-05
Python實現(xiàn)讀取excel中的圖片功能
這篇文章主要介紹了如何利用Python實現(xiàn)讀取Excel中的圖片的功能，文中的實現(xiàn)步驟講解詳細，對我們學習Python有一定幫助，需要的可以參考一下
2022-01-01
裝了python再裝anaconda該怎么操作保姆級教程
本文詳細介紹了在已安裝Python的情況下如何正確安裝和配置Anaconda環(huán)境,包括Python環(huán)境變量的配置、Anaconda的下載安裝以及環(huán)境變量配置,文中通過代碼就介紹的非常詳細,需要的朋友可以參考下
2024-11-11
Python實現(xiàn)郵件的批量發(fā)送的示例代碼
下面小編就為大家分享一篇Python實現(xiàn)郵件的批量發(fā)送的示例代碼，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧
2018-01-01
python輸出結果刷新及進度條的實現(xiàn)操作
這篇文章主要介紹了python輸出結果刷新及進度條的實現(xiàn)操作，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧
2020-07-07
詳解Python如何獲取和理用戶的連續(xù)輸入
這篇文章主要為大家詳細介紹了如何在?Python?中獲取和處理用戶的連續(xù)輸入,包括基本的輸入方法、輸入驗證、異常處理以及一些常見的應用場景,感興趣的可以了解下
2024-03-03
Python+Tkinter制作在線個性簽名工具
這篇文章主要為大家分享如何利用Python中的Tkinter庫制作一個簡易的在線個性簽名生成工具，文中的示例代碼講解詳細，感興趣的小伙伴可以了解一下
2022-03-03
python 引用傳遞和值傳遞詳解(實參,形參)
這篇文章主要介紹了python 引用傳遞和值傳遞詳解(實參,形參)。具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧
2020-06-06
pytorch查看模型weight與grad方式
這篇文章主要介紹了pytorch查看模型weight與grad方式，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧
2020-06-06
Python結合Window計劃任務監(jiān)測郵件的示例代碼
這篇文章主要介紹了Python結合Window計劃任務監(jiān)測郵件的示例代碼，文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值，需要的朋友們下面隨著小編來一起學習學習吧
2020-08-08