CN113554550B - 图像处理模型的训练方法、装置、电子设备及存储介质 - Google Patents

图像处理模型的训练方法、装置、电子设备及存储介质 Download PDF

Info

Publication number
CN113554550B
CN113554550B CN202110732721.1A CN202110732721A CN113554550B CN 113554550 B CN113554550 B CN 113554550B CN 202110732721 A CN202110732721 A CN 202110732721A CN 113554550 B CN113554550 B CN 113554550B
Authority
CN
China
Prior art keywords
image
feature
sample
difference information
sample image
Prior art date
Legal status (The legal status is an assumption and is not a legal conclusion. Google has not performed a legal analysis and makes no representation as to the accuracy of the status listed.)
Active
Application number
CN202110732721.1A
Other languages
English (en)
Other versions
CN113554550A (zh
Inventor
宋希彬
张良俊
Current Assignee (The listed assignees may be inaccurate. Google has not performed a legal analysis and makes no representation or warranty as to the accuracy of the list.)
Beijing Baidu Netcom Science and Technology Co Ltd
Original Assignee
Beijing Baidu Netcom Science and Technology Co Ltd
Priority date (The priority date is an assumption and is not a legal conclusion. Google has not performed a legal analysis and makes no representation as to the accuracy of the date listed.)
Filing date
Publication date
Application filed by Beijing Baidu Netcom Science and Technology Co Ltd filed Critical Beijing Baidu Netcom Science and Technology Co Ltd
Priority to CN202110732721.1A priority Critical patent/CN113554550B/zh
Publication of CN113554550A publication Critical patent/CN113554550A/zh
Application granted granted Critical
Publication of CN113554550B publication Critical patent/CN113554550B/zh
Active legal-status Critical Current
Anticipated expiration legal-status Critical

Links

Classifications

    • GPHYSICS
    • G06COMPUTING; CALCULATING OR COUNTING
    • G06TIMAGE DATA PROCESSING OR GENERATION, IN GENERAL
    • G06T3/00Geometric image transformations in the plane of the image
    • G06T3/40Scaling of whole images or parts thereof, e.g. expanding or contracting
    • G06T3/4053Scaling of whole images or parts thereof, e.g. expanding or contracting based on super-resolution, i.e. the output image resolution being higher than the sensor resolution
    • G06T3/4076Scaling of whole images or parts thereof, e.g. expanding or contracting based on super-resolution, i.e. the output image resolution being higher than the sensor resolution using the original low-resolution images to iteratively correct the high-resolution images
    • GPHYSICS
    • G06COMPUTING; CALCULATING OR COUNTING
    • G06FELECTRIC DIGITAL DATA PROCESSING
    • G06F18/00Pattern recognition
    • G06F18/20Analysing
    • G06F18/21Design or setup of recognition systems or techniques; Extraction of features in feature space; Blind source separation
    • G06F18/214Generating training patterns; Bootstrap methods, e.g. bagging or boosting

Landscapes

  • Engineering & Computer Science (AREA)
  • Theoretical Computer Science (AREA)
  • Physics & Mathematics (AREA)
  • Data Mining & Analysis (AREA)
  • General Physics & Mathematics (AREA)
  • Artificial Intelligence (AREA)
  • Computer Vision & Pattern Recognition (AREA)
  • Evolutionary Biology (AREA)
  • Evolutionary Computation (AREA)
  • Bioinformatics & Computational Biology (AREA)
  • General Engineering & Computer Science (AREA)
  • Bioinformatics & Cheminformatics (AREA)
  • Life Sciences & Earth Sciences (AREA)
  • Image Processing (AREA)
  • Image Analysis (AREA)

Abstract

本公开提供了一种图像处理模型的训练方法、装置、电子设备及存储介质,涉及计算机技术领域,尤其涉及深度学习、计算机视觉等人工智能技术领域。具体实现方案为:获取第一样本图像,对第一样本图像进行多次特征增强处理,以分别得到多次特征增强处理后的多个标注样本图像,从多个标注样本图像之中确定第二样本图像,获取第一样本图像和第二样本图像之间的标注特征差异信息,以及根据第一样本图像、多个标注样本图像,以及标注特征差异信息训练初始的图像处理模型,以得到目标图像处理模型,由此有效地提升了模型的收敛效率,降低了对真值图像的依赖程度,提升训练得到的图像处理模型针对图像特征的表达建模能力,提升图像处理模型的图像处理效果。

Description

图像处理模型的训练方法、装置、电子设备及存储介质
技术领域
本公开涉及计算机技术领域,尤其涉及深度学习、计算机视觉等人工智能技术领域,具体涉及一种图像处理模型的训练方法、装置、电子设备及存储介质。
背景技术
人工智能是研究使计算机来模拟人的某些思维过程和智能行为(如学习、推理、思考、规划等)的学科,既有硬件层面的技术也有软件层面的技术。人工智能硬件技术一般包括如传感器、专用人工智能芯片、云计算、分布式存储、大数据处理等技术;人工智能软件技术主要包括计算机视觉技术、语音识别技术、自然语言处理技术以及机器学习/深度学习、大数据处理技术、知识图谱技术等几大方向。
相关技术中,图像超分辨率是指由一幅低分辨率图像恢复出高分辨率图像。图像超分辨率具有广大的应用前景,可应用于多种应用场景中,例如,图像分割,物体检测,深度估计等应用场景。
发明内容
本公开提供了一种图像处理模型的训练方法、图像处理方法、装置、电子设备、存储介质以及计算机程序产品。
根据本公开的第一方面,提供了一种图像处理模型的训练方法,包括:获取第一样本图像;对所述第一样本图像进行多次特征增强处理,以分别得到所述多次特征增强处理后的多个标注样本图像;从所述多个标注样本图像之中确定第二样本图像;获取所述第一样本图像和所述第二样本图像之间的标注特征差异信息;以及根据所述第一样本图像、所述多个标注样本图像,以及所述标注特征差异信息训练初始的图像处理模型,以得到目标图像处理模型。
根据本公开的第二方面,提供了一种图像处理方法:包括:获取待处理图像,所述待处理图像具有对应的待处理图像特征;将所述待处理图像输入至如上述的图像处理模型的训练方法训练得到的目标图像处理模型之中,以得到所述目标图像处理模型输出的目标特征差异信息;以及根据所述目标特征差异信息对所述待处理图像特征进行特征增强处理,以得到目标图像特征,所述目标图像特征被融合至所述待处理图像之中以得到目标图像。
根据本公开的第三方面,提供了一种图像处理模型的训练装置,包括:第一获取模块,用于获取第一样本图像;第一处理模块,用于对所述第一样本图像进行多次特征增强处理,以分别得到所述多次特征增强处理后的多个标注样本图像;确定模块,用于从所述多个标注样本图像之中确定第二样本图像;第二获取模块,用于获取所述第一样本图像和所述第二样本图像之间的标注特征差异信息;以及训练模块,用于根据所述第一样本图像、所述多个标注样本图像,以及所述标注特征差异信息训练初始的图像处理模型,以得到目标图像处理模型。
根据本公开的第四方面,提供了一种图像处理装置,包括:第三获取模块,用于获取待处理图像,所述待处理图像具有对应的待处理图像特征;输入模块,用于将所述待处理图像输入至如上述的图像处理模型的训练装置训练得到的目标图像处理模型之中,以得到所述目标图像处理模型输出的目标特征差异信息;以及第三处理模块,用于根据所述目标特征差异信息对所述待处理图像特征进行特征增强处理,以得到目标图像特征,所述目标图像特征被融合至所述待处理图像之中以得到目标图像。
根据本公开的第五方面,提供了一种电子设备,包括:至少一个处理器;以及与至少一个处理器通信连接的存储器;其中,存储器存储有可被至少一个处理器执行的指令,指令被至少一个处理器执行,以使至少一个处理器能够执行如第一方面的图像处理模型的训练方法,或者执行如第二方面的图像处理方法。
根据本公开的第六方面,提供了一种存储有计算机指令的非瞬时计算机可读存储介质,计算机指令用于使计算机执行如第一方面的图像处理模型的训练方法,或者执行如第二方面的图像处理方法。
根据本公开的第七方面,提供了一种计算机程序产品,包括计算机程序,计算机程序在被处理器执行时实现如第一方面的图像处理模型的训练方法,或者执行如第二方面的图像处理方法。
应当理解,本部分所描述的内容并非旨在标识本公开的实施例的关键或重要特征,也不用于限制本公开的范围。本公开的其它特征将通过以下的说明书而变得容易理解。
附图说明
附图用于更好地理解本方案,不构成对本公开的限定。其中:
图1是根据本公开第一实施例的示意图;
图2是本公开实施例中图像处理模型的训练装置的架构示意图;
图3是本公开实施例中升采样训练单元的结构示意图;
图4是本公开实施例中残差学习结构示意图;
图5是本公开实施例中的残差学习模块的结构示意图;
图6是根据本公开第二实施例的示意图;
图7是本公开实施例中通道级特征增强处理流程示意图;
图8是本公开实施例中像素级特征增强处理流程示意图;
图9是本公开实施例中以非线性方式构建样本图像的示意图;
图10是根据本公开第三实施例的示意图;
图11是本公开实施例中图像处理方法的流程示意图;
图12是根据本公开第四实施例的示意图;
图13是根据本公开第五实施例的示意图;
图14是根据本公开第六实施例的示意图;
图15示出了可以用来实施本公开的实施例的图像处理模型的训练方法的示例电子设备的示意性框图。
具体实施方式
以下结合附图对本公开的示范性实施例做出说明,其中包括本公开实施例的各种细节以助于理解,应当将它们认为仅仅是示范性的。因此,本领域普通技术人员应当认识到,可以对这里描述的实施例做出各种改变和修改,而不会背离本公开的范围和精神。同样,为了清楚和简明,以下的描述中省略了对公知功能和结构的描述。
图1是根据本公开第一实施例的示意图。
其中,需要说明的是,本实施例的图像处理模型的训练方法的执行主体为图像处理模型的训练装置,该装置可以由软件和/或硬件的方式实现,该装置可以配置在电子设备中,电子设备可以包括但不限于终端、服务器端等。
本公开实施例涉及人工智能技术领域,具体涉及计算机视觉和深度学习技术领域。
其中,人工智能(Artificial Intelligence),英文缩写为AI。它是研究、开发用于模拟、延伸和扩展人的智能的理论、方法、技术及应用系统的一门新的技术科学。
深度学习是学习样本数据的内在规律和表示层次,这些学习过程中获得的信息对诸如文字,图像和声音等数据的解释有很大的帮助。深度学习的最终目标是让机器能够像人一样具有分析学习能力,能够识别文字、图像和声音等数据。
计算机视觉,指用摄影机和电脑代替人眼对目标进行识别、跟踪和测量等机器视觉,并进一步做图形处理,使电脑处理成为更适合人眼观察或传送给仪器检测的图像。
本公开实施例可应用于图像处理场景下,而图像处理场景可以例如,采用一些硬件设备或者软件计算处理逻辑对待处理图像进行识别,以识别得到相应的图像特征,并采用该图像特征辅助后续的检测应用,将本公开实施例提供的图像处理模型的训练方法应用于图像处理场景中,有效地提升了模型的收敛效率,降低了对真值图像的依赖程度,提升训练得到的图像处理模型针对图像特征的表达建模能力,提升图像处理模型的图像处理效果。
如图1所示,该图像处理模型的训练方法,包括:
S101:获取第一样本图像。
其中,用于训练模型的图像,可以被称为样本图像,该样本图像的数量可以是一张或者多张,该样本图像也可以是从多个视频帧之中提取出的部分帧视频图像,对此不做限制。
其中的第一样本图像,可以是作为输入以训练模型的样本图像,第一样本图像可以是低分辨率的样本图像(则本公开实施例中,可以图像特征是分辨率进行示例,图像处理模型支持对低分辨率的第一样本图像进行分辨率增强处理,以得到相对高分辨率的输出图像,或者图像特征也可以配置为其它任意可能的特征,对此不做限制)。
本公开实施例在获取一张或者多张的第一样本图像之后,可以触发对第一样本图像进行多次特征增强处理,以分别得到多次特征增强处理后的多个标注样本图像的步骤。
S102:对第一样本图像进行多次特征增强处理,以分别得到多次特征增强处理后的多个标注样本图像。
其中,对第一样本图像进行多次特征增强处理得到的样本图像,可以被称为标注样本图像,相应地,对第一样本图像进行多次特征增强处理,可以分别得到多次特征增强处理后的多个标注样本图像。
其中,标注样本图像可以被用于作为训练图像处理模型时的参考样本图像。
本公开实施例中,上述的第一样本图像和多个标注样本图像可以是在训练图像处理模型之前预先生成的,举例而言,可以在已有的包含高分辨率的深度图像(该高分辨率的深度图像可以被称为标注样本图像)的数据集中,选取N张标注样本图像组成训练数据集,而后,对N张标注样本图像,可以使用非线性方式对其进行图像构建,以生成与N张标注样本图像分别对应的低分辨率的深度图像(该低分辨率的深度图像可以被称为第一样本图像),从而组成高分辨-低分辨率深度图像对,作为训练数据。
本公开实施例中,在预先生成多个标注样本图像时,可以对第一样本图像进行当次的特征增强处理,以得到当次的标注样本图像;在进行下次的特征增强处理时,对当次的标注样本图像进行特征增强处理,以得到下次的标注样本图像,直至执行特征增强处理的次数满足设定次数。
如图2所示,图2是本公开实施例中图像处理模型的训练装置的架构示意图,可以低分辨率的图像(第一样本图像)作为输入,包含多个升采样训练单元和降采样训练单元。每个升采样单元的输出作为下一个升采样单元的输入,每个降采样训练单元的输出作为下一个降采样单元的输入。降采样训练单元的数量与升采样训练单元的数量相同,可以采用K标识,表示整体倍数与基本倍数的倍数关系,整个训练过程中,可以配置第一个训练单元(升采样训练单元和降采样训练单元)参考监督信号(标注特征差异信息)辅助进行训练,其它训练单元与第一个训练单元共享权重,从而保证整个训练过程可以收敛,且后续的训练单元可以不配置参考监督信号(标注特征差异信息)。
也即是说,输入数据可以是:第一样本图像,而后采用图2所示与输入数据相连的第一个升采样训练单元对第一样本图像进行特征增强处理以得到标注样本图像,而后,从多个标注样本图像之中可以采样得到第二样本图像,并触发后续采用第一样本图像和第二样本图像之间的标注特征差异信息作为整体模型训练用的参考监督信号,以训练图像处理模型的方法步骤。
上述的当次的标注样本图像,即可以是当次采用图2中的一个升采样训练单元处理输入的样本图像之后得到的样本图像,而下次的标注样本图像,可以是采用一个升采样训练单元处理输入的当次的标注样本图像之后得到的样本图像。
而不同标注样本图像对应的标注特征差异信息不相同,标注特征差异信息指示标注样本图像和第一样本图像之间的特征差异程度,可以例如是标注样本图像A的分辨率是第一样本图像的分辨率的2倍,标注样本图像B的分辨率是第一样本图像的分辨率的4倍,标注样本图像C的分辨率是第一样本图像的分辨率的8倍,对此不做限制。
其中的设定次数,可以是结合图像处理模型的应用场景需求自适应配置,对此不做限制。
也即是说,本公开实施例中,是接续地、多通道地对输入的第一样本图像进行特征增强处理,各次图像特征增强处理可以均配置为相同倍数的增强处理(例如,每个升采样训练单元支持对输入图像的分辨率增强两倍),从而可以简化整体的图像特征增强处理逻辑,实现采用接续地、多通道的图像增强处理方式即可以得到2倍、4倍、8倍的高分辨率的标注样本图像。
本公开实施例中,上述图2中的升采样训练单元可采用残差结构,包括卷积操作和图像重建操作,通过第一层卷积获得特征后,可以分为两个分支,其中一个分支通过卷积操作获得图像的高频特征,而后,将高频特征与另一分支进行像素级加法,再通过图像重建模块获得高分辨率的图像。
本公开实施例中,为了支持下述实施例的举例说明,还对上述图2中的升采样训练单元的架构进行了进一步的举例说明,如图3所示,图3是本公开实施例中升采样训练单元的结构示意图,以低分辨率的图像(第一样本图像)作为升采样训练单元的输入(此处需要说明的是,针对第一个升采样训练单元,可以将第一样本图像作为其输入,而针对接续的升采样训练单元,可以是采用前一个升采样训练单元输出的特征增强后的标注样本图像作为输入),从而获得高分辨率的图像。
如图4所示,图4是本公开实施例中残差学习结构示意图,该残差学习结构可以被用于构成图2中的升采样训练单元,图4中包括M个残差学习(Residual Network,ResNet)模块,各个残差学习(Residual Network,ResNet)模块可以分别包括多个特征增强单元,如图5所示,图5是本公开实施例中的残差学习模块的结构示意图,图5中示出了一个残差学习(Residual Network,ResNet)模块可以由N个特征增强单元构成。
S103:从多个标注样本图像之中确定第二样本图像。
上述对第一样本图像进行多次特征增强处理,以分别得到多次特征增强处理后的多个标注样本图像之后,可以从多个标注样本图像之中确定第二样本图像。
举例而言,可以从多个标注样本图像之中采样得到一张或者多张的第二样本图像,从而可以基于第一样本图像和第二样本图像构成高分辨-低分辨率深度图像对,作为训练数据。
可选地,一些实施例中,从多个标注样本图像之中确定第二样本图像,可以是确定目标特征差异信息,并确定与目标特征差异信息匹配的标注特征差异信息,并将匹配的标注特征差异信息对应的标注样本图像作为第二样本图像,其中的目标特征差异信息,可以是根据实际的训练场景需求自适应配置的,从而可以有效地满足图像处理场景中的个性化训练需求,能够实现不同特征差异程度的图像处理效果。
举例而言,目标特征差异信息可以是2倍、4倍、6倍等,即训练需求可以配置为使得目标图像处理模型具有针对第一样本图像的图像特征增强2倍、4倍、6倍等。
举例而言,假设目标特征差异信息可以是2倍,则从多个标注样本图像之中确定图像特征差异信息是2倍的标注样本图像作为第二样本图像,后续可以辅助以图像特征差异信息是2倍作为标注特征差异信息来训练图像处理模型。
S104:获取第一样本图像和第二样本图像之间的标注特征差异信息。
其中,第二样本图像的分辨率是第一样本图像分辨率的2倍,则可以确定标注特征差异信息是2倍数的分辨率,根据实际的应用场景需求,也可以配置第二样本图像的分辨率是第一样本图像分辨率的4倍或者8倍,对此不做限制。
其中,可以将第一样本图像和第二样本图像构成高分辨-低分辨率深度图像对,作为训练数据。
其中,第一样本图像和第二样本图像相对应,表征在训练模型的过程中,采用第一样本图像和第二样本图像之间的特征差异信息作为监督信号,以对模型的训练效果进行监督,即,采用第一样本图像和第二样本图像之间的特征差异信息作为标注特征差异信息,该特征差异信息可以具体例如是第一样本图像和第二样本图像之间分辨率差异信息,对此不做限制。
本公开实施例中,第一样本图像可以是低分辨率的样本图像,则相应的,第二样本图像可以是相对高分辨率的样本图像,例如,第二样本图像的分辨率可以是第一样本图像的分辨率的2倍、4倍、6倍等,对此不做限制,从而可以基于不同的标注特征差异信息作为训练模型用的监督信号,使得训练得到的图像处理模型具有不同程度的图像特征增强能力。
本公开实例中,可以采用第一样本图像和第二样本图像之间的特征差异作为标注特征差异,以对整个模型训练过程作为监督,后续在训练过程中,可以对第一样本图像进行多次的特征增强处理(由图2中多个升采样训练单元来完成),从而得到具有不同特征差异程度(例如,4倍分辨率、6倍分辨率、8倍分辨率的第三样本图像),以辅助对模型的整体训练过程。
本公开实施例中,多次升采样训练单元处理之后的高倍数分辨率的图像还可以被添加至训练数据集之中,以反向采用降采样训练单元进行特征减弱处理,以得到低分辨率的图像作为训练数据集的扩充,下述的举例说明可以一并结合图2展开,对此不做限制。
S105:根据第一样本图像、多个标注样本图像,以及标注特征差异信息训练初始的图像处理模型,以得到目标图像处理模型。
上述在从多个标注样本图像之中确定第二样本图像,获取第一样本图像和第二样本图像之间的标注特征差异信息之后,可以根据第一样本图像、多个标注样本图像,以及标注特征差异信息训练初始的图像处理模型,以得到目标图像处理模型。
也即是说,本公开实施例中,支持采用高分辨-低分辨率深度图像对之间的标注特征差异信息作为监督信号,而后在训练模型的过程中直接是根据第一样本图像以及预先进行特征增强处理后的多个标注样本图像来训练初始的图像处理模型,以得到目标图像处理模型。
其中的初始的图像处理模型可以为人工智能中的任一种能够执行图像处理任务的模型,比如机器学习模型或者神经网络模型等,对此不做限制。
本实施例中,通过获取第一样本图像,并对第一样本图像进行多次特征增强处理,以分别得到多次特征增强处理后的多个标注样本图像,从多个标注样本图像之中确定第二样本图像,获取第一样本图像和第二样本图像之间的标注特征差异信息,以及根据第一样本图像、多个标注样本图像,以及标注特征差异信息训练初始的图像处理模型,以得到目标图像处理模型,由于是从增强处理后的多个标注样本图像之中采样得到第二样本图像,并且将第一样本图像和第二样本图像之间的特征差异信息作为标注特征差异信息,以作为整体模型训练用的参考监督信号,有效地提升了模型的收敛效率,降低了对真值图像的依赖程度,提升训练得到的图像处理模型针对图像特征的表达建模能力,提升图像处理模型的图像处理效果。
图6是根据本公开第二实施例的示意图。
如图6所示,该图像处理模型的训练方法,包括:
S601:获取第一样本图像。
S602:对第一样本图像进行多次特征增强处理,以分别得到多次特征增强处理后的多个标注样本图像。
S601-S602的描述说明可以参见上述实施例,在此不再赘述。
本公开实施例在对第一样本图像进行特征增强处理,以得到标注样本图像,可以是对第一样本图像进行多次的特征增强处理,以得到多个标注样本图像,从而可以得到具有不同特征差异程度的多个标注样本图像,能够有效辅助训练得到的图像处理模型针对图像特征的不同程度的增强处理效果,并且不需要预先准备具有不同特征差异程度的样本图像,降低了对多样化图像特征的样本图像的依赖程度,提升了模型训练的便捷性,并且同时保障训练效果。
举例而言,可以将第一样本图像输入至上述图2中的第一个升采样训练单元中,并采用升采样训练单元对第一样本图像进行特征增强处理,以得到第一个标注样本图像,而后,将第一个标注样本图像作为第二个升采样训练单元的输入,得到进一步特征增强处理后的标注样本图像。
当然,也可以采用其它任意可能的方式对第一样本图像进行特征增强处理,以得到多个标注样本图像,例如图像合成的方式等,对此不做限制。
可选地,一些实施例中,对第一样本图像进行多次特征增强处理,以分别得到多次特征增强处理后的多个标注样本图像,包括:对第一样本图像进行多次通道级特征增强处理,以分别得到多次通道级特征增强处理后的多个标注样本图像;对第一样本图像进行多次像素级特征增强处理,以分别得到多次像素级特征增强处理后的多个标注样本图像。
也即是说,本公开实施例中支持对第一样本图像进行通道级特征增强处理,或者,支持对第一样本图像进行像素级特征增强处理,或者,采用通道级特征增强处理结合像素级特征增强处理方式处理第一样本图像,对此不做限制。
其中,对第一样本图像进行像素级特征增强处理的方式可以例如,获取第一样本图像中各个像素点的像素特征(例如深度特征、分辨率特征),而后,对像素特征(例如深度特征、分辨率特征)进行相应的增强处理,对此不做限制。
通过对第一样本图像进行通道级特征增强处理,以得到标注样本图像,和/或对第一样本图像进行像素级特征增强处理,以得到标注样本图像,从而能够有效地提升特征增强处理的灵活性,提升特征增强处理效果,从而辅助提升整体图像处理方法的适用性,辅助扩展图像处理方法的应用场景。
可选地,一些实施例中,对第一样本图像进行多次通道级特征增强处理,可以是确定与第一样本图像对应的第一图像特征;对第一图像特征进行卷积操作,以得到卷积图像特征,并对第一图像特征进行特征重组操作,以得到重组图像特征;融合卷积图像特征和重组图像特征,以得到融合图像特征;采用柔性最大值传输函数softmax处理融合图像特征,以得到参考描述信息;以及根据参考描述信息处理第一图像特征,以得到增强图像特征,其中,增强图像特征被用于处理第一样本图像以得到对应的第三样本图像,从而能够实现准确地、便捷地对输入的图像进行特征增强处理,较大程度地提升图像特征增强处理的效果,有效地辅助提升训练得到的图像处理模型针对图像特征的增强处理能力。
举例而言,如图7所示,图7是本公开实施例中通道级特征增强处理流程示意图,给定第一图像特征F(c×w×h)作为输入,通道级特征增强处理:可以首先通过卷积+特征重组reshape操作获得两种特征:卷积图像特征Qc(c×(h*w))和重组图像特征Hc((h*w)×c),而后,将前述两种特征采用矩阵乘法操作进行融合,获得融合图像特征:矩阵Mc(c×c),并采用柔性最大值传输函数softmax处理融合图像特征:矩阵Mc(c×c),以得到参考描述信息:权重Mc’(c×c)。另外,通过对第一图像特征F(c×w×h)进行卷积操作获得新的特征Fc’(c×h×w),该新的特征Fc’(c×h×w)可以是与前述的卷积图像特征的形式相一致,再通过Mc’和Fc’进行矩阵乘法操作,以根据参考描述信息处理第一图像特征,以得到增强图像特征Fh(c×h×w),而后,还可以通过对增强后的特征Fh与Fc’进行像素级加法获得通道增强后的特征Fc,最终的输入特征F=Fc,a为可学习的参数,对增强后的特征Fh与Fc’进行像素级加法的步骤,可以视为将增强图像特征Fh(c×h×w)对第一样本图像中相应的图像特征进行增强,从而形成标注样本图像的过程。
当然,也可以采用其它任意可能的方式实现对第一样本图像进行通道级特征增强处理,比如建模的方式、数学运算的方式、工程学方式等等,对此不做限制。
举例而言,如图8所示,图8是本公开实施例中像素级特征增强处理流程示意图,给定第一图像特征F(c×w×h)作为输入,像素级特征增强:可以首先通过卷积+特征重组reshape操作获得两种特征Qp(c×(h*w))和Hp((h*w)×c),通过针对前述两种特征采用矩阵乘法操作进行融合,获得融合图像特征:Mp((h*w)×(h*w)),并对Mp((h*w)×(h*w))进行柔性最大值传输函数softmax获得新的权重Mp’((h*w)×(h*w))。另外,还可以通过对第一图像特征F进行卷积操作获得新的特征Fp’(c×h×w),以及通过Mp’和Fp’进行矩阵乘法操作,以获得增强图像特征Fh(c×h×w),通过对增强图像特征Fh与Fp’进行像素级加法,获得像素级特征增强后的特征Fp,该像素级特征增强后的特征Fp被用于处理第一样本图像以形成标注样本图像。
S603:从多个标注样本图像之中确定第二样本图像。
S603的描述说明可以参见上述实施例,在此不再赘述。
S604:对多个标注样本图像分别进行多次特征削弱处理,以得到多个第三样本图像。
如图9所示,图9是本公开实施例中以非线性方式构建样本图像的示意图,标注样本图像是高分辨率的图像,可以对标注样本图像进行至少一次的卷积操作处理,以得到卷积操作特征,而后,根据卷积操作特征对标注样本图像进行图像重建,从而得到低分辨率图像作为第三样本图像,图9中所示的非线性方式构建样本图像的处理逻辑可以配置在图像处理模型的训练装置中的降采样训练单元中,经由该降采样训练单元,以高分辨率的图像作为输入,获得低分辨率的图像,包括卷积操作和图像重建操作。
上述对多个标注样本图像分别进行多次特征削弱处理,以得到多个第三样本图像可以被用于对训练用的第一样本图像进行灵活地扩充。
S605:根据多个第三样本图像对已获取第一样本图像进行补充,以得到多个第一样本图像。
举例而言,可以对多个标注样本图像分别进行多次特征削弱处理,以得到多个第三样本图像,而后,可以直接将多个第三样本图像作为第一样本图像,以对训练用的第一样本图像进行灵活地扩充,从而有效地保障了样本多样性,能够较大程度地降低了对真值图像的训练需求,从而有效地降低了图像处理模型的训练成本,实现了训练资源消耗和训练效果的有效平衡。
S606:将第一样本图像和多个标注样本图像输入至初始的图像处理模型之中,以得到图像处理模型输出的多个预测特征差异信息,多个预测特征差异信息,是预测得到的第一样本图像分别与多个标注样本图像之间的图像特征差异信息。
也即是说,本公开实施例中的初始的图像处理模型能够分别对第一样本图像和多个标注样本图像之间的特征差异信息进行预测(特征差异信息例如分辨率之间的倍数差异),而后将预测得到的特征差异信息作为预测差异信息,该预测差异信息可以被用于后续辅助对输入的样本图像进行图像特征增强处理。
S607:确定多个预测特征差异信息分别与标注特征差异信息之间对应的多个损失值。
本公开实施例中,可以针对初始的图像处理模型预先配置损失函数,在训练初始的图像处理模型的过程中,将第一样本图像和标注样本图像分别对应的图像特征(例如,分辨率),以及标注特征差异信息、预测特征差异信息作为损失函数的输入参数,并确定损失函数的输出值作为与该预测特征差异信息对应的损失值,并依据相应的方式获取与每个预测特征差异信息均对应的损失值,而后,将损失值与设定的损失阈值进行比对,以确定是否满足收敛时机,对此不做限制。
当然,也可以采用其它任意可能的方式来判定图像处理模型的收敛时机,比如针对每个训练单元均配置不同的参考监督信号来判定收敛时机,或,当训练次数满足一定的次数,则直接确定图像处理模型收敛等,对此不做限制。
S608:如果多个损失值均小于损失阈值,则将训练得到的图像处理模型作为目标图像处理模型。
也即是说,本公开实施例中,为了实现较大程度提升训练得到的图像处理模型针对图像特征的表达建模能力,实现能够准确地、及时地判定出图像处理模型的收敛时机,使得训练得到的图像处理模型针对图像特征具备了不同程度的增强处理效果,还可以在获取与每个预测特征差异信息均对应的损失值之后,将各个损失值分别与预先设定的损失阈值进行比对,如果多个损失值均小于损失阈值,则将训练得到的图像处理模型作为目标图像处理模型。
本实施例中,通过获取第一样本图像,并对第一样本图像进行多次特征增强处理,以分别得到多次特征增强处理后的多个标注样本图像,从多个标注样本图像之中确定第二样本图像,获取第一样本图像和第二样本图像之间的标注特征差异信息,以及根据第一样本图像、多个标注样本图像,以及标注特征差异信息训练初始的图像处理模型,以得到目标图像处理模型,由于是从增强处理后的多个标注样本图像之中采样得到第二样本图像,并且将第一样本图像和第二样本图像之间的特征差异信息作为标注特征差异信息,以作为整体模型训练用的参考监督信号,有效地提升了模型的收敛效率,降低了对真值图像的依赖程度,提升训练得到的图像处理模型针对图像特征的表达建模能力,提升图像处理模型的图像处理效果。通过对多个标注样本图像分别进行多次特征削弱处理,以得到多个第三样本图像,根据多个第三样本图像对已获取第一样本图像进行补充,以得到多个第一样本图像,从而有效地保障了样本多样性,能够较大程度地降低了对真值图像的训练需求,从而有效地降低了图像处理模型的训练成本,实现了训练资源消耗和训练效果的有效平衡。实现较大程度提升训练得到的图像处理模型针对图像特征的表达建模能力,实现能够准确地、及时地判定出图像处理模型的收敛时机,使得训练得到的图像处理模型针对图像特征具备了不同程度的增强处理效果。
图10是根据本公开第三实施例的示意图。
如图10所示,该图像处理方法,包括:
S1001:获取待处理图像,待处理图像具有对应的待处理图像特征。
其中,当前待对其进行处理的图像,可以被称为待处理图像。
该待处理图像的数量可以是一张或者多张,该待处理图像也可以是从多个视频帧之中提取出的部分帧视频图像,对此不做限制。
S1002:将待处理图像输入至如上述的图像处理模型的训练方法训练得到的目标图像处理模型之中,以得到目标图像处理模型输出的目标特征差异信息。
上述在获取待处理图像之后,可以将待处理图像输入至如上述图像处理模型的训练方法训练得到的目标图像处理模型之中,以得到目标图像处理模型输出的目标特征差异信息,该目标特征差异信息,可以用于表征需求的图像特征与待处理图像特征之间的特征差异情况,需求的图像特征可以例如为需求的分辨率,而待处理图像特征,可以例如为待处理图像对应的分辨率,从而目标特征差异信息,可以用于描述需求的分辨率和待处理图像对应的分辨率之间的分辨率差异情况。
S1003:根据目标特征差异信息对待处理图像特征进行特征增强处理,以得到目标图像特征,目标图像特征被融合至待处理图像之中以得到目标图像。
上述在获得目标图像处理模型输出的目标特征差异信息之后,可以采用目标特征差异信息对待处理图像特征进行特征增强处理,并将处理后的图像特征作为目标图像特征,而后,可以将目标图像特征融合至待处理图像,以形成目标图像。
举例而言,如图11所示,图11是本公开实施例中图像处理方法的流程示意图,假设待处理图像是低分辨率图像,将低分辨率图像输入至目标图像处理模型之中,该目标图像处理模型之中可以是采用上述的训练方法结合训练数据集训练得到的,该目标图像处理模型可以具体是非线性结构,支持采用上述的图像特征增强方法来对低分辨率图像进行图像处理,以输出高分辨率图像。
本实施例中,通过获取待处理图像,待处理图像具有对应的待处理图像特征,并将待处理图像输入至如上述的图像处理模型的训练方法训练得到的目标图像处理模型之中,以得到目标图像处理模型输出的目标特征差异信息,以及根据目标特征差异信息对待处理图像特征进行特征增强处理,以得到目标图像特征,目标图像特征被融合至待处理图像之中以得到目标图像,从而在采用训练得到的目标图像处理模型处理待处理图像时,能够表达建模出更为准确的特征差异信息,有效地辅助提升针对待处理图像的特征增强效果,从而有效地提升图像处理模型的图像处理效果。
图12是根据本公开第四实施例的示意图。
如图12所示,该图像处理模型的训练装置120,包括:
第一获取模块1201,用于获取第一样本图像;
第一处理模块1202,用于对所述第一样本图像进行多次特征增强处理,以分别得到所述多次特征增强处理后的多个标注样本图像;
确定模块1203,用于从所述多个标注样本图像之中确定第二样本图像;
第二获取模块1204,用于获取所述第一样本图像和所述第二样本图像之间的标注特征差异信息;以及
训练模块1205,用于根据所述第一样本图像、所述多个标注样本图像,以及所述标注特征差异信息训练初始的图像处理模型,以得到目标图像处理模型。
在本公开的一些实施例中,如图13所示,图13是根据本公开第五实施例的示意图,该图像处理模型的训练装置130,包括:第一获取模块1301、第一处理模块1302、确定模块1303、第二获取模块1304、训练模块1305,其中,确定模块1303,具体用于:
确定目标特征差异信息;
确定与所述目标特征差异信息匹配的标注特征差异信息,并将所述匹配的标注特征差异信息对应的标注样本图像作为所述第二样本图像。
在本公开的一些实施例中,图像处理模型的训练装置130,还包括:
第二处理模块1306,用于对所述多个标注样本图像分别进行多次特征削弱处理,以得到多个第三样本图像;
补充模块1307,用于根据所述多个第三样本图像对已获取所述第一样本图像进行补充,以得到多个所述第一样本图像。
在本公开的一些实施例中,其中,第一处理模块1302,包括:
第一处理子模块13021,用于对所述第一样本图像进行多次通道级特征增强处理,以分别得到所述多次通道级特征增强处理后的多个标注样本图像;
第二处理子模块13022,用于对所述第一样本图像进行多次像素级特征增强处理,以分别得到所述多次像素级特征增强处理后的多个标注样本图像。
在本公开的一些实施例中,其中,第一处理模块1302,具体用于:
对第一样本图像进行当次的特征增强处理,以得到当次的标注样本图像;
在进行下次的特征增强处理时,对所述当次的标注样本图像进行特征增强处理,以得到下次的标注样本图像,直至执行所述特征增强处理的次数满足设定次数。
在本公开的一些实施例中,其中,第一处理子模块13021,具体用于:
确定与所述第一样本图像对应的第一图像特征;
对所述第一图像特征进行卷积操作,以得到卷积图像特征,并对所述第一图像特征进行特征重组操作,以得到重组图像特征;
融合所述卷积图像特征和所述重组图像特征,以得到融合图像特征;
采用柔性最大值传输函数处理所述融合图像特征,以得到参考描述信息;以及
根据所述参考描述信息处理所述第一图像特征,以得到增强图像特征,其中,所述增强图像特征被用于处理所述第一样本图像以得到对应的第三样本图像。
在本公开的一些实施例中,其中,训练模块1305,具体用于:
将所述第一样本图像和所述多个标注样本图像输入至所述初始的图像处理模型之中,以得到所述图像处理模型输出的多个预测特征差异信息,所述多个预测特征差异信息,是预测得到的所述第一样本图像分别与所述多个标注样本图像之间的图像特征差异信息;
确定所述多个预测特征差异信息分别与所述标注特征差异信息之间对应的多个损失值;
如果所述多个损失值均小于损失阈值,则将训练得到的所述图像处理模型作为所述目标图像处理模型。
可以理解的是,本实施例附图13中的图像处理模型的训练装置130与上述实施例中的图像处理模型的训练装置120,第一获取模块1301与上述实施例中的第一获取模块1201,第一处理模块1302与上述实施例中的第一处理模块1202,确定模块1303与上述实施例中的确定模块1203,第二获取模块1304与上述实施例中的第二获取模块1204,训练模块1305与上述实施例中的训练模块1205可以具有相同的功能和结构。
需要说明的是,前述对图像处理模型的训练方法的解释说明也适用于本实施例的图像处理模型的训练装置。
本实施例中,通过获取第一样本图像,并对第一样本图像进行多次特征增强处理,以分别得到多次特征增强处理后的多个标注样本图像,从多个标注样本图像之中确定第二样本图像,获取第一样本图像和第二样本图像之间的标注特征差异信息,以及根据第一样本图像、多个标注样本图像,以及标注特征差异信息训练初始的图像处理模型,以得到目标图像处理模型,由于是从增强处理后的多个标注样本图像之中采样得到第二样本图像,并且将第一样本图像和第二样本图像之间的特征差异信息作为标注特征差异信息,以作为整体模型训练用的参考监督信号,有效地提升了模型的收敛效率,降低了对真值图像的依赖程度,提升训练得到的图像处理模型针对图像特征的表达建模能力,提升图像处理模型的图像处理效果。
图14是根据本公开第六实施例的示意图。
如图14所示,该图像处理装置140,包括:
第三获取模块1401,用于获取待处理图像,所述待处理图像具有对应的待处理图像特征;
输入模块1402,用于将所述待处理图像输入至如上述的图像处理模型的训练装置训练得到的目标图像处理模型之中,以得到所述目标图像处理模型输出的目标特征差异信息;以及
第三处理模块1403,用于根据所述目标特征差异信息对所述待处理图像特征进行特征增强处理,以得到目标图像特征,所述目标图像特征被融合至所述待处理图像之中以得到目标图像。
需要说明的是,前述对图像处理方法的解释说明也适用于本实施例的图像处理装置。
本实施例中,通过获取待处理图像,待处理图像具有对应的待处理图像特征,并将待处理图像输入至如上述的图像处理模型的训练方法训练得到的目标图像处理模型之中,以得到目标图像处理模型输出的目标特征差异信息,以及根据目标特征差异信息对待处理图像特征进行特征增强处理,以得到目标图像特征,目标图像特征被融合至待处理图像之中以得到目标图像,从而在采用训练得到的目标图像处理模型处理待处理图像时,能够表达建模出更为准确的特征差异信息,有效地辅助提升针对待处理图像的特征增强效果,从而有效地提升图像处理模型的图像处理效果。
根据本公开的实施例,本公开还提供了一种电子设备、一种可读存储介质和一种计算机程序产品。
图15示出了可以用来实施本公开的实施例的图像处理模型的训练方法的示例电子设备的示意性框图。电子设备旨在表示各种形式的数字计算机,诸如,膝上型计算机、台式计算机、工作台、个人数字助理、服务器、刀片式服务器、大型计算机、和其它适合的计算机。电子设备还可以表示各种形式的移动装置,诸如,个人数字处理、蜂窝电话、智能电话、可穿戴设备和其它类似的计算装置。本文所示的部件、它们的连接和关系、以及它们的功能仅仅作为示例,并且不意在限制本文中描述的和/或者要求的本公开的实现。
如图15所示,设备1500包括计算单元1501,其可以根据存储在只读存储器(ROM)1502中的计算机程序或者从存储单元1508加载到随机访问存储器(RAM)1503中的计算机程序,来执行各种适当的动作和处理。在RAM 1503中,还可存储设备1500操作所需的各种程序和数据。计算单元1501、ROM 1502以及RAM 1503通过总线1504彼此相连。输入/输出(I/O)接口1505也连接至总线1504。
设备1500中的多个部件连接至I/O接口1505,包括:输入单元1506,例如键盘、鼠标等;输出单元1507,例如各种类型的显示器、扬声器等;存储单元1508,例如磁盘、光盘等;以及通信单元1509,例如网卡、调制解调器、无线通信收发机等。通信单元1509允许设备1500通过诸如因特网的计算机网络和/或各种电信网络与其它设备交换信息/数据。
计算单元1501可以是各种具有处理和计算能力的通用和/或专用处理组件。计算单元1501的一些示例包括但不限于中央处理单元(CPU)、图形处理单元(GPU)、各种专用的人工智能(AI)计算芯片、各种运行机器学习模型算法的计算单元、数字信号处理器(DSP)、以及任何适当的处理器、控制器、微控制器等。计算单元1501执行上文所描述的各个方法和处理,例如图像处理模型的训练方法,或者图像处理方法。例如,在一些实施例中,图像处理模型的训练方法,或者图像处理方法可被实现为计算机软件程序,其被有形地包含于机器可读介质,例如存储单元1508。在一些实施例中,计算机程序的部分或者全部可以经由ROM1502和/或通信单元1509而被载入和/或安装到设备1500上。当计算机程序加载到RAM1503并由计算单元1501执行时,可以执行上文描述的图像处理模型的训练方法,或者图像处理方法的一个或多个步骤。备选地,在其它实施例中,计算单元1501可以通过其它任何适当的方式(例如,借助于固件)而被配置为执行图像处理模型的训练方法,或者图像处理方法。
本文中以上描述的系统和技术的各种实施方式可以在数字电子电路系统、集成电路系统、场可编程门阵列(FPGA)、专用集成电路(ASIC)、专用标准产品(ASSP)、芯片上系统的系统(SOC)、负载可编程逻辑设备(CPLD)、计算机硬件、固件、软件、和/或它们的组合中实现。这些各种实施方式可以包括:实施在一个或者多个计算机程序中,该一个或者多个计算机程序可在包括至少一个可编程处理器的可编程系统上执行和/或解释,该可编程处理器可以是专用或者通用可编程处理器,可以从存储系统、至少一个输入装置、和至少一个输出装置接收数据和指令,并且将数据和指令传输至该存储系统、该至少一个输入装置、和该至少一个输出装置。
用于实施本公开的方法的程序代码可以采用一个或多个编程语言的任何组合来编写。这些程序代码可以提供给通用计算机、专用计算机或其它可编程数据处理装置的处理器或控制器,使得程序代码当由处理器或控制器执行时使流程图和/或框图中所规定的功能/操作被实施。程序代码可以完全在机器上执行、部分地在机器上执行,作为独立软件包部分地在机器上执行且部分地在远程机器上执行或完全在远程机器或服务器上执行。
在本公开的上下文中,机器可读介质可以是有形的介质,其可以包含或存储以供指令执行系统、装置或设备使用或与指令执行系统、装置或设备结合地使用的程序。机器可读介质可以是机器可读信号介质或机器可读储存介质。机器可读介质可以包括但不限于电子的、磁性的、光学的、电磁的、红外的、或半导体系统、装置或设备,或者上述内容的任何合适组合。机器可读存储介质的更具体示例会包括基于一个或多个线的电气连接、便携式计算机盘、硬盘、随机存取存储器(RAM)、只读存储器(ROM)、可擦除可编程只读存储器(EPROM或快闪存储器)、光纤、便捷式紧凑盘只读存储器(CD-ROM)、光学储存设备、磁储存设备、或上述内容的任何合适组合。
为了提供与用户的交互,可以在计算机上实施此处描述的系统和技术,该计算机具有:用于向用户显示信息的显示装置(例如,CRT(阴极射线管)或者LCD(液晶显示器)监视器);以及键盘和指向装置(例如,鼠标或者轨迹球),用户可以通过该键盘和该指向装置来将输入提供给计算机。其它种类的装置还可以用于提供与用户的交互;例如,提供给用户的反馈可以是任何形式的传感反馈(例如,视觉反馈、听觉反馈、或者触觉反馈);并且可以用任何形式(包括声输入、语音输入或者、触觉输入)来接收来自用户的输入。
可以将此处描述的系统和技术实施在包括后台部件的计算系统(例如,作为数据服务器)、或者包括中间件部件的计算系统(例如,应用服务器)、或者包括前端部件的计算系统(例如,具有图形用户界面或者网络浏览器的用户计算机,用户可以通过该图形用户界面或者该网络浏览器来与此处描述的系统和技术的实施方式交互)、或者包括这种后台部件、中间件部件、或者前端部件的任何组合的计算系统中。可以通过任何形式或者介质的数字数据通信(例如,通信网络)来将系统的部件相互连接。通信网络的示例包括:局域网(LAN)、广域网(WAN)、互联网及区块链网络。
计算机系统可以包括客户端和服务器。客户端和服务器一般远离彼此并且通常通过通信网络进行交互。通过在相应的计算机上运行并且彼此具有客户端-服务器关系的计算机程序来产生客户端和服务器的关系。服务器可以是云服务器,又称为云计算服务器或云主机,是云计算服务体系中的一项主机产品,以解决了传统物理主机与VPS服务("Virtual Private Server",或简称"VPS")中,存在的管理难度大,业务扩展性弱的缺陷。服务器也可以为分布式系统的服务器,或者是结合了区块链的服务器。
应该理解,可以使用上面所示的各种形式的流程,重新排序、增加或删除步骤。例如,本发公开中记载的各步骤可以并行地执行也可以顺序地执行也可以不同的次序执行,只要能够实现本公开公开的技术方案所期望的结果,本文在此不进行限制。
上述具体实施方式,并不构成对本公开保护范围的限制。本领域技术人员应该明白的是,根据设计要求和其它因素,可以进行各种修改、组合、子组合和替代。任何在本公开的精神和原则之内所作的修改、等同替换和改进等,均应包含在本公开保护范围之内。

Claims (17)

1.一种图像处理模型的训练方法,包括:
获取第一样本图像;
对所述第一样本图像进行多次特征增强处理,以分别得到所述多次特征增强处理后的多个标注样本图像,其中,不同所述标注样本图像对应的标注特征差异信息不相同,所述标注特征差异信息指示所述标注样本图像和所述第一样本图像之间的特征差异程度;
确定目标特征差异信息;
确定与所述目标特征差异信息匹配的标注特征差异信息,并将所述匹配的标注特征差异信息对应的所述标注样本图像作为第二样本图像;
获取所述第一样本图像和所述第二样本图像之间的标注特征差异信息;以及
根据所述第一样本图像、所述多个标注样本图像,以及所述标注特征差异信息训练初始的图像处理模型,以得到目标图像处理模型,其中,在所述初始的图像处理模型的训练过程中,所述第一样本图像和所述多个标注样本图像用于确定多个预测特征差异信息,所述多个预测特征差异信息,是预测得到的所述第一样本图像分别与所述多个标注样本图像之间的图像特征差异信息。
2.根据权利要求1所述的方法,在所述根据所述第一样本图像、所述多个标注样本图像,以及所述标注特征差异信息训练初始的图像处理模型,以得到目标图像处理模型之前,还包括:
对所述多个标注样本图像分别进行多次特征削弱处理,以得到多个第三样本图像;
根据所述多个第三样本图像对已获取所述第一样本图像进行补充,以得到多个所述第一样本图像。
3.根据权利要求1所述的方法,其中,所述对所述第一样本图像进行多次特征增强处理,以分别得到所述多次特征增强处理后的多个标注样本图像,包括:
对所述第一样本图像进行多次通道级特征增强处理,以分别得到所述多次通道级特征增强处理后的多个标注样本图像;
对所述第一样本图像进行多次像素级特征增强处理,以分别得到所述多次像素级特征增强处理后的多个标注样本图像。
4.根据权利要求1所述的方法,其中,所述对所述第一样本图像进行多次特征增强处理,以分别得到所述多次特征增强处理后的多个标注样本图像,包括:
对第一样本图像进行当次的特征增强处理,以得到当次的标注样本图像;
在进行下次的特征增强处理时,对所述当次的标注样本图像进行特征增强处理,以得到下次的标注样本图像,直至执行所述特征增强处理的次数满足设定次数。
5.根据权利要求3所述的方法,其中,所述对所述第一样本图像进行多次通道级特征增强处理,包括:
确定与所述第一样本图像对应的第一图像特征;
对所述第一图像特征进行卷积操作,以得到卷积图像特征,并对所述第一图像特征进行特征重组操作,以得到重组图像特征;
融合所述卷积图像特征和所述重组图像特征,以得到融合图像特征;
采用柔性最大值传输函数处理所述融合图像特征,以得到参考描述信息;以及
根据所述参考描述信息处理所述第一图像特征,以得到增强图像特征,其中,所述增强图像特征被用于处理所述第一样本图像以得到对应的第三样本图像。
6.根据权利要求1所述的方法,其中,所述根据所述第一样本图像、所述多个标注样本图像,以及所述标注特征差异信息训练初始的图像处理模型,以得到目标图像处理模型,包括:
将所述第一样本图像和所述多个标注样本图像输入至所述初始的图像处理模型之中,以得到所述图像处理模型输出的多个预测特征差异信息;
确定所述多个预测特征差异信息分别与所述标注特征差异信息之间对应的多个损失值;
如果所述多个损失值均小于损失阈值,则将训练得到的所述图像处理模型作为所述目标图像处理模型。
7.一种图像处理方法,包括:
获取待处理图像,所述待处理图像具有对应的待处理图像特征;
将所述待处理图像输入至如上述权利要求1-6任一项所述的图像处理模型的训练方法训练得到的目标图像处理模型之中,以得到所述目标图像处理模型输出的目标特征差异信息;以及
根据所述目标特征差异信息对所述待处理图像特征进行特征增强处理,以得到目标图像特征,所述目标图像特征被融合至所述待处理图像之中以得到目标图像。
8.一种图像处理模型的训练装置,包括:
第一获取模块,用于获取第一样本图像;
第一处理模块,用于对所述第一样本图像进行多次特征增强处理,以分别得到所述多次特征增强处理后的多个标注样本图像,其中,不同所述标注样本图像对应的标注特征差异信息不相同,所述标注特征差异信息指示所述标注样本图像和所述第一样本图像之间的特征差异程度;
确定模块,用于确定目标特征差异信息;确定与所述目标特征差异信息匹配的标注特征差异信息,并将所述匹配的标注特征差异信息对应的所述标注样本图像作为第二样本图像;
第二获取模块,用于获取所述第一样本图像和所述第二样本图像之间的标注特征差异信息;以及
训练模块,用于根据所述第一样本图像、所述多个标注样本图像,以及所述标注特征差异信息训练初始的图像处理模型,以得到目标图像处理模型,其中,在所述初始的图像处理模型的训练过程中,所述第一样本图像和所述多个标注样本图像用于确定多个预测特征差异信息,所述多个预测特征差异信息,是预测得到的所述第一样本图像分别与所述多个标注样本图像之间的图像特征差异信息。
9.根据权利要求8所述的装置,不同所述标注样本图像对应的标注特征差异信息不相同,所述标注特征差异信息指示所述标注样本图像和所述第一样本图像之间的特征差异程度,
其中,所述确定模块,具体用于:
确定目标特征差异信息;
确定与所述目标特征差异信息匹配的标注特征差异信息,并将所述匹配的标注特征差异信息对应的标注样本图像作为所述第二样本图像。
10.根据权利要求8所述的装置,还包括:
第二处理模块,用于对所述多个标注样本图像分别进行多次特征削弱处理,以得到多个第三样本图像;
补充模块,用于根据所述多个第三样本图像对已获取所述第一样本图像进行补充,以得到多个所述第一样本图像。
11.根据权利要求8所述的装置,其中,所述第一处理模块,包括:
第一处理子模块,用于对所述第一样本图像进行多次通道级特征增强处理,以分别得到所述多次通道级特征增强处理后的多个标注样本图像;
第二处理子模块,用于对所述第一样本图像进行多次像素级特征增强处理,以分别得到所述多次像素级特征增强处理后的多个标注样本图像。
12.根据权利要求8所述的装置,其中,所述第一处理模块,具体用于:
对第一样本图像进行当次的特征增强处理,以得到当次的标注样本图像;
在进行下次的特征增强处理时,对所述当次的标注样本图像进行特征增强处理,以得到下次的标注样本图像,直至执行所述特征增强处理的次数满足设定次数。
13.根据权利要求11所述的装置,其中,所述第一处理子模块,具体用于:
确定与所述第一样本图像对应的第一图像特征;
对所述第一图像特征进行卷积操作,以得到卷积图像特征,并对所述第一图像特征进行特征重组操作,以得到重组图像特征;
融合所述卷积图像特征和所述重组图像特征,以得到融合图像特征;
采用柔性最大值传输函数处理所述融合图像特征,以得到参考描述信息;以及
根据所述参考描述信息处理所述第一图像特征,以得到增强图像特征,其中,所述增强图像特征被用于处理所述第一样本图像以得到对应的第三样本图像。
14.根据权利要求8所述的装置,其中,所述训练模块,具体用于:
将所述第一样本图像和所述多个标注样本图像输入至所述初始的图像处理模型之中,以得到所述图像处理模型输出的多个预测特征差异信息;
确定所述多个预测特征差异信息分别与所述标注特征差异信息之间对应的多个损失值;
如果所述多个损失值均小于损失阈值,则将训练得到的所述图像处理模型作为所述目标图像处理模型。
15.一种图像处理装置,包括:
第三获取模块,用于获取待处理图像,所述待处理图像具有对应的待处理图像特征;
输入模块,用于将所述待处理图像输入至如上述权利要求8-14任一项所述的图像处理模型的训练装置训练得到的目标图像处理模型之中,以得到所述目标图像处理模型输出的目标特征差异信息;以及
第三处理模块,用于根据所述目标特征差异信息对所述待处理图像特征进行特征增强处理,以得到目标图像特征,所述目标图像特征被融合至所述待处理图像之中以得到目标图像。
16.一种电子设备,包括:
至少一个处理器;以及
与所述至少一个处理器通信连接的存储器;其中,
所述存储器存储有可被所述至少一个处理器执行的指令,所述指令被所述至少一个处理器执行,以使所述至少一个处理器能够执行权利要求1-6中任一项所述的方法,或者执行权利要求7所述的方法。
17.一种存储有计算机指令的非瞬时计算机可读存储介质,其中,所述计算机指令用于使所述计算机执行根据权利要求1-6中任一项所述的方法,或者执行权利要求7所述的方法。
CN202110732721.1A 2021-06-30 2021-06-30 图像处理模型的训练方法、装置、电子设备及存储介质 Active CN113554550B (zh)

Priority Applications (1)

Application Number Priority Date Filing Date Title
CN202110732721.1A CN113554550B (zh) 2021-06-30 2021-06-30 图像处理模型的训练方法、装置、电子设备及存储介质

Applications Claiming Priority (1)

Application Number Priority Date Filing Date Title
CN202110732721.1A CN113554550B (zh) 2021-06-30 2021-06-30 图像处理模型的训练方法、装置、电子设备及存储介质

Publications (2)

Publication Number Publication Date
CN113554550A CN113554550A (zh) 2021-10-26
CN113554550B true CN113554550B (zh) 2023-08-04

Family

ID=78131106

Family Applications (1)

Application Number Title Priority Date Filing Date
CN202110732721.1A Active CN113554550B (zh) 2021-06-30 2021-06-30 图像处理模型的训练方法、装置、电子设备及存储介质

Country Status (1)

Country Link
CN (1) CN113554550B (zh)

Families Citing this family (1)

* Cited by examiner, † Cited by third party
Publication number Priority date Publication date Assignee Title
CN116664988B (zh) * 2023-07-24 2023-11-21 广立微(上海)技术有限公司 图片自动标注方法、装置、计算机设备和存储介质

Citations (4)

* Cited by examiner, † Cited by third party
Publication number Priority date Publication date Assignee Title
CN111626350A (zh) * 2020-05-25 2020-09-04 腾讯科技(深圳)有限公司 目标检测模型训练方法、目标检测方法及装置
CN111738262A (zh) * 2020-08-21 2020-10-02 北京易真学思教育科技有限公司 目标检测模型训练、检测方法、装置、设备及存储介质
WO2020233200A1 (zh) * 2019-05-17 2020-11-26 北京字节跳动网络技术有限公司 用于训练模型的方法和装置、用于预测信息的方法和装置
CN112541482A (zh) * 2020-12-25 2021-03-23 北京百度网讯科技有限公司 深度信息补全模型训练方法、装置、设备以及存储介质

Family Cites Families (1)

* Cited by examiner, † Cited by third party
Publication number Priority date Publication date Assignee Title
CN110428378B (zh) * 2019-07-26 2022-02-08 北京小米移动软件有限公司 图像的处理方法、装置及存储介质

Patent Citations (4)

* Cited by examiner, † Cited by third party
Publication number Priority date Publication date Assignee Title
WO2020233200A1 (zh) * 2019-05-17 2020-11-26 北京字节跳动网络技术有限公司 用于训练模型的方法和装置、用于预测信息的方法和装置
CN111626350A (zh) * 2020-05-25 2020-09-04 腾讯科技(深圳)有限公司 目标检测模型训练方法、目标检测方法及装置
CN111738262A (zh) * 2020-08-21 2020-10-02 北京易真学思教育科技有限公司 目标检测模型训练、检测方法、装置、设备及存储介质
CN112541482A (zh) * 2020-12-25 2021-03-23 北京百度网讯科技有限公司 深度信息补全模型训练方法、装置、设备以及存储介质

Also Published As

Publication number Publication date
CN113554550A (zh) 2021-10-26

Similar Documents

Publication Publication Date Title
CN113538235B (zh) 图像处理模型的训练方法、装置、电子设备及存储介质
CN113177472B (zh) 动态手势识别方法、装置、设备以及存储介质
CN113177451B (zh) 图像处理模型的训练方法、装置、电子设备及存储介质
CN113361363B (zh) 人脸图像识别模型的训练方法、装置、设备及存储介质
CN114187624B (zh) 图像生成方法、装置、电子设备及存储介质
CN113361572B (zh) 图像处理模型的训练方法、装置、电子设备以及存储介质
WO2020093724A1 (zh) 生成信息的方法和装置
CN113379877B (zh) 人脸视频生成方法、装置、电子设备及存储介质
CN114693934B (zh) 语义分割模型的训练方法、视频语义分割方法及装置
CN113627536B (zh) 模型训练、视频分类方法,装置,设备以及存储介质
CN115456167B (zh) 轻量级模型训练方法、图像处理方法、装置及电子设备
CN116309983B (zh) 虚拟人物模型的训练方法、生成方法、装置和电子设备
CN113592932A (zh) 深度补全网络的训练方法、装置、电子设备及存储介质
CN114913325B (zh) 语义分割方法、装置及计算机程序产品
CN112929689A (zh) 视频插帧方法、装置、设备以及存储介质
CN113554550B (zh) 图像处理模型的训练方法、装置、电子设备及存储介质
CN114882333B (zh) 数据处理模型的训练方法、装置、电子设备及存储介质
CN114078097A (zh) 图像去雾模型的获取方法、装置和电子设备
CN113657468A (zh) 预训练模型的生成方法、装置、电子设备和存储介质
CN116468112B (zh) 目标检测模型的训练方法、装置、电子设备和存储介质
CN113379750A (zh) 语义分割模型的半监督学习方法、相关装置及产品
CN115239889B (zh) 3d重建网络的训练方法、3d重建方法、装置、设备和介质
CN116402914A (zh) 用于确定风格化图像生成模型的方法、装置及产品
CN113361575B (zh) 模型训练方法、装置和电子设备
CN113344200B (zh) 用于训练可分离卷积网络的方法、路侧设备及云控平台

Legal Events

Date Code Title Description
PB01 Publication
PB01 Publication
SE01 Entry into force of request for substantive examination
SE01 Entry into force of request for substantive examination
GR01 Patent grant
GR01 Patent grant