三行代码帮你自动画图
【摘要】 一周之前，百度开源的自动画图功能爆火，它可以三行就实现各种风格的自动画图。首先安装包!pip install paddlehub然后就可以用三行代码画图了import paddlehub as hub module = hub.Module(name="ernie_vilg") results = module.generate_image(text_prompts=["波涛汹涌的大海"])...
一周之前，百度开源的自动画图功能爆火，它可以三行就实现各种风格的自动画图。
首先安装包
!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除了画图外还有很多功能，下面我们也顺带做点探索。
-
人像分割
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')
此外，PaddleHub还具有一些Linux命令行工具。
-
hub list
!hub list用于显示PaddleHub下安装的模型
其中ernie_vilg、humanseg_lite等都是我们在《三行代码帮你自动画图》中分享过的模型。
2. hub uninstall
用于卸载模型，我们以humanseg_lite为例
!hub uninstall humanseg_lite
3. hub install
用于安装模型，我们再把humanseg_lite装回去
!hub install humanseg_lite
4. hub show
用于显示已安装模型的信息，还是以humanseg_lite为例
!hub show humanseg_lite
5. hub run
用于运行模型，以lac模型为例
!hub run lac --input_text "今天是个好日子
此外，还有hub version，hub config等。
【版权声明】本文为华为云社区用户原创内容，转载时必须标注文章的来源（华为云社区）、文章链接、文章作者等基本信息， 否则作者和本社区有权追究责任。如果您发现本社区中有涉嫌抄袭的内容，欢迎发送邮件进行举报，并提供相关证据，一经查实，本社区将立刻删除涉嫌侵权内容，举报邮箱：
cloudbbs@huaweicloud.com
- 点赞
- 收藏
- 关注作者
评论（0）