Skip to content

Latest commit

 

History

History
61 lines (51 loc) · 2.1 KB

P01_PaddleHub人像扣图.md

File metadata and controls

61 lines (51 loc) · 2.1 KB
# Author: Acer Zhang
# Datetime: 2021/9/17 
# Copyright belongs to the author.
# Please indicate the source for reprinting.
import os

from qgui import CreateQGUI
from qgui.banner_tools import GitHub
from qgui.notebook_tools import ChooseFileTextButton, ChooseDirTextButton, RunButton, BaseButton
from qgui import MessageBox

import paddlehub as hub
import cv2

human_seg = hub.Module(name="deeplabv3p_xception65_humanseg")

# 预测主程序
def infer(args):
    img_path = args["文件输入框"].get()
    out_path = args["保存位置"].get()

    # 简单做个判断，保证输入是正确的
    if not os.path.exists(img_path):
        MessageBox.info("请选择要分割的图片")
        # 不选择就不做预测了
        return 1
    if not os.path.exists(out_path):
        MessageBox.info("请选择图片保存目录")
        return 2

    # 执行分割
    print("开始分割")
    human_seg.segmentation(images=[cv2.imread(img_path)],
                           visualization=True,
                           output_dir=out_path)

    # 打开结果文件夹 不支持MAC系统
    os.startfile(out_path)
    print("处理完毕")


# 创建主界面
main_gui = CreateQGUI(title="基于PaddleHub的人像扣图小工具")

# 在界面最上方添加一个按钮，链接到GitHub主页
main_gui.add_banner_tool(GitHub("https://github.com/QPT-Family/QGUI"))
# 在主界面部分添加一个文件选择工具
main_gui.add_notebook_tool(ChooseFileTextButton(name="文件输入框"))
# 再加个文件夹选择工具
main_gui.add_notebook_tool(ChooseDirTextButton(name="保存位置"))
# 添加一个运行按钮
main_gui.add_notebook_tool(RunButton(infer))
main_gui.set_navigation_info(title="基于PaddleHub的人像扣图小工具",info="该项目基于PaddleHub和QGUI制作的人像扣图小工具，在右侧选择需要扣图的图片和保存位置，点击开始运行即可获取结果。")
# 简单加个简介
main_gui.set_navigation_about(author="GT",
                              version="0.0.1",
                              github_url="https://github.com/QPT-Family/QGUI")
# 跑起来~
main_gui.run()