滑稽，使用paddle輕鬆搞定摳圖，媽媽再也不用擔心我不會摳圖了

寫在前面： 我是「虐貓人薛定諤i」，一個不滿足於現狀，有夢想，有追求的00後
$\quad$
本博客主要記錄和分享自己畢生所學的知識，歡迎關注，第一時間獲取更新。
$\quad$
不忘初心，方得始終。
$\quad$

❤❤❤❤❤❤❤❤❤❤

環境搭建

需要安裝paddle和paddlehub，如果不會的小夥伴，可以參考paddle的官網。
傳送門

快速安裝教程

效果展示

代碼實現

import os
import paddlehub as hub

# 加載模型
humanseg = hub.Module(name='deeplabv3p_xception65_humanseg')

# 圖片路徑
path = "D:/Code/Python/res/"

# 判斷路徑是否存在
if os.path.exists(path):
    files = os.listdir(path)
    imgs = []
    for item in files:
        imgs.append(path + item)
    # 批量摳圖
    results = humanseg.segmentation(data={'image': imgs},
                                    visualization=True,
                                    output_dir='D:/Code/Python/result/')
    print("---- 程序結束 ----")
else:
    print("輸入的路徑不存在！！！")
    print("---- 程序結束 ----")

總結分析

這裏說一下我遇到的問題，網上的教程大部分都說，運行完成後，我們可以在項目下看到humanseg_output目錄，但是我運行後，代碼沒有報錯，卻沒有這個目錄。其實，這是因爲paddle升級了，網上的那是以前的版本，新版本需要自己指定一個目錄來存放摳圖結果，而且，還要將visualization設置爲True。
最後，我還要感謝ZackSock大佬。

蒟蒻寫博客不易，加之本人水平有限，寫作倉促，錯誤和不足之處在所難免，謹請讀者和各位大佬們批評指正。
如需轉載，請署名作者並附上原文鏈接，蒟蒻非常感激
名稱：虐貓人薛定諤i
博客地址：https://blog.csdn.net/Deep___Learning