我是張量流和 keras 的新手，我正在嘗試為街景門牌號碼 (SVHN) 數(shù)據(jù)集創(chuàng)建 CNN 模型。數(shù)據(jù)集包含彩色圖像，我想將它們轉(zhuǎn)換為灰度圖像。我在網(wǎng)上發(fā)現(xiàn)了一些代碼，聲稱他們正在將圖像轉(zhuǎn)換為灰度，但它只是改變顏色。人們正在閱讀帶有灰色色彩圖的第二張圖像。有什么辦法可以真正將這張圖像變成灰度圖嗎？（我不知道如何用這種編程語言處理圖像。如果這是一個愚蠢的問題，請原諒我并提供一個簡短的解釋。）我在下面提供了圖像和代碼，我將不勝感激。代碼：import matplotlibimport matplotlib.pyplot as pltimport numpy as np#Read picture:picture = plt.imread('google.jpg')print("google logo's shape is: ",picture.shape) #(500, 500, 3)#saving picture as an np array:pic_array = np.array(picture)#Turning image to grayscalegrayscale_pic = np.expand_dims(np.dot(pic_array[...,:3],[0.299, 0.587, 0.144]),axis = 0)#Dimensions shifted, (probly my mistake):grayscale_pic = np.moveaxis(grayscale_pic, 0, -1)print("shape of grayscale pic = ", grayscale_pic.shape) # (500, 500, 1)plt.imshow(picture) #Figure_1plt.show()plt.imshow(grayscale_pic) #Figure_2plt.show()
3 回答
海綿寶寶撒
TA貢獻1809條經(jīng)驗 獲得超8個贊
有多種方法可以做到這一點。一種可能的方法是利用PIL(Pillow)庫：
from PIL import Image
import matplotlib.pyplot as plt
picture = Image.open('google.jpg')
grayscale_pic = picture.convert('LA')
grayscale_pic.save('grayscale.png')
fig,ax = plt.subplots(nrows=1, ncols=2)
plt.subplot(1,2,1)
plt.imshow(picture)
plt.subplot(1,2,2)
plt.imshow(grayscale_pic)
plt.show()
輸出：
慕田峪7331174
TA貢獻1828條經(jīng)驗 獲得超13個贊
opencv您可以使用以下命令將普通圖像轉(zhuǎn)換為灰度圖像：
import cv2 gray = cv2.cvtColor(picture,cv2.COLOR_RGB2GRAY)
如果你numpy更喜歡opencv，那么你可以使用這個：
gray = np.dot(picture[...,:3], [0.2989, 0.5870, 0.1140])
慕桂英3389331
TA貢獻2036條經(jīng)驗 獲得超8個贊
您可以與權(quán)重一起使用matplotlib：
import numpy as np
import matplotlib.pyplot as plt
an_image = plt.imread('google.png')
rgb_weights = [0.2989, 0.5870, 0.1140]
grayscale_image = np.dot(an_image[..., :3], rgb_weights)
plt.axis('off')
plt.imshow(grayscale_image, cmap=plt.get_cmap("gray"), aspect='auto')
plt.show()
輸出：
如果刪除aspect='auto'參數(shù)：
或者你可以使用opencv
import cv2
an_image = cv2.imread("google.png")
grey_image = cv2.cvtColor(an_image, cv2.COLOR_BGR2GRAY)
或者你可以使用PIL圖書館
from PIL import Image
img = Image.open('google.png').convert('LA')
LA模式為 L（8 位像素，黑白），帶有 ALPHA 設(shè)計用于 .gif 和 .png。如果您的圖像是 .jpeg，請使用L.
輸出：
添加回答
舉報
0/150
提交
取消