受到疫情反復(fù)的影響,很多小伙伴原定的出游計劃都被打斷了,也就拍不出美美的照片了,今天小千就來給大家分享一篇利用Python自動摳圖并自動合成全新背景照片,讓你足不出戶游遍全球。
功能簡介
我們今天實現(xiàn)的小功能就是,利用python來提取我們的人體圖像,然后將我們提取的人像跟背景向結(jié)合,最后通過flask在界面上進行顯示,其流程如下圖所示:
提取人像照片
首先是人像的提取,對于人像的提取,可以利用removebg庫,該庫需要大家去它的官網(wǎng)進行注冊,然后按照教程獲取API密鑰,并安裝removebg庫。
由于該庫免費使用的次數(shù)少,而且需要花錢進行續(xù)費,因此小編決定采用百度AI提供的接口,對于大家個人使用和驗證來說,百度AI提供的免費次數(shù)足夠使用。
對于百度API的使用,大家也已經(jīng)不陌生了,大家只需要創(chuàng)建對應(yīng)的應(yīng)用,然后獲取到API的ID、KEY和SECRET_KEY即可調(diào)用其接口。
程序如下圖所示:
上述的程序中,我們首先讀入二進制的圖片信息,然后通過百度API提供的接口,上傳我們的圖片信息,然后獲取我們需要的前景圖片信息,將其保存到本地文件夾下即可。
有了我們的前景,也即人體的圖像,接下來就是如何將其與背景結(jié)合,程序如下圖所示:
上述的程序中,我們首先利用Image讀入背景圖片和前景圖片,然后將前景圖片縮放到合適的尺寸,然后將前景圖片進行分離,分離出前景圖片中alpha通道的信息,alpha通道控制著圖片的透明度信息,我們將前景粘貼到背景圖片時,需要將alpha通道提取出來作為mask傳入,否則組合出來的圖片將不會有透明的效果。
我們上面分別將處理了人像提取和前景背景圖片的組合,并將兩個函數(shù)組合在一起命名為travel_image。如下圖所示:
程序中傳入的是前景圖片和背景圖片的路徑,以及組合好之后組合圖片的名字,rate指的是前景圖片的大小調(diào)整比例。函數(shù)返回的是組合好的圖片的保存路徑。
把程序跟Flask結(jié)合
接下來我們要做的就是將程序與flask結(jié)合。其部分程序如下圖所示:
上述程序中,我們首先獲取上傳的兩張圖片,這里我們規(guī)定上傳的圖片是2張,分別是前景圖片和背景圖片;
而且前景圖片和背景圖片的命名也有需求,就是背景圖片的名稱排序必須要在前景圖片名稱之前。程序判斷如果上傳的是2張圖片,便會調(diào)用我們的travel_image函數(shù),返回我們的保存路徑;
然后將路徑信息傳給html文件,html文件便會根據(jù)圖片的地址來讀取圖片,并在頁面上進行顯示;
最后,我們來展示一下整個程序的執(zhí)行過程
學(xué)習(xí)Python,可以參考千鋒Python培訓(xùn)班推出的Python開發(fā)學(xué)習(xí)路線,結(jié)合千鋒Python培訓(xùn)機構(gòu)名師精心錄制的全套Python視頻教程,可以讓你對學(xué)習(xí)Python需要掌握的知識有個清晰的了解,并快速入門Python開發(fā)。千鋒Python培訓(xùn)機構(gòu)視頻教程分為三個大塊:Python基礎(chǔ)視頻教程、Python高級視頻教程、Python高手晉級視頻教程。視頻講解通俗易懂,入門Python開發(fā)僅用此套視頻足矣。想要獲取免費Python學(xué)習(xí)路線和學(xué)習(xí)資料可以添加我們的Python技術(shù)交流qq群:790693323 加群找群管理領(lǐng)取即可,Python相關(guān)技術(shù)問題也可以加群解決,等你來哦~~~~