【人人都是深度学习师】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

 

原图

 

 

线稿图

 

 

分析

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

 

参考

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