快捷導(dǎo)航

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

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

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

pytorch 庫

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 庫函數(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)灰度圖像實(shí)例就是小編分享給大家的全部內(nèi)容了，希望能給大家一個(gè)參考，也希望大家多多支持腳本之家。

您可能感興趣的文章:

相關(guān)文章

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