Python快速實現(xiàn)一鍵摳圖功能的全過程
更新時間：2021年06月29日 11:11:37 作者：XerCis
你有沒想過,Python也能成為這樣的一種工具:在只有一張圖片,需要細致地摳出人物的情況下,能幫你減少摳圖步驟,這篇文章主要給大家介紹了關(guān)于Python快速實現(xiàn)一鍵摳圖功能的相關(guān)資料,需要的朋友可以參考下
簡介
使用百度深度學習框架paddlepaddle對人像圖片進行自動化摳圖
安裝
根據(jù)PaddlePaddle官網(wǎng)命令安裝
如
pip install paddlepaddle-gpu -i https://mirror.baidu.com/pypi/simple pip install paddlehub -i https://mirror.baidu.com/pypi/simple
初試
1.jpg
2.jpg
3.jpg
4.jpg
5.jpg
import paddlehub as hub
from pathlib import Path
paths = [str(i) for i in Path('.').glob('*.jpg')] # 當前路徑下所有.jpg文件
human_seg = hub.Module(name='deeplabv3p_xception65_humanseg')
results = human_seg.segmentation(paths=paths, visualization=True, output_dir='output')
# results = human_seg.segmentation(paths=paths, use_gpu=True, visualization=True, output_dir='output') # 使用GPU
print(results)
代碼會自動下載圖像分割模型deeplabv3p_xception65_humanseg到C:\Users\Administrator\.paddlehub\modules
效果
| 文件名 | 原圖 | 效果 |
|---|---|---|
| 1.jpg | |
 |
| 2.jpg | |
 |
| 3.jpg | |
 |
| 4.jpg | |
 |
| 5.jpg | |
 |
詳解
人像分割API
def segmentation(images=None,
paths=None,
batch_size=1,
use_gpu=False,
visualization=False,
output_dir='humanseg_output')
參數(shù)
- images(list[numpy.ndarray])：圖片數(shù)據(jù)，BGR格式
- paths(list[str])：圖片路徑
- batch_size(int)：批量處理數(shù)量
- use_gpu(bool)：是否使用 GPU
- visualization(bool)：是否將識別結(jié)果保存為圖片
- output_dir(str)：圖片保存路徑
遇到的坑
1. 報錯RuntimeError: Environment Variable CUDA_VISIBLE_DEVICES is not set correctly. If you wanna use gpu, please set CUDA_VISIBLE_DEVICES as cuda_device_id.
import os os.environ['CUDA_VISIBLE_DEVICES'] = '0'
或
set CUDA_VISIBLE_DEVICES=0
參考文獻
一款Python實用神器，5 行 Python 代碼 實現(xiàn)一鍵批量扣圖
總結(jié)
到此這篇關(guān)于Python快速實現(xiàn)一鍵摳圖功能的文章就介紹到這了,更多相關(guān)Python一鍵摳圖內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！
相關(guān)文章
淺談python中的@以及@在tensorflow中的作用說明
這篇文章主要介紹了淺談python中的@以及@在tensorflow中的作用說明，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2021-03-03
python3+PyQt5使用數(shù)據(jù)庫窗口視圖
這篇文章主要為大家詳細介紹了python3+PyQt5使用數(shù)據(jù)庫窗口視圖，具有一定的參考價值，感興趣的小伙伴們可以參考一下2018-04-04
基于Python實現(xiàn)中秋佳節(jié)月餅搶購腳本
這篇文章主要介紹了Python版中秋佳節(jié)月餅搶購腳本,今天要用的是一個測試工具的庫Selenium，今天我們就是用它去實現(xiàn)自動化搶購月餅，其實就是用這個工具"模擬"人為操作瀏覽器相應的操作，比如登陸，勾選購物車商品，下單購買等等操作，需要的朋友可以參考下2022-09-09
這篇文章主要介紹了Python中time模塊和datetime模塊的用法示例,主要演示了一些時間日期的打印和計算,需要的朋友可以參考下2016-02-02
一篇文章從零開始創(chuàng)建conda環(huán)境、常用命令的使用及pycharm配置項目環(huán)境
在Conda中創(chuàng)建新環(huán)境是一個非常有用的做法,尤其是當你需要為不同的項目安裝不同版本的軟件包時,這篇文章主要給大家介紹了關(guān)于從零開始創(chuàng)建conda環(huán)境、常用命令的使用及pycharm配置項目環(huán)境的相關(guān)資料,需要的朋友可以參考下2024-07-07
讀萬卷書不如行萬里路，學的扎不扎實要通過實戰(zhàn)才能看出來，本篇文章手把手帶用python來做幾個整蠱的小程序,大家可以在過程中查缺補漏，看看自己掌握程度怎么樣，發(fā)給朋友加固一下友誼2021-10-10
這篇文章主要介紹了python的去重以及數(shù)據(jù)合并的用法說明，具有很好的參考價值，希望對大家有所幫助。如有錯誤或未考慮完全的地方，望不吝賜教2023-02-02
今天小編就為大家分享一篇解決PySide+Python子線程更新UI線程的問題，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2019-01-01
這篇文章主要介紹了Python 中 Elias Delta 編碼，下面的文章我們將使用 python 實現(xiàn) Elias Delta 編碼。具體詳細內(nèi)容,需要的朋友可以參考一下2021-11-11
這篇文章主要為大家詳細介紹了Python如何通過PyCAN和CANard庫實現(xiàn)編寫測試腳本,文中的示例代碼講解詳細,感興趣的小伙伴可以跟隨小編一起學習一下2025-02-02