Python——Removebg|證件照換底

第一步：安裝removebg庫

第二步 申請API接口

這裏需要用自己郵箱註冊，並登陸
https://www.remove.bg/

第三步，代碼實現

from removebg import RemoveBg#導入removebg庫
import os
rmbg = RemoveBg("VNzVdb7uBAgryt7xkUcDDRXT", "error.log")#替換成自己的API接口
rmbg.remove_background_from_img_file("/Users/鄭玉婷/Pictures/Camera Roll/1.jpg")#圖片地址，摳完的也會自動保存進去

批量摳圖

from removebg import RemoveBg
import os
rmbg = RemoveBg("VNzVdb7uBAgryt7xkUcDDRXT", "error.log")
path = '%s/picture' % os.getcwd()#在源碼同級的目錄下新建一個圖片文件夾放圖片
for pic in os.listdir(path):
    rmbg.remove_background_from_img_file("%s/%s" % (path, pic))

效果圖

單張

批量

總結

Removebg模塊的效果還是不錯的，用證件照會更好一些，不過由於素材緊缺以及其他原因所以用了別的一些圖片，輪廓清晰的效果更好。