图像处理方法、装置、电子设备及存储介质制造方法及图纸

本申请公开了一种图像处理方法、装置、电子设备及存储介质，该图像处理方法包括：获取待上色的第一灰度图像，其中，第一灰度图像包括待上色的人像区域；将第一灰度图像输入至预先训练的人像抠图模型，得到人像区域对应的人像掩膜图像作为第一掩膜图像，其中，人像掩膜图像中人像区域的灰度值与除人像区域以外的背景区域的灰度值不同；基于第一灰度图像以及第一掩膜图像，获取人像区域对应的第二灰度图像；将第一灰度图像以及第二灰度图像输入至预先训练的上色模型，得到对第一灰度图像中的人像区域上色后的第一人像图像。本方法可以实现对图像中的人像区域进行针对性上色，降低算力的消耗。的消耗。的消耗。

【技术实现步骤摘要】
图像处理方法、装置、电子设备及存储介质


[0001]本申请涉及图像处理
，更具体地，涉及一种图像处理方法、装置、电子设备及存储介质。

技术介绍

[0002]随着科技水平和生活水平的快速进步，电子设备(例如智能手机、平板电脑等)已经成为人们生活中常用的电子产品之一。人们可以利用电子设备拍摄和保存大量的照片，并且一些情况下会有针对保存的黑白照片进行上色，使得黑白照片转换为彩色照片的需求，但是，相关技术中针对黑白照片进行上色时会消耗大量的算力，并且处理速度较慢。

技术实现思路

