教你用Python合成人像,足不出戶遊遍全球!

千鋒武漢發表於2021-04-07

      受到疫情反覆的影響,很多小夥伴原定的出遊計劃都被打斷了,也就拍不出美美的照片了,今天小千就來給大家分享一篇利用Python自動摳圖並自動合成全新背景照片,讓你足不出戶遊遍全球。

      功能簡介

      我們今天實現的小功能就是,利用python來提取我們的人體影像,然後將我們提取的人像跟背景向結合,最後透過flask在介面上進行顯示,其流程如下圖所示:

1

      提取人像照片

      首先是人像的提取,對於人像的提取,可以利用removebg庫,該庫需要大家去它的官網進行註冊,然後按照教程獲取API金鑰,並安裝removebg庫。

      由於該庫免費使用的次數少,而且需要花錢進行續費,因此小編決定採用百度AI提供的介面,對於大家個人使用和驗證來說,百度AI提供的免費次數足夠使用。

      對於百度API的使用,大家也已經不陌生了,大家只需要建立對應的應用,然後獲取到API的ID、KEY和SECRET_KEY即可呼叫其介面。

      程式如下圖所示:

2

      上述的程式中,我們首先讀入二進位制的圖片資訊,然後透過百度API提供的介面,上傳我們的圖片資訊,然後獲取我們需要的前景圖片資訊,將其儲存到本地資料夾下即可。

      有了我們的前景,也即人體的影像,接下來就是如何將其與背景結合,程式如下圖所示:

3

      上述的程式中,我們首先利用Image讀入背景圖片和前景圖片,然後將前景圖片縮放到合適的尺寸,然後將前景圖片進行分離,分離出前景圖片中alpha通道的資訊,alpha通道控制著圖片的透明度資訊,我們將前景貼上到背景圖片時,需要將alpha通道提取出來作為mask傳入,否則組合出來的圖片將不會有透明的效果。

      我們上面分別將處理了人像提取和前景背景圖片的組合,並將兩個函式組合在一起命名為travel_image。如下圖所示:

4

      程式中傳入的是前景圖片和背景圖片的路徑,以及組合好之後組合圖片的名字,rate指的是前景圖片的大小調整比例。函式返回的是組合好的圖片的儲存路徑。

      把程式跟Flask結合

      接下來我們要做的就是將程式與flask結合。其部分程式如下圖所示:

5

      上述程式中,我們首先獲取上傳的兩張圖片,這裡我們規定上傳的圖片是2張,分別是前景圖片和背景圖片;

      而且前景圖片和背景圖片的命名也有需求,就是背景圖片的名稱排序必須要在前景圖片名稱之前。程式判斷如果上傳的是2張圖片,便會呼叫我們的travel_image函式,返回我們的儲存路徑;

      然後將路徑資訊傳給html檔案,html檔案便會根據圖片的地址來讀取圖片,並在頁面上進行顯示;

      最後,我們來展示一下整個程式的執行過程

6


來自 “ ITPUB部落格 ” ,連結:http://blog.itpub.net/31548651/viewspace-2766812/,如需轉載,請註明出處,否則將追究法律責任。

相關文章