Python快速實(shí)現(xiàn)一鍵摳圖功能的全過程
簡介
使用百度深度學(xué)習(xí)框架paddlepaddle對(duì)人像圖片進(jìn)行自動(dòng)化摳圖
安裝
根據(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')] # 當(dāng)前路徑下所有.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)
代碼會(huì)自動(dòng)下載圖像分割模型deeplabv3p_xception65_humanseg到C:\Users\Administrator\.paddlehub\modules
效果
| 文件名 | 原圖 | 效果 |
|---|---|---|
| 1.jpg | |
 |
| 2.jpg | |
 |
| 3.jpg | |
 |
| 4.jpg | |
 |
| 5.jpg | |
 |
詳解
人像分割A(yù)PI
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)：是否將識(shí)別結(jié)果保存為圖片
- output_dir(str)：圖片保存路徑
遇到的坑
1. 報(bào)錯(cuò)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
參考文獻(xiàn)
一款Python實(shí)用神器，5 行 Python 代碼 實(shí)現(xiàn)一鍵批量扣圖
總結(jié)
到此這篇關(guān)于Python快速實(shí)現(xiàn)一鍵摳圖功能的文章就介紹到這了,更多相關(guān)Python一鍵摳圖內(nèi)容請(qǐng)搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！
相關(guān)文章
在Python中,frozenset是一種與set類似的數(shù)據(jù)類型,但具有不可變性,本文將詳細(xì)介紹frozenset的作用、用法以及與set的區(qū)別,感興趣的可以了解下2024-01-01
python實(shí)現(xiàn)m3u8格式轉(zhuǎn)換為mp4視頻格式
這篇文章主要為大家詳細(xì)介紹了python實(shí)現(xiàn)m3u8格式轉(zhuǎn)換為mp4視頻格式，具有一定的參考價(jià)值，感興趣的小伙伴們可以參考一下2018-02-02
Python實(shí)現(xiàn)Opencv cv2.Canny()邊緣檢測
這篇博客將介紹Canny邊緣檢測的概念，并利用cv2.Canny()實(shí)現(xiàn)邊緣檢測，具有一定的參考價(jià)值，感興趣的小伙伴們可以參考一下2021-07-07
本文主要介紹了exe反編譯為.py文件的方法，文中通過示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值，需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧2023-03-03
python密碼學(xué)周期置換密碼學(xué)習(xí)
這篇文章主要為大家介紹了python密碼學(xué)周期置換密碼的學(xué)習(xí)，有需要的朋友可以借鑒參考下，希望能夠有所幫助，祝大家多多進(jìn)步，早日升職加薪2022-05-05
Python導(dǎo)入引用其他文件的函數(shù)實(shí)戰(zhàn)案例(推薦!)
這篇文章主要給大家介紹了關(guān)于Python導(dǎo)入引用其他文件的函數(shù)的相關(guān)資料,文中通過代碼以及圖文介紹的非常詳細(xì),對(duì)大家學(xué)習(xí)或者使用Python具有一定的參考借鑒價(jià)值,需要的朋友可以參考下2024-01-01
本篇文章主要介紹了Django實(shí)現(xiàn)組合搜索的方法示例，小編覺得挺不錯(cuò)的，現(xiàn)在分享給大家，也給大家做個(gè)參考。一起跟隨小編過來看看吧2018-01-01
Python使用pyglet庫完整實(shí)現(xiàn)漢諾塔游戲流程詳解
這篇文章主要介紹了Python使用pyglet庫完整實(shí)現(xiàn)漢諾塔游戲流程，漢諾塔問題是一個(gè)遞歸問題，也可以使用非遞歸法來解決，這個(gè)問題不僅是一個(gè)數(shù)學(xué)和邏輯問題，也是一個(gè)很好的教學(xué)工具，可以用來教授遞歸、算法和邏輯思考等概念，需要的朋友可以參考下2007-02-02
詳解Python的Django框架中的模版相關(guān)知識(shí)
這篇文章主要介紹了Python的Django框架中的模版相關(guān)知識(shí),模版的存在大大簡化了創(chuàng)作頁面時(shí)HTML的相關(guān)工作,需要的朋友可以參考下2015-07-07
這篇文章主要介紹了PyCharm 專業(yè)版安裝圖文教程，文中通過示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值，需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧2020-02-02