[0003]本申请提出了一种图像处理方法、装置、电子设备及存储介质，可以实现对图像中的人像区域进行针对性上色，降低算力的消耗。
[0004]第一方面，本申请实施例提供了一种图像处理方法，所述方法包括：获取待上色的第一灰度图像，其中，所述第一灰度图像包括待上色的人像区域；将所述第一灰度图像输入至预先训练的人像抠图模型，得到所述人像区域对应的人像掩膜图像作为第一掩膜图像，其中，所述人像掩膜图像中所述人像区域的灰度值与除所述人像区域以外的背景区域的灰度值不同；基于所述第一灰度图像以及所述第一掩膜图像，获取所述人像区域对应的第二灰度图像；将所述第一灰度图像以及所述第二灰度图像输入至预先训练的上色模型，得到对所述人像区域上色后的第一人像图像。
[0005]第二方面，本申请实施例提供了一种图像处理装置，所述装置包括：第一获取模块、人像分割模块、第二获取模块以及图像上色模块，其中，所述第一获取模块用于获取待上色的第一灰度图像，其中，所述第一灰度图像包括待上色的人像区域；所述人像分割模块用于将所述第一灰度图像输入至预先训练的人像抠图模型，得到所述人像区域对应的人像掩膜图像作为第一掩膜图像，其中，所述人像掩膜图像中所述人像区域的灰度值与除所述人像区域以外的背景区域的灰度值不同；所述第二获取模块用于基于所述第一灰度图像以及所述第一掩膜图像，获取所述人像区域对应的第二灰度图像；所述图像上色模块用于将所述第一灰度图像以及所述第二灰度图像输入至预先训练的上色模型，得到对所述人像区域上色后的第一人像图像。
[0006]第三方面，本申请实施例提供了一种电子设备，包括：一个或多个处理器；存储器；一个或多个应用程序，其中所述一个或多个应用程序被存储在所述存储器中并被配置为由所述一个或多个处理器执行，所述一个或多个应用程序配置用于执行上述第一方面提供的图像处理方法。
[0007]第四方面，本申请实施例提供了一种计算机可读取存储介质，所述计算机可读取存储介质中存储有程序代码，所述程序代码可被处理器调用执行上述第一方面提供的图像处理方法。
[0008]本申请提供的方案，通过获取待上色的第一灰度图像，其中，第一灰度图像包括待上色的人像区域；将第一灰度图像输入至预先训练的人像抠图模型，得到人像区域对应的人像掩膜图像作为第一掩膜图像，其中，人像掩膜图像中人像区域的灰度值与除人像区域以外的背景区域的灰度值不同；基于第一灰度图像以及第一掩膜图像，获取人像区域对应的第二灰度图像；将第一灰度图像以及第二灰度图像输入至预先训练的上色模型，得到对第一灰度图像中的人像区域上色后的第一人像图像。由此，可以实现对图像中的人像区域进行针对性上色，从而降低算力的消耗，并提升对图像进行上色的速度。
附图说明
[0009]为了更清楚地说明本申请实施例中的技术方案，下面将对实施例描述中所需要使用的附图作简单地介绍，显而易见地，下面描述中的附图仅仅是本申请的一些实施例，对于本领域技术人员来讲，在不付出创造性劳动的前提下，还可以根据这些附图获得其他的附图。
[0010]图1示出了根据本申请一个实施例的图像处理方法流程图。
[0011]图2示出了本申请实施例提供的人像掩膜图像的一种示意图。
[0012]图3示出了根据本申请另一个实施例的图像处理方法流程图。
[0013]图4示出了本申请实施例提供的人像抠图模型的一种结构示意图。
[0014]图5示出了本申请实施例提供的上色模型的一种结构示意图。
[0015]图6示出了根据本申请又一个实施例的图像处理方法流程图。
[0016]图7示出了根据本申请一个实施例的图像处理装置的一种框图。
[0017]图8是本申请实施例的用于执行根据本申请实施例的图像处理方法的电子设备的框图。
[0018]图9是本申请实施例的用于保存或者携带实现根据本申请实施例的图像处理方法的程序代码的存储单元。
具体实施方式
[0019]为了使本
的人员更好地理解本申请方案，下面将结合本申请实施例中的附图，对本申请实施例中的技术方案进行清楚、完整地描述。
[0020]相关技术中，针对黑白图像进行上色的方案主要分为两种，一种是基于传统图像处理的方式，另一种是基于深度学习的方式。其中，基于传统图像处理的方式进行上色时，稳定鲁棒性差；而基于深度学习的方式中，是针对整张图像进行上色，上色区域范围广，计算量大，不适合用于电子设备运行，并且针对人像图像进行上色时，人像背景复杂，需要大体量模型才能获取到不突兀的背景上色效果。
[0021]针对上述问题，专利技术人提出了本申请实施例提供的图像处理方法、装置、电子设备以及存储介质，可以实现对图像中的人像区域进行针对性上色，从而降低算力的消耗，并提升对图像进行上色的速度。其中，具体的图像处理方法在后续的实施例中进行详细的说明。
[0022]下面再结合附图对本申请实施例提供的图像处理方法进行详细介绍。
[0023]请参阅图1，图1示出了本申请一个实施例提供的图像处理方法的流程示意图。在具体的实施例中，所述图像处理方法应用于如图7所示的图像处理装置400以及配置有所述
图像处理装置400的电子设备100(图8)。下面将以电子设备为例，说明本实施例的具体流程，当然，可以理解的，本实施例所应用的电子设备可以为智能手机、平板电脑、智能手表、电子书等，在此不做限定。下面将针对图1所示的流程进行详细的阐述，所述图像处理方法具体可以包括以下步骤：
[0024]步骤S110：获取待上色的第一灰度图像，其中，所述第一灰度图像包括待上色的人像区域。
[0025]在本申请实施例中，待上色的第一灰度图像可以是黑白人像图像对应的灰度图像。电子设备可以获取待进行上色的第一灰度图像，第一灰度图像为包含有人像区域的图像，该人像区域为待上色的区域，也就是说，第一灰度图像包括人像区域以及除人像区域以外的背景区域。其中，人像区域是指人体区域所在的区域，人体区域可以包括身体区域、头部区域、头发区域等。可以理解地，一些年代较为久远的照片大多为黑白人像图像，黑白人像图像的视觉效果很难与彩色人像图像相比，或者一些设备支持以黑白模式拍摄，以黑白模式拍摄后存储的图像是黑白的，但是后期需要将黑白图像中的人像区域进行上色，因此存在用户将黑白人像图像转换为彩色人像图像的需求。
[0026]在一些实施方式，电子设备可以从本地获取待上色的第一灰度图像。可本文档来自技高网...

