【人人都是深度学习师】3行代码实现AI自动根据彩色图生成线稿图-图像分割

基础环境

基础环境使用镜像:registry.cn-hangzhou.aliyuncs.com/allen135681/easyml:ubuntu18.04-nvidia_cuda10.0-base-Miniconda3-py39_4.9.2_paddlepaddle2.0.1_paddlehub2.0.4_paddlenlp2.0.0rc10   pyin paddle

额外安装:

pip list|grep "paddle\|cv"
opencv-python      4.2.0.32
paddlehub          2.0.4
paddlepaddle       2.0.1

代码

hub install Extract_Line_Draft==1.0.0
import paddlehub as hub
Extract_Line_Draft_test = hub.Module(name="Extract_Line_Draft")
Extract_Line_Draft_test.ExtractLine("testImage.png", use_gpu=True) # execute predict

【人人都是深度学习师】3行代码实现AI自动根据彩色图生成线稿图-图像分割_第1张图片

【人人都是深度学习师】3行代码实现AI自动根据彩色图生成线稿图-图像分割_第2张图片

 

原图

【人人都是深度学习师】3行代码实现AI自动根据彩色图生成线稿图-图像分割_第3张图片 【人人都是深度学习师】3行代码实现AI自动根据彩色图生成线稿图-图像分割_第4张图片

 

线稿图

【人人都是深度学习师】3行代码实现AI自动根据彩色图生成线稿图-图像分割_第5张图片 【人人都是深度学习师】3行代码实现AI自动根据彩色图生成线稿图-图像分割_第6张图片

 

分析

提取线稿(Extract_Line_Draft),该模型可自动根据彩色图生成线稿图。该PaddleHub Module支持API预测及命令行预测。

 

参考

https://www.paddlepaddle.org.cn/hubdetail?name=Extract_Line_Draft&en_category=ImageSegmentation

你可能感兴趣的:(计算机视觉,迁移学习,模型动物园,图像分割,线稿图,深度学习,人工智能,机器学习)