Python利用AI接口實現摳圖并改圖片底色
更新時間：2022年10月08日 10:45:55 作者：小圓-
時候我們的證件照需要換底色，又來不及去照相館拍照，用ps也不好摳圖，所以今天給你們分享一下如何用python來摳圖，并換底色，需要的可以參考一下
前言
嗨嗨，大家好呀 ，我是小圓。有時候我們的證件照需要換底色，又來不及去照相館拍照，用ps也不好摳圖，所以今天給你們分享一下如何用python來摳圖，并換底色
一、注冊百度AI賬號
創(chuàng)建人像分割應用
百度人像分割主頁：按步驟注冊，登錄，實名認證即可。
在控制臺主頁找到人體分析
創(chuàng)建應用
里面的需要填寫的內容可以隨便寫，新用戶要去領取免費資源，不然使用不了。
創(chuàng)建完成在應用列表記錄 API Key、Secret Key的值 ，稍后要用。
至此，注冊賬號和創(chuàng)建應用的任務就完成了。
二、代碼實現
1.引入庫
import os
import requests
import base64
import cv2
import numpy as np
from PIL import Image
from pathlib import Path
???????path = os.getcwd()
paths = list(Path(path).glob('*'))
2.獲取Access Token
def get_access_token():
url = 'https://aip.baidubce.com/oauth/2.0/token'
data = {
'grant_type': 'client_credentials', # 固定值
'client_id': '替換成你的API Key', # 在開放平臺注冊后所建應用的API Key
'client_secret': '替換成你的Secret Key' # 所建應用的Secret Key
}
res = requests.post(url, data=data)
res = res.json()
access_token = res['access_token']
return access_token
3.核心代碼
def removebg():
try:
request_url = "https://aip.baidubce.com/rest/2.0/image-classify/v1/body_seg"
# 二進制方式打開圖片文件
f = open(name, 'rb')
img = base64.b64encode(f.read())
params = {"image":img}
access_token = get_access_token()
request_url = request_url + "?access_token=" + access_token
headers = {'content-type': 'application/x-www-form-urlencoded'}
response = requests.post(request_url, data=params, headers=headers)
if response:
res = response.json()["foreground"]
png_name=name.split('.')[0]+".png"
with open(png_name,"wb") as f:
data = base64.b64decode(res)
f.write(data)
fullwhite(png_name) #png圖片底色填充，視情況舍去
png_jpg(png_name) #png格式轉jpg,視情況舍去
os.remove(png_name) #刪除原png圖片，視情況舍去
print(name+"\t處理成功！")
except Exception as e:
pass4.圖片底色填充
def fullwhite(png_name):
im = Image.open(png_name)
x,y = im.size
try:
p = Image.new('RGBA', im.size, (255,255,255)) # 使用白色來填充背景,視情況更改
p.paste(im, (0, 0, x, y), im)
p.save(png_name)
except:
pass
5.圖片壓縮
#compress_rate：數值越小照片越模糊
def resize(compress_rate = 0.5):
im = Image.open(name)
w, h = im.size
im_resize = im.resize((int(w*compress_rate), int(h*compress_rate)))
resize_w, resieze_h = im_resize.size
#quality 代表圖片質量，值越低越模糊
im_resize.save(name)
im.close()
6.獲取圖圖片大小
def get_size():
size = os.path.getsize(name)
return size / 1024
7.png格式轉jpg
def png_jpg(png_name):
im = Image.open(png_name)
bg=Image.new('RGB',im.size,(255,255,255))
bg.paste(im)
jpg_name = png_name.split('.')[0]+".jpg"
#quality 代表圖片質量，值越低越模糊
bg.save(jpg_name,quality=70)
im.close()
8.主函數
if __name__ == '__main__':
for i in paths:
name = os.path.basename(i.name)
if(name==os.path.basename(__file__)):
continue
size = get_size()
##照片壓縮
while size >=900:
size = get_size()
resize()
removebg()
print(" ")9.完整代碼
#人像分割
import os
import requests
import base64
import cv2
import numpy as np
from PIL import Image
from pathlib import Path
path = os.getcwd()
paths = list(Path(path).glob('*'))
def get_access_token():
url = 'https://aip.baidubce.com/oauth/2.0/token'
data = {
'grant_type': 'client_credentials', # 固定值
'client_id': '替換成你的API Key', # 在開放平臺注冊后所建應用的API Key
'client_secret': '替換成你的Secret Key' # 所建應用的Secret Key
}
res = requests.post(url, data=data)
res = res.json()
access_token = res['access_token']
return access_token
def png_jpg(png_name):
im = Image.open(png_name)
bg=Image.new('RGB',im.size,(255,255,255))
bg.paste(im)
jpg_name = png_name.split('.')[0]+".jpg"
#quality 代表圖片質量，值越低越模糊
bg.save(jpg_name,quality=70)
im.close()
#compress_rate：數值越小照片越模糊
def resize(compress_rate = 0.5):
im = Image.open(name)
w, h = im.size
im_resize = im.resize((int(w*compress_rate), int(h*compress_rate)))
resize_w, resieze_h = im_resize.size
#quality 代表圖片質量，值越低越模糊
im_resize.save(name)
im.close()
def get_size():
size = os.path.getsize(name)
return size / 1024
def fullwhite(png_name):
im = Image.open(png_name)
x,y = im.size
try:
# 使用白色來填充背景
# (alpha band as paste mask).
p = Image.new('RGBA', im.size, (255,255,255))
p.paste(im, (0, 0, x, y), im)
p.save(png_name)
except:
pass
def removebg():
try:
request_url = "https://aip.baidubce.com/rest/2.0/image-classify/v1/body_seg"
# 二進制方式打開圖片文件
f = open(name, 'rb')
img = base64.b64encode(f.read())
params = {"image":img}
access_token = get_access_token()
request_url = request_url + "?access_token=" + access_token
headers = {'content-type': 'application/x-www-form-urlencoded'}
response = requests.post(request_url, data=params, headers=headers)
if response:
res = response.json()["foreground"]
png_name=name.split('.')[0]+".png"
with open(png_name,"wb") as f:
data = base64.b64decode(res)
f.write(data)
fullwhite(png_name)
png_jpg(png_name)
os.remove(png_name)
print(name+"\t處理成功！")
except Exception as e:
pass
if __name__ == '__main__':
for i in paths:
name = os.path.basename(i.name)
if(name==os.path.basename(__file__)):
continue
size = get_size()
##照片壓縮
while size >=900:
size = get_size()
resize()
removebg()
print(" ")[重要]使用前注意事項
1. 該程序會覆蓋原文件，使用前請備份文件，以免造成數據丟失
2. 將程序復制到和待處理的照片同目錄下，雙擊程序即可運行
到此這篇關于Python利用AI接口實現摳圖并改圖片底色的文章就介紹到這了,更多相關Python 摳圖內容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關文章希望大家以后多多支持腳本之家！
相關文章
這篇文章主要介紹了python中Tkinter實現分頁標簽的示例代碼，文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值，需要的朋友們下面隨著小編來一起學習學習吧2021-04-04
這篇文章主要介紹了python解析XML文件的方法，大家參考使用吧2013-12-12
今天小編大家分享一篇PYQT5實現控制臺顯示功能的方法，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2019-06-06
python使用xlrd實現檢索excel中某列含有指定字符串記錄的方法
這篇文章主要介紹了python使用xlrd實現檢索excel中某列含有指定字符串記錄的方法,涉及Python使用xlrd模塊檢索Excel的技巧,非常具有實用價值,需要的朋友可以參考下2015-05-05
短鏈接，通俗來說，就是將長的URL網址，通過程序計算等方式，轉換為簡短的網址字符串。本文將用Python語言實現這一效果，需要的可以參考一下2022-11-11
這篇文章主要介紹了Python實現獲取磁盤剩余空間的2種方法,結合具體實例形式分析了Python操作計算機硬件的相關實現技巧,需要的朋友可以參考下2017-06-06
這篇文章主要介紹了Django中使用Json返回數據的實現方法，文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值，需要的朋友們下面隨著小編來一起學習學習吧2020-06-06
通過python操作注冊表主要有兩種方式，一種是通過python的內置模塊 _winreg，另一種方式就是 Win32 Extension For Python的win32api模塊，這里主要簡單看看用內置模塊 _winreg如何操作注冊表2020-02-02
語音識別是一門交叉學科。近二十年來，語音識別技術取得顯著進步，開始從實驗室走向市場。人們預計，未來10年內，語音識別技術將進入工業(yè)、家電、通信、汽車電子、醫(yī)療、家庭服務、消費電子產品等各個領域2021-11-11
這篇文章主要為大家介紹了Python 異步之推導式示例詳解，有需要的朋友可以借鑒參考下，希望能夠有所幫助，祝大家多多進步，早日升職加薪2023-03-03