Python圖片批量自動摳圖去背景的代碼詳解
今天發(fā)現(xiàn)個好東西啊，叫片刻摳圖，是一個在線對圖片自動摳圖去除背景的網(wǎng)站。只要上傳圖片，就可以自動把背景去掉把目標(biāo)對象摳出來。
不管是動物、汽車或各種產(chǎn)品，還是人物，都可以全自動做到令人驚艷的摳圖效果！而且還說可以做到發(fā)絲級 AI 自動摳圖，作為一個頭發(fā)存留不多的程序員，突然莫名感覺到一絲來自這個世界的惡意。
這個摳圖有什么用？比如可以把人物摳出來換個背景：
也可以把產(chǎn)品摳出來做成新的商品宣傳圖：（做電商的朋友肯定懂）
這么好用的東西，現(xiàn)在最關(guān)鍵的是：完全免費(fèi)！
emmmmmm，這個消息被我的幾十個前任女友知道了，她們紛紛發(fā)來照片，要我給她把照片背景給去掉，把她美美的人像給摳出來，之后好換上馬爾代夫的背景。
這么多的圖片要一張張傳上去，再下載回來，同樣的事情做幾十遍那我肯定不會做的。
那么，必須要寫個 Python 自動化腳本程序來搞定了。
說干就干，二話不說就掏出網(wǎng)頁自動化的神器Selenium，寫簡單幾行代碼：
""" Selenium是第三方庫，需要先安裝,在Teminal里執(zhí)行：pip install selenium 驅(qū)動程序 chromedriver 需要根據(jù)瀏覽器 chrome 的版本下載 有問題的可以加我：youbafu （全網(wǎng)統(tǒng)一id） from selenium import webdriver from selenium.webdriver.chrome.service import Service s = Service(r'D:\driver\chromedriver.exe') # 初始化瀏覽器為chrome瀏覽器 browser = webdriver.Chrome(service=s) # 打開 片刻摳圖 browser.get(r'https://www.pickwant.com')
就完成了自動打開網(wǎng)頁：
但這是遠(yuǎn)遠(yuǎn)不夠的，我們需要實(shí)現(xiàn)自動上傳圖片，然后自動下載。
通過幾次手動操作得出規(guī)律，我們只要進(jìn)入到編輯界面
然后循環(huán)執(zhí)行以下步驟：
1、點(diǎn)擊網(wǎng)頁右上角的重新上傳按鈕。
2、選擇要上傳的圖片，點(diǎn)打開按鈕。
3、等幾秒摳好圖后，點(diǎn)網(wǎng)頁右上角下載按鈕。
4、彈出對話框后，再點(diǎn)確認(rèn)下載就會自動保存圖片。
就可以依次把文件夾里的前女友們的圖片完成自動摳圖了。
要做出自動化腳本，接下來就是找到這些按鈕，然后模擬點(diǎn)擊就可以了。在重新下載按鈕上點(diǎn)鼠標(biāo)右鍵，在彈出的菜單中點(diǎn)檢查，彈出了網(wǎng)頁的元素查看界面，再重復(fù)操作一次，就定位到了這個按鈕的網(wǎng)頁代碼：
可以看到重新上傳是個 span ，這里可以取它的上級 div，它的class是border_wrap upload_wrap，那么就可以寫代碼定位到這個元素，然后執(zhí)行click()操作了。
import time from selenium import webdriver from selenium.webdriver.chrome.service import Service from selenium.webdriver.common.by import By s = Service(r'D:\driver\chromedriver.exe') # 初始化瀏覽器為chrome瀏覽器 browser = webdriver.Chrome(service=s) # 打開 片刻摳圖 browser.get(r'https://www.pickwant.com/#/editor') time.sleep(1) # 定位到 重新上傳 按鈕 upload = browser.find_element(By.CSS_SELECTOR, '.border_wrap.upload_wrap') upload.click()
這里要特別說明的是，根據(jù) CSS 的基本語法，id用#表示，class用.來表示，如果中間有空格也是用.來拼接，下級標(biāo)簽用>來拼接。
<div data-v-0373fdbc="" class="border_wrap upload_wrap"></div>
對于這樣的 div 標(biāo)簽，其class是 border_wrap upload_wrap ，所以 By.CSS_SELECTOR 的查找元素的參數(shù)就是 ‘.border_wrap.upload_wrap’ 。
代碼執(zhí)行后，彈出一個選擇文件的對話框：
這時候問題來了，這個對話框是 Windows 系統(tǒng)級別的，不受Selenium控制，怎么辦？
不要慌，這難不倒霸夫老師，既然是 Windows 系統(tǒng)彈窗，那總得受系統(tǒng)控制吧！
那就祭出 pywin32 庫來對付它，在終端命令行（Terminal）里輸入命令安裝庫：
pip install pywin32
然后再打開查看 Windows 窗口信息的神器spy++，拿到窗口的類名和層次結(jié)構(gòu)關(guān)系，就可以寫出以下的自動選擇文件并點(diǎn)擊按鈕的函數(shù)代碼了:
import win32con # win32庫里包含了很多模塊
import win32gui # 這里導(dǎo)入具體用到的模塊
def 上傳圖片(文件路徑):
dialog = win32gui.FindWindow('#32770', u'打開') # 對話框
ComboBoxEx32 = win32gui.FindWindowEx(dialog, 0, 'ComboBoxEx32', None)
ComboBox = win32gui.FindWindowEx(ComboBoxEx32, 0, 'ComboBox', None)
Edit = win32gui.FindWindowEx(ComboBox, 0, 'Edit', None) # 上面三句依次尋找對象，直到找到輸入框Edit對象的句柄
button = win32gui.FindWindowEx(dialog, 0, 'Button', None) # 打開按鈕
time.sleep(0.5)
win32gui.SendMessage(Edit, win32con.WM_SETTEXT, None, 文件路徑) # 往輸入框輸入絕對地址
time.sleep(0.5)
win32gui.SendMessage(dialog, win32con.WM_COMMAND, 1, button) # 點(diǎn) 打開 按鈕
time.sleep(5)先寫好這個函數(shù)備用，等下要上傳哪張圖片，只要調(diào)用函數(shù)時把圖片的文件路徑作為參數(shù)傳入，就可以完成目標(biāo)了。
上傳圖片后，只要稍等一兩秒就自動完成摳圖了。這時候再點(diǎn)網(wǎng)頁右上角下載按鈕，等彈出對話框后，再點(diǎn)確認(rèn)下載就會自動保存圖片。
前面我們學(xué)習(xí)了定位元素的方法了，再定位這兩個按鈕對我們來說不是什么困難了。
但是，你都能堅持看到這里了，霸夫老師必須要教你個更牛X的更方便的技巧了：
在下載按鈕上點(diǎn)鼠標(biāo)右鍵，在彈出的菜單中點(diǎn)檢查，彈出了網(wǎng)頁的元素查看界面，再重復(fù)操作一次，就定位到了這個按鈕的網(wǎng)頁代碼。然后在它的上一級的這個div標(biāo)簽上點(diǎn)鼠標(biāo)右鍵，選擇Copy，再選擇Copy selector。
然后回到python代碼編輯窗口，粘貼就可以獲得準(zhǔn)確的定位參數(shù)：
#app > div > div:nth-child(1) > div.col-md-12.wonx_row.top_bar > div.border_wrap.download_wrap
這樣就省去了辛辛苦苦去理解和轉(zhuǎn)化參數(shù)，大大提高效率，接下來就可以輕松寫出定位元素和點(diǎn)擊的代碼：
download = browser.find_element(By.CSS_SELECTOR, '#app > div > div:nth-child(1) > div.col-md-12.wonx_row.top_bar > div.border_wrap.download_wrap') download.click() # 點(diǎn)擊 下載 time.sleep(0.5) comfirm = browser.find_element(By.CSS_SELECTOR, '#app > div > div.container > div.dialog-content > div > div.download_wrap') comfirm.click() # 點(diǎn)擊 確認(rèn)下載 time.sleep(1)
單張圖片的上傳和下載流程都寫完了，接下來要寫個獲取文件夾下所有圖片的函數(shù)，來實(shí)現(xiàn)依次把每個前任的圖片都完成摳圖的任務(wù)。
我把她們的照片都放在電腦的 E: 盤下的前任女友這個文件夾里，那么只要讀取這個文件夾下的所有圖片，依次執(zhí)行執(zhí)行一遍上傳和下載操作就可以了。
文件夾路徑 = r'E:\前任女友'
names = os.listdir(文件夾路徑) # 列舉出該目錄下所有的文件名
for name in names:
path = os.path.join(文件夾路徑, name) # 拼接出圖片完整路徑
# 點(diǎn)擊重新上傳() # 等待封裝
上傳圖片(path)
# 點(diǎn)擊下載() # 等待封裝所有的邏輯都理清楚了，模塊也基本寫好了，接下來就把代碼整合一下，來看完整代碼：
"""
片刻摳圖
圖片批量自動摳圖去背景
作者：有霸夫
有問題請加vx：youbafu
import time
from selenium import webdriver
from selenium.webdriver.chrome.service import Service
from selenium.webdriver.common.by import By
import win32con
import win32gui
import os
s = Service(r'D:\driver\chromedriver.exe')
# 初始化瀏覽器為chrome瀏覽器
browser = webdriver.Chrome(service=s)
# 打開 片刻摳圖
browser.get(r'https://www.pickwant.com')
time.sleep(1)
browser.get(r'https://www.pickwant.com/#/editor')
time.sleep(2)
def 點(diǎn)擊重新上傳():
while True:
try:
# 定位到 重新上傳 按鈕
upload = browser.find_element(By.CSS_SELECTOR, '.border_wrap.upload_wrap')
upload.click()
except:
print('請先掃描登錄或完成驗(yàn)證碼識別')
else:
print('點(diǎn)擊 重新上傳')
break
finally:
time.sleep(1)
def 上傳圖片(文件路徑):
dialog = win32gui.FindWindow('#32770', u'打開') # 對話框
ComboBoxEx32 = win32gui.FindWindowEx(dialog, 0, 'ComboBoxEx32', None)
ComboBox = win32gui.FindWindowEx(ComboBoxEx32, 0, 'ComboBox', None)
Edit = win32gui.FindWindowEx(ComboBox, 0, 'Edit', None) # 上面三句依次尋找對象，直到找到輸入框Edit對象的句柄
button = win32gui.FindWindowEx(dialog, 0, 'Button', None) # 打開按鈕
if Edit is not None and button is not None:
time.sleep(0.5)
win32gui.SendMessage(Edit, win32con.WM_SETTEXT, None, 文件路徑) # 往輸入框輸入絕對地址
print('上傳：', 文件路徑)
win32gui.SendMessage(dialog, win32con.WM_COMMAND, 1, button) # 點(diǎn) 打開 按鈕
time.sleep(5)
def 點(diǎn)擊下載():
download = browser.find_element(By.CSS_SELECTOR, '.border_wrap.download_wrap')
download.click()
time.sleep(0.5)
confirm = browser.find_element(By.CSS_SELECTOR, '.dialog-content>.content_wrap>.download_wrap')
confirm.click()
print('點(diǎn)擊 下載')
文件夾路徑 = r'E:\前任女友'
names = os.listdir(文件夾路徑) # 列舉出該目錄下所有的文件名
for name in names:
path = os.path.join(文件夾路徑, name) # 拼接出圖片完整路徑
點(diǎn)擊重新上傳()
上傳圖片(path)
點(diǎn)擊下載()
print('恭喜！全部完成')在完整代碼中加入了異常處理，如果遇到需要掃碼登錄或者驗(yàn)證碼，手動操作一下，還會繼續(xù)執(zhí)行下去的。
到此這篇關(guān)于Python圖片批量自動摳圖去背景的文章就介紹到這了,更多相關(guān)Python批量自動摳圖去背景內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！
相關(guān)文章
Python圖片轉(zhuǎn)gif方式(將靜態(tài)圖轉(zhuǎn)化為分塊加載的動態(tài)圖)
這篇文章主要介紹了Python圖片轉(zhuǎn)gif方式(將靜態(tài)圖轉(zhuǎn)化為分塊加載的動態(tài)圖),具有很好的參考價值,希望對大家有所幫助,如有錯誤或未考慮完全的地方,望不吝賜教2023-11-11
python 實(shí)現(xiàn)判斷ip連通性的方法總結(jié)
下面小編就為大家分享一篇python 實(shí)現(xiàn)判斷ip連通性的方法總結(jié)，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2018-04-04
python實(shí)現(xiàn)京東訂單推送到測試環(huán)境，提供便利操作示例
這篇文章主要介紹了python實(shí)現(xiàn)京東訂單推送到測試環(huán)境，提供便利操作,涉及Python基于requests模塊的網(wǎng)絡(luò)請求操作相關(guān)使用技巧,需要的朋友可以參考下2019-08-08
Python實(shí)現(xiàn)對百度云的文件上傳(實(shí)例講解)
下面小編就為大家?guī)硪黄狿ython實(shí)現(xiàn)對百度云的文件上傳(實(shí)例講解)。小編覺得挺不錯的，現(xiàn)在就分享給大家，也給大家做個參考。一起跟隨小編過來看看吧2017-10-10
在本篇文章里小編給大家整理的是一篇關(guān)于python中設(shè)置代碼自動提示的方法，有需要的朋友們可以學(xué)習(xí)參考下。2020-07-07
python3利用Socket實(shí)現(xiàn)通信的方法示例
這篇文章主要介紹了python3利用Socket實(shí)現(xiàn)通信的方法示例，文中通過示例代碼介紹的非常詳細(xì)，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值，需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧2019-05-05
這篇文章主要介紹了如何利用Python編寫一個簡易版的天天跑酷游戲，文中的示例代碼講解詳細(xì)，對我們學(xué)習(xí)Python有一定幫助，需要的可以參考一下2022-03-03
這篇文章主要介紹了python查看自己安裝的所有庫并導(dǎo)出,主要包括查看安裝的庫通過命令查詢，導(dǎo)出庫安裝文件執(zhí)行命令，本文給大家介紹的非常詳細(xì)，對大家的學(xué)習(xí)或工作具有一定的參考借鑒價值，需要的朋友可以參考下2022-06-06
Python 在OpenCV里實(shí)現(xiàn)仿射變換—坐標(biāo)變換效果
這篇文章主要介紹了Python 在OpenCV里實(shí)現(xiàn)仿射變換—坐標(biāo)變換效果，本文通過一個例子給大家介紹的非常詳細(xì)，具有一定的參考借鑒價值,需要的朋友可以參考下2019-08-08
Python callable()函數(shù)用法實(shí)例分析
這篇文章主要介紹了Python callable()函數(shù)用法,結(jié)合實(shí)例形式分析了Python callable()函數(shù)的功能、使用方法及相關(guān)操作注意事項,需要的朋友可以參考下2018-03-03