利用python生成照片墻的示例代碼
PIL(Python Image Library)是python的第三方圖像處理庫,但是由于其強大的功能與眾多的使用人數(shù),幾乎已經(jīng)被認為是python官方圖像處理庫了。其官方主頁為:PIL。
PIL歷史悠久,原來是只支持python2.x的版本的,后來出現(xiàn)了移植到python3的庫pillow,pillow號稱是friendly fork for PIL,其功能和PIL差不多,但是支持python3。本文只使用了PIL那些最常用的特性與用法,主要參考自:http://www.effbot.org/imagingbook。
Part 1:利用python生成照片墻
(1)簡要介紹思路:
(1)通過給定字符串生成一張圖片;(2)然后將該圖片的每個像素的寬擴張edge_len倍,高也擴張edge_len倍,假設(shè)edge_len=60,那么原文字圖片的每個像素就變成了60*60像素的一個圖片;(3)原文字圖片的每個像素的透明度不同,顯示文字的地方,透明度低(不透明),這2個字周邊的地方,透明度高(透明),我們根據(jù)原文字圖片每個像素的透明度,來設(shè)定放到這個像素(其實寬高已經(jīng)擴大了60倍)圖片的透明度(trans_alpha方法實現(xiàn))。
(2)導入庫文件
from PIL import Image, ImageDraw, ImageFontimport os
(3)由文字生成圖像
def gen_text_img(text, font_size=20, font_path=None): # args:文字內(nèi)容,文字字體大小,字體路徑 font = ImageFont.truetype(font_path, font_size) if font_path is not None else None (width, length) = font.getsize(text) # 獲取文字大小 text_img = Image.new(’RGBA’, (width, length)) draw = ImageDraw.Draw(text_img) # 第一個tuple表示未知(left,up),之后是文字,然后顏色,最后設(shè)置字體 draw.text((0, 0), text, fill=(0, 0, 0), font=font) text_img.save(’./temp_pic.jpg’) return text_img
(4)透明度調(diào)節(jié)
def trans_alpha(img, pixel): ’’’ 根據(jù)rgba的pixel調(diào)節(jié)img的透明度 這里傳進來的pixel是一個四元組(r,g,b,alpha) ’’’ _, _, _, alpha = img.split() alpha = alpha.point(lambda i: pixel[-1]*10) img.putalpha(alpha) #Part2有介紹 return img
(5)根據(jù)透明度參數(shù)對放大文字圖進行像素覆蓋
def picture_wall_mask(text_img, edge_len, pic_dir='./user'): # 根據(jù)文字圖gen_text_img像生成對應(yīng)的照片墻,輸入:文字圖像,各個照片邊長,照片所在路徑 new_img = Image.new(’RGBA’, (text_img.size[0] * edge_len, text_img.size[1] * edge_len)) file_list = os.listdir(pic_dir) img_index = 0 for x in range(0, text_img.size[0]): for y in range(0, text_img.size[1]): pixel = text_img.getpixel((x, y))#Part2有介紹 file_name = file_list[img_index % len(file_list)] try:img = Image.open(os.path.join(pic_dir, file_name)).convert(’RGBA’)#Part2有介紹img = img.resize((edge_len, edge_len))img = trans_alpha(img, pixel)new_img.paste(img, (x * edge_len, y * edge_len)) #指定區(qū)域替換,Part2有介紹img_index += 1 except Exception as e:print(f'open file {file_name} failed! {e}') return new_img
(6)生成照片墻
def main(text=’’, font_size = 20, edge_len = 60,pic_dir = './user', out_dir = './out/', font_path = ’./demo.ttf’): ’’’ 生成照片墻 :param text: Text of picture wall, if not defined this will generage a rectangle picture wall :param font_size: font size of a clear value :param edge_len: sub picture’s egde length ’’’ if len(text) >= 1: text_ = ’ ’.join(text)#將字符串用空格分隔開,提高展示效果 #text_ = text print(f'generate text wall for ’{text_}’ with picture path:{pic_dir}') text_img = gen_text_img(text_, font_size, font_path) # text_img.show() img_ascii = picture_wall_mask(text_img, edge_len, pic_dir) # img_ascii.show() img_ascii.save(out_dir + os.path.sep + ’_’.join(text) + ’.jpg’)
(7)函數(shù)執(zhí)行與傳參
if __name__ == ’__main__’: main(text=’python’)
文件目錄結(jié)構(gòu)如下,以供參考:out存放生成的照片墻,user存放貼上去的圖片。
完整代碼參考:git鏈接
(8)樣圖
Part 2:圖像處理過程中中學習到的幾個知識點:
(1)python PNG圖片顯示導入庫文件
僅適用于顯示png格式的圖片
import matplotlib.pyplot as plt # plt 用于顯示圖片import matplotlib.image as mpimg # mpimg 用于讀取圖片
顯示圖片
lena = mpimg.imread(’temp_pic.png’) # 讀取和代碼處于同一目錄下的 lena.png# 此時 lena 就已經(jīng)是一個 np.array 了,可以對它進行任意處理lena.shape #(512, 512, 3)plt.imshow(lena) # 顯示圖片plt.axis(’off’) # 不顯示坐標軸plt.show()
(2)PIL中圖像格式轉(zhuǎn)換img.convert()函數(shù)
在數(shù)字圖像處理中,針對不同的圖像格式有其特定的處理算法。所以,在做圖像處理之前,我們需要考慮清楚自己要基于哪種格式的圖像進行算法設(shè)計及其實現(xiàn)。本文基于這個需求,使用python中的圖像處理庫PIL來實現(xiàn)不同圖像格式的轉(zhuǎn)換。對于彩色圖像,不管其圖像格式是PNG,還是BMP,或者JPG,在PIL中,使用Image模塊的open()函數(shù)打開后,返回的圖像對象的模式都是“RGB”。而對于灰度圖像,不管其圖像格式是PNG,還是BMP,或者JPG,打開后,其模式為“L”。對于PNG、BMP和JPG彩色圖像格式之間的互相轉(zhuǎn)換都可以通過Image模塊的open()和save()函數(shù)來完成。具體說就是,在打開這些圖像時,PIL會將它們解碼為三通道的“RGB”圖像。用戶可以基于這個“RGB”圖像,對其進行處理。處理完畢,使用函數(shù)save(),可以將處理結(jié)果保存成PNG、BMP和JPG中任何格式。這樣也就完成了幾種格式之間的轉(zhuǎn)換。同理,其他格式的彩色圖像也可以通過這種方式完成轉(zhuǎn)換。當然,對于不同格式的灰度圖像,也可通過類似途徑完成,只是PIL解碼后是模式為“L”的圖像。
而對于Part 1中convert()函數(shù)的使用,推薦一篇博文Python圖像處理庫PIL中圖像格式轉(zhuǎn)換以供參考學習。
(3)img.paste()函數(shù)
第一個參數(shù)是用來覆蓋的圖片,第二個參數(shù)是覆蓋的位置,參考博文。
(4)img.getpixel()函數(shù)
查看圖像存儲值,參考博文。
(5)python圖像處理:給圖像添加透明度(alpha通道)
主要介紹img.putalpha()函數(shù)的用法,參考博文
到此這篇關(guān)于利用python生成照片墻的示例代碼的文章就介紹到這了,更多相關(guān)python生成照片墻內(nèi)容請搜索好吧啦網(wǎng)以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持好吧啦網(wǎng)!
相關(guān)文章:
