[发明专利]物体简笔画的自动生成方法及装置

说明书

本发明提供了一种物体简笔画的自动生成方法及装置，通过对图像的物体检测、物体特征提取和匹配，快速选取物体对应的简笔画素材，解决了现有技术从拍摄图像中生成物体简笔画时，对图像尺寸、图像标注、系统资源有较高要求的问题，避免图像尺寸对算法的影响，减少复杂图像标注的消耗，有效提高系统的处理效率。

技术领域

本发明涉及图像处理领域，更具体地说，涉及一种物体简笔画的自动生成方法及装置。

背景技术

随着智能手机等智能终端设备性能的提高，越来越多的人习惯使用智能手机拍照和录制视频。智能手机的摄像设备的改进方向主要有两个：一是不断提高的硬件性能，拍摄的照片的质量越来越接近专业单反相机；而是日渐丰富的软件功能，能够提供特定场景模式、物体识别、物体美化等功能，满足人们多样的拍摄需求。

目前流行的一种功能是在拍摄图像上识别特定物体，并生成特定物体的简笔画，即物体简笔画生成方案，这种功能能够增加拍照的趣味性和互动性。例如，在拍摄天空图像时，天空中的云朵形态多样，依照云朵的边缘作出简笔画，使得在拍摄天空的这个特定场景下时，增强云朵的拍摄效果。

同时，物体简笔画生成方案也可以用于工业领域，例如产品海报的制作。现有的产品海报制作方法是采用人工的方式，根据产品的实物图片，使用图像处理软件进行处理，这样的方式效率较低，不利于大批量产品的海报生产。因此，需要一种简笔画生成方法，能够自动将拍摄的图像生成具有简明线条的简笔画，自动完成产品海报的制作，提高产品海报的制作效率。

然而现有的图像生成简笔画的方案大多利用结构化学习的算法，进行图像生成，特别是利用生成式对抗网络GAN进行图像生成算法。这些算法存在以下缺点：1、需要大量复杂的人工标注数据，实现同样的功能对样本的标注成本极高；2、图像生成的算法对图像大小敏感；3、结构化算法实现同样的功能，训练时候需要更大显存和更多计算资源。

因此，需要一种物体简笔画的自动生成方法，能够对所有尺寸和类型的拍摄图像进行图像简笔画生成处理，采用深度学习的方式，减少人工标注消耗的人力物力，避免对智能终端处理资源的大量消耗。

发明内容

有鉴于此，本发明的目的在于提供一种物体简笔画的自动生成方法及装置，用于解决现有技术从拍摄图像中生成物体简笔画时，对图像尺寸、图像标注、系统资源有较高要求的问题，避免图像尺寸对算法的影响，减少复杂图像标注的消耗，有效提高系统的处理效率。

为了解决上述技术问题，现提出的方案如下：

一种物体简笔画的自动生成方法，包括：

对输入图像进行物体检测，获取所述图像中的物体区域；

提取所述物体区域的特征向量；

将所述物体区域的特征向量与注册简笔画特征库中的特征向量模板进行相似度匹配，获得最相似的特征向量模板对应的注册简笔画素材；

根据所述注册简笔画素材在所述输入图像上生成物体简笔画。

优选的，所述对输入图像进行物体检测，获取所述图像中的物体区域，具体包括：

基于物体检测模型D确定所述图像中的物体数量，以及至少一个物体的物体区域；

其中，所述物体区域包括物体的位置坐标范围；

所述物体检测模型D的训练方法，具体包括：

获取已标注物体外轮廓坐标信息的图像集合作为训练样本A；

基于MobileNet V2主干网络，利用非极大值抑制NMS算法对所述训练样本A进行候选框的合并，训练得到物体检测模型D。

优选的，所述提取所述物体区域的特征向量，具体包括：