【技术保护点】

【技术特征摘要】
1.一种图像处理方法，其特征在于，所述方法包括：获取待上色的第一灰度图像，其中，所述第一灰度图像包括待上色的人像区域；将所述第一灰度图像输入至预先训练的人像抠图模型，得到所述人像区域对应的人像掩膜图像作为第一掩膜图像，其中，所述人像掩膜图像中所述人像区域的灰度值与除所述人像区域以外的背景区域的灰度值不同；基于所述第一灰度图像以及所述第一掩膜图像，获取所述人像区域对应的第二灰度图像；将所述第一灰度图像以及所述第二灰度图像输入至预先训练的上色模型，得到对所述人像区域上色后的第一人像图像。2.根据权利要求1所述的方法，其特征在于，所述人像抠图模型包括第一编码模块以及第一解码模块，所述将所述第一灰度图像输入至预先训练的人像抠图模型，得到所述人像区域对应的人像掩膜图像作为第一掩膜图像，包括：将所述第一灰度图像输入至所述第一编码模块，并通过所述第一解码模块对所述第一编码模块输出的图像特征进行解码，得到所述第一解码模块输出的所述人像区域对应的人像掩膜图像作为第一掩膜图像。3.根据权利要求2所述的方法，其特征在于，所述第一编码模块包括N层编码层，所述第一解码模块包括N层解码层，所述N层编码层依次级联，所述N层编码层与所述N层解码层一一对应，所述将所述第一灰度图像输入至所述第一编码模块，并通过所述第一解码模块对所述第一编码模块输出的图像特征进行解码，得到所述第一解码模块输出的所述人像区域对应的人像掩膜图像作为第一掩膜图像，包括：将所述第一灰度图像输入至所述第一编码模块，并将N层编码层中第N层编码层提取的中间特征输入至所述第N层编码层对应的解码层；针对第4层编码层至第N
‑
1层编码层中的每个编码层，将所述每个编码层中的提取的中间特征、所述每个编码层的下一层编码层对应的解码层解码后的中间特征以及经过目标处理后的所述每个编码层的下一层编码层的提取的中间特征进行融合后，输入至所述每个编码层对应的解码层，所述目标处理包括卷积处理和上采样处理；将第3层编码层提取的中间特征与所述第4层编码层对应的解码层解码后的中间特征进行融合后，输入至所述第3层编码层对应的解码层；将所述第3层编码层对应的解码层解码后的中间特征，输入至第2层编码层对应的解码层；将所述第2层编码层对应的解码层解码后的中间特征，输入至第1层编码层对应的解码层，得到所述第一解码模块输出的所述人像区域对应的人像掩膜图像作为第一掩膜图像。4.根据权利要求1所述的方法，其特征在于，所述上色模型包括第二编码模块、第三编码模块以及第二解码模块，所述将所述第一灰度图像以及所述第二灰度图像输入至预先训练的上色模型，得到对所述人像区域上色后的第一人像图像，包括：将所述第一灰度图像输入至所述第二编码模块，得到所述第二编码模块输出的第一图像特征；将所述第二灰度图像输入至所述第三编码模块，得到所述第三编码模块输出的第二图像特征；
将所述第一图像特征与所述第二图像特征进行融合，得到第三图像特征；将所述第三图像特征输入至所述第二解码模...

【专利技术属性】
技术研发人员：王凡祎，张严浩，冯天鹏，
申请(专利权)人：OPPO重庆智能科技有限公司，
类型：发明
国别省市：