三行代码帮你自动画图

一周之前，百度开源的自动画图功能爆火，它可以三行就实现各种风格的自动画图。

首先安装包

!pip install paddlehub

然后就可以用三行代码画图了

import paddlehub as hub
module = hub.Module(name="ernie_vilg")
results = module.generate_image(text_prompts=["波涛汹涌的大海"])

这个包还提供了不同的画风，比如我们把ernie_vilg改成stable_diffusion

module = hub.Module(name="stable_diffusion")
results = module.generate_image(text_prompts=["波涛汹涌的大海"])

事实上，paddlehub除了画图外还有很多功能，下面我们也顺带做点探索。

1. 人像分割

import paddlehub as hub
module = hub.Module(name="humanseg_lite")
res = module.segment(
paths = ["human.png"],
visualization=True,
output_dir='humanseg_output')

可以看到，效果还是不错的，如果图片是白底的话，效果应该更好。

2. 人脸检测

import paddlehub as hub
module = hub.Module(name="ultra_light_fast_generic_face_detector_1mb_640")
res = module.face_detection(
paths = ["human.png"],
visualization=True,
output_dir='humanseg_output')

3. 人体姿势检测

import paddlehub as hub
module = hub.Module(name="openpose_body_estimation")
res = module.predict(
img = "human.png",
visualization=True,
save_path='humanseg_output')