飞桨是百度自主研发的开源深度学习平台。包含深度学习核心训练和推理框架、工具组件、基础模型库、端到端开发套件、 预测部署和开发训练。
今天要说的PaddleHub是飞桨中的一个工具组件,包含了大量的预训练模型,不需要自己训练可以拿来直接使用,也可以根据自己的需要做一些少量的训练进行微调。
预训练模型,从应用场景可以分为文本、视频、图像、语音、工业应用五大类模型库到目前为止共400多个模型。
生成证件照的思路,从一张人像图片生成证件照,需要两大步,第一步利用PaddleHub中的图像分割模型,把人像从原图中扣出,第二步使用Python图像处理库PIL (Python Imaging Library) ,对图像进行处理,调整图像大小,设置图像底色即可保存证件照。
一,部署PaddleHub
参考官网安装方式
pip install --upgrade paddlepaddle -i https://mirror.baidu.com/pypi/simple
pip install --upgrade paddlehub -i https://mirror.baidu.com/pypi/simple
二,选择图像分割模型
这里使用推荐的模型 deeplabv3p_xception65_humanseg
在当前页面向下找到学习模型,点击更多模型进入选择模型界面。从
在左侧导航栏中从应用场景=>图像=>图像分割 右侧会出现多个模型,可以根据自己需求选择对应的模型,这里选择deeplabv3p_xception65_humanseg
三,测试
使用下面的脚本对图像进行分割扣图
import paddlehub as hub
import cv2
human_seg = hub.Module(name="deeplabv3p_xception65_humanseg")
result = human_seg.segment(images=[cv2.imread('/PATH/TO/IMAGE')],visualization=True,output_dir="output")
在模型页面下方有API预测的相关命令
四,生成证件照
PaddleHub部署完成后,就可以使用python调用PaddleHub生成证件照了下面是生成证件照的代码及程序界面
代码如下:
###########################################
# 生成证件照 配置环境
# sudo apt install python3-tk
# pip install Pillow
# 先导入python环境再执行此脚本
# source ~/paddlehub_env/bin/activate
###########################################
import tkinter as tk
from tkinter import filedialog
import os
from PIL import Image, ImageDraw, ImageFont, ImageTk
import paddlehub as hub
import sys
import cv2
import
标签:PaddleHub,模型,生成,install,图像,import,证件照
From: https://blog.csdn.net/qq_40089934/article/details/141093626