CN114067007A - 图像处理方法与装置、神经网络训练的方法与装置 - Google Patents

图像处理方法与装置、神经网络训练的方法与装置 Download PDF

Info

Publication number
CN114067007A
CN114067007A CN202010754067.XA CN202010754067A CN114067007A CN 114067007 A CN114067007 A CN 114067007A CN 202010754067 A CN202010754067 A CN 202010754067A CN 114067007 A CN114067007 A CN 114067007A
Authority
CN
China
Prior art keywords
image
training
images
region
model
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.)
Pending
Application number
CN202010754067.XA
Other languages
English (en)
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.)
Peking University
Huawei Technologies Co Ltd
Original Assignee
Peking University
Huawei Technologies 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 Peking University, Huawei Technologies Co Ltd filed Critical Peking University
Priority to CN202010754067.XA priority Critical patent/CN114067007A/zh
Priority to PCT/CN2021/086836 priority patent/WO2022021938A1/zh
Publication of CN114067007A publication Critical patent/CN114067007A/zh
Pending legal-status Critical Current

Links

Images

Classifications

    • GPHYSICS
    • G06COMPUTING; CALCULATING OR COUNTING
    • G06TIMAGE DATA PROCESSING OR GENERATION, IN GENERAL
    • G06T9/00Image coding
    • G06T9/002Image coding using neural networks
    • GPHYSICS
    • G06COMPUTING; CALCULATING OR COUNTING
    • G06NCOMPUTING ARRANGEMENTS BASED ON SPECIFIC COMPUTATIONAL MODELS
    • G06N3/00Computing arrangements based on biological models
    • G06N3/02Neural networks
    • G06N3/04Architecture, e.g. interconnection topology
    • GPHYSICS
    • G06COMPUTING; CALCULATING OR COUNTING
    • G06NCOMPUTING ARRANGEMENTS BASED ON SPECIFIC COMPUTATIONAL MODELS
    • G06N3/00Computing arrangements based on biological models
    • G06N3/02Neural networks
    • G06N3/04Architecture, e.g. interconnection topology
    • G06N3/045Combinations of networks
    • GPHYSICS
    • G06COMPUTING; CALCULATING OR COUNTING
    • G06NCOMPUTING ARRANGEMENTS BASED ON SPECIFIC COMPUTATIONAL MODELS
    • G06N3/00Computing arrangements based on biological models
    • G06N3/02Neural networks
    • G06N3/08Learning methods
    • GPHYSICS
    • G06COMPUTING; CALCULATING OR COUNTING
    • G06NCOMPUTING ARRANGEMENTS BASED ON SPECIFIC COMPUTATIONAL MODELS
    • G06N3/00Computing arrangements based on biological models
    • G06N3/02Neural networks
    • G06N3/08Learning methods
    • G06N3/084Backpropagation, e.g. using gradient descent
    • GPHYSICS
    • G06COMPUTING; CALCULATING OR COUNTING
    • G06TIMAGE DATA PROCESSING OR GENERATION, IN GENERAL
    • G06T9/00Image coding
    • HELECTRICITY
    • H04ELECTRIC COMMUNICATION TECHNIQUE
    • H04NPICTORIAL COMMUNICATION, e.g. TELEVISION
    • H04N1/00Scanning, transmission or reproduction of documents or the like, e.g. facsimile transmission; Details thereof
    • H04N1/41Bandwidth or redundancy reduction
    • HELECTRICITY
    • H04ELECTRIC COMMUNICATION TECHNIQUE
    • H04NPICTORIAL COMMUNICATION, e.g. TELEVISION
    • H04N19/00Methods or arrangements for coding, decoding, compressing or decompressing digital video signals
    • H04N19/10Methods or arrangements for coding, decoding, compressing or decompressing digital video signals using adaptive coding
    • H04N19/134Methods or arrangements for coding, decoding, compressing or decompressing digital video signals using adaptive coding characterised by the element, parameter or criterion affecting or controlling the adaptive coding
    • H04N19/136Incoming video signal characteristics or properties
    • H04N19/14Coding unit complexity, e.g. amount of activity or edge presence estimation
    • HELECTRICITY
    • H04ELECTRIC COMMUNICATION TECHNIQUE
    • H04NPICTORIAL COMMUNICATION, e.g. TELEVISION
    • H04N19/00Methods or arrangements for coding, decoding, compressing or decompressing digital video signals
    • H04N19/42Methods or arrangements for coding, decoding, compressing or decompressing digital video signals characterised by implementation details or hardware specially adapted for video compression or decompression, e.g. dedicated software implementation

Landscapes

  • Engineering & Computer Science (AREA)
  • Theoretical Computer Science (AREA)
  • Physics & Mathematics (AREA)
  • Multimedia (AREA)
  • General Physics & Mathematics (AREA)
  • Evolutionary Computation (AREA)
  • Signal Processing (AREA)
  • Artificial Intelligence (AREA)
  • General Health & Medical Sciences (AREA)
  • Data Mining & Analysis (AREA)
  • Computational Linguistics (AREA)
  • Molecular Biology (AREA)
  • Computing Systems (AREA)
  • General Engineering & Computer Science (AREA)
  • Biophysics (AREA)
  • Mathematical Physics (AREA)
  • Software Systems (AREA)
  • Biomedical Technology (AREA)
  • Life Sciences & Earth Sciences (AREA)
  • Health & Medical Sciences (AREA)
  • Image Analysis (AREA)

Abstract

本申请公开了人工智能领域中的一种图像处理方法与装置、神经网络训练的方法与装置。该图像处理方法包括:确定待处理图像中的多个区域图像中每个区域图像对应的纹理复杂度信息;根据所述每个区域图像的纹理复杂度信息,确定所述每个区域图像对应的图像压缩模型,其中,不同的纹理复杂度信息对应不同的图像压缩模型;利用所述每个区域图像对应的压缩模型对所述每个区域图像进行压缩。该方法通过对于待处理图像中不同纹理复杂度的区域图像,使用与该区域图像纹理复杂度相对应的压缩模型进行图像的压缩,提高待处理图像的整体压缩效果。

Description

图像处理方法与装置、神经网络训练的方法与装置
技术领域
本申请涉及人工智能领域,并且更具体地,涉及一种图像处理方法与装置、神经网络训练的方法与装置。
背景技术
人工智能(artificial intelligence,AI)是利用数字计算机或者数字计算机控制的机器模拟、延伸和扩展人的智能,感知环境、获取知识并使用知识获得最佳结果的理论、方法、技术及应用系统。换句话说,人工智能是计算机科学的一个分支,它企图了解智能的实质,并生产出一种新的能以人类智能相似的方式作出反应的智能机器。人工智能也就是研究各种智能机器的设计原理与实现方法,使机器具有感知、推理与决策的功能。人工智能领域的研究包括机器人,自然语言处理,计算机视觉,决策与推理,人机交互,推荐与搜索,AI基础理论等。
图像压缩能够减小图像数据中的冗余信息,因此,图像压缩对于提高图像的存储效率和传输效率有着重要意义。传统的图像压缩方法如联合图像专家组(jointphotographic experts group,JPEG)在中高码率区域有较好的压缩效果,在低码率区域,传统的图像压缩方法的压缩效果不够理想。
可以通过神经网络对图像进行压缩,该方法主要利用神经网络和对应的非线性变换提取图像特征,从而达到压缩目的。这种方法相比传统的图像压缩方法,可以省去复杂的参数设计和模块设计。在解压时,可以采用神经网络进行解码以重构图像。如何提高神经网络对图像的压缩性能,成为一个亟需解决的技术问题。
发明内容
本申请提供一种图像处理方法与装置、及神经网络训练的方法与装置,能够提高神经网络对图像的压缩效果。
第一方面,提供了一种图像处理方法,所述方法包括:确定待处理图像中的多个区域图像中每个区域图像对应的纹理复杂度信息;根据所述每个区域图像对应的纹理复杂度信息,确定所述每个区域图像对应的图像压缩模型,其中,不同的纹理复杂度信息对应不同的图像压缩模型;利用所述每个区域图像对应的图像压缩模型对所述每个区域图像进行压缩。
一张完整的图像的不同区域中,图像的纹理复杂度可能并不相同。例如,在天空、海滩等背景区域,图像的纹理复杂度较低;在包括人物等目标的兴趣区域或前景区域,图像复杂度较高。
通过将待处理图像划分为多个区域图像,对每个区域图像分别使用与该图像纹理复杂度相对应的压缩模型进行图像的压缩,可以对待处理图像中不同纹理复杂度的区域进行与该纹理复杂度相适应的压缩处理,以提高对待处理图像整体的压缩效果。
结合第一方面,在一些可能的实现方式中,所述方法还包括:利用与压缩所述每个区域图像的图像压缩模型对应的图像解压模型,对所述每个区域图像压缩后得到的图像特征进行解压缩,得到所述每个区域图像对应的区域解压图像;对所述多个区域解压图像进行拼接处理和优化处理,得到恢复后的待处理图像,所述优化处理包括对所述多个区域解压图像的边沿进行调整。
由于对一张完整的图像进行了划分,将各个用于处理不同纹理复杂度的图像解压模型通过解压得到的区域解压图像进行拼接,两个相邻的区域解压图像在拼接之后可能出现线条不连续或颜色差异等。通过对一个或多个区域解压图像的边沿区域的像素进行调整,可以使得对经过压缩、解压、拼接处理后的完整图像与处理前的图像之间的图像失真度更小。
结合第一方面,在一些可能的实现方式中,所述确定待处理图像中的多个区域图像中每个区域图像对应的纹理复杂度信息,包括:计算每个区域图像中每个像素的梯度大小;根据所述每个像素的梯度大小,确定每个区域图像的纹理复杂度信息。
可以根据像素的亮度或其他颜色的表示方式,确定像素的梯度大小。
可以利用该区域图像中各个像素的梯度大小的中位数或平均数表示该区域图像的纹理复杂度。
结合第一方面,在一些可能的实现方式中,所述方法还包括:将所述待处理图像划分为所述多个区域图像,所述多个区域图像不重叠,且所述多个区域图像包括所述待处理图像中的全部像素。
多个区域图像包括所述待处理图像中的全部像素并且该多个区域图像不重叠,可以降低码率。
第二方面,提供一种神经网络训练方法,所述方法包括:确定训练图像中的多个训练区域图像中每个训练区域图像对应的纹理复杂度信息;根据所述每个训练区域图像对应的训练纹理复杂度信息,确定所述每个训练区域图像对应的编解码模型,其中,不同的纹理复杂度信息对应不同的编解码模型,每个所述编解码模型用于对输入的所述训练区域图像进行压缩,并对压缩结果进行解压得到多个解压训练区域图像;根据率失真调整所述编解码模型的参数,所述率失真根据所述解压训练区域图像和所述训练区域图像得到。
通过在训练过程中,针对每种纹理复杂度的训练区域图像,训练与该纹理复杂度相对应的编解码器,使得每个编解码器对于与之相对应的纹理复杂度的图片的压缩性能更好。
一张完整的图像中,图像的纹理复杂度可能并不相同。例如,在天空、海滩等背景区域,图像的纹理复杂度较低;在包括人物等目标的兴趣区域或前景区域,图像复杂度较高。将一张完整图像划分为多个个区域,从而利用每个区域的图像对编解码器进行训练,能够使得训练数据更加符合编解码器的图像对应的纹理复杂度,从而提升每个编解码器对于与之相对应的纹理复杂度的图片的压缩性能。
结合第二方面,在一些可能的实现方式中,通过融合模型对所述多个解压训练区域图像进行拼接处理和优化处理,得到所述训练恢复图像,所述优化处理包括对至少一个所述解压训练区域图像的边沿进行调整;根据所述训练恢复图像与所述训练图像之间的图像失真度,调整所述融合模型的参数。
通过采用融合模型,能够使得编解码器处理后的图像在拼接后图像失真度更小。
还可以根据所述训练恢复图像与所述训练图像之间的图像失真度,调整所述编解码模型的参数。
可选地,可以通过“端到端”的方式训练图像处理过程中所需的神经网络模型。
根据训练恢复图像与所述待处理训练图像的图像失真度,计算率失真,从而根据率失真调整编解码模型、融合模型的参数,完成对编解码模型和融合模型的训练,从而可以实现对图像处理过程中所需的神经网络模型的“端到端”的训练,降低了神经网络训练的复杂度。
应当理解,率失真可以根据图像失真度和码率确定。码率用于表示图像的压缩程度,可以根据编解码模型的压缩结果确定。
应当理解,融合模型还可以对解压训练区域图像的边沿之外的其他区域进行调整。
结合第二方面,在一些可能的实现方式中,所述确定训练图像中的多个训练区域图像中每个训练区域图像对应的纹理复杂度信息,包括:计算每个训练区域图像中每个像素的梯度大小;根据所述每个像素的梯度大小,确定每个训练区域图像的纹理复杂度信息。
结合第二方面,在一些可能的实现方式中,所述方法还包括:将所述训练图像划分为所述多个训练区域图像,所述多个训练区域图像不重叠,且所述多个训练区域图像包括所述训练图像中的全部像素。
第三方面,提供一种电子设备,图像处理装置,其特征在于,包括存储模块和处理模块;所述存储模块用于存储程序指令;当所述程序指令在所述处理器中执行时,所述处理模块用于:确定待处理图像中的多个区域图像中每个区域图像对应的纹理复杂度信息;根据所述每个区域图像对应的纹理复杂度信息,确定所述每个区域图像对应的图像压缩模型,其中,不同的纹理复杂度信息对应不同的图像压缩模型;利用所述每个区域图像对应的图像压缩模型对所述每个区域图像进行压缩。
结合第三方面,在一些可能的实现方式中,所述处理模块还用于:利用与压缩所述每个区域图像的图像压缩模型对应的图像解压模型,对所述每个区域图像压缩后得到的图像特征进行解压缩,得到所述每个区域图像对应的区域解压图像;对所述多个区域解压图像进行拼接处理和优化处理,得到恢复后的待处理图像,所述优化处理包括对所述多个区域解压图像的边沿进行像素调整。
结合第三方面,在一些可能的实现方式中,所述处理模块还用于:计算每个区域图像中每个像素的梯度大小;根据所述每个像素的梯度大小,确定每个区域图像的纹理复杂度信息。
结合第三方面,在一些可能的实现方式中,所述处理模块还用于:将所述待处理图像划分为所述多个区域图像,所述多个区域图像不重叠,且所述多个区域图像包括所述待处理图像中的全部像素。
第四方面,提供一种神经网络训练装置,包括存储模块和处理模块;所述存储模块用于存储程序指令,当所述程序指令在所述处理器中执行时,所述处理模块用于:确定训练图像中的多个训练区域图像中每个训练区域图像对应的纹理复杂度信息;根据所述每个训练区域图像对应的训练纹理复杂度信息,确定所述每个训练区域图像对应的编解码模型,其中,不同的纹理复杂度信息对应不同的编解码模型,每个所述编解码模型用于对输入的所述训练区域图像进行压缩,并对压缩结果进行解压得到多个解压训练区域图像;根据率失真调整所述编解码模型的参数,所述率失真根据所述解压训练区域图像和所述训练区域图像得到。
结合第四方面,在一些可能的实现方式中,所述处理模块还用于:通过融合模型对所述多个解压训练区域图像进行拼接处理和优化处理,得到训练恢复图像,所述优化处理包括对至少一个所述解压训练区域图像的边沿进行像素调整;根据所述训练恢复图像与所述训练图像的图像失真度,调整融合模型的参数。
还可以根据所述训练恢复图像与所述训练图像之间的图像失真度,调整所述编解码模型的参数。
结合第四方面,在一些可能的实现方式中,所述处理模块还用于:计算每个训练区域图像中每个像素的梯度大小;根据所述每个像素的梯度大小,确定每个训练区域图像的纹理复杂度信息。
结合第四方面,在一些可能的实现方式中,所述处理模块还用于:将所述训练图像划分为所述多个训练区域图像,所述多个训练区域图像不重叠,且所述多个训练区域图像包括所述训练图像中的全部像素。
第五方面,提供一种电子设备,包括存储器和处理器,所述存储器用于存储程序指令;当所述程序指令在所述处理器中执行时,所述处理器用于执行第一方面或第二方面所述的方法。
上述第五方面中的处理器既可以是中央处理器(central processing unit,CPU),也可以是CPU与神经网络运算处理器的组合,这里的神经网络运算处理器可以包括图形处理器(graphics processing unit,GPU)、神经网络处理器(neural-networkprocessing unit,NPU)和张量处理器(tensor processing unit,TPU)等等。其中,TPU是谷歌(google)为机器学习全定制的人工智能加速器专用集成电路。
第六方面,提供一种计算机可读介质,该计算机可读介质存储用于设备执行的程序代码,该程序代码包括用于执行第一方面或第二方面中的任意一种实现方式中的方法。
第七方面,提供一种包含指令的计算机程序产品,当该计算机程序产品在计算机上运行时,使得计算机执行上述第一方面或第二方面中的任意一种实现方式中的方法。
第八方面,提供一种芯片,所述芯片包括处理器与数据接口,所述处理器通过所述数据接口读取存储器上存储的指令,执行上述第一方面或第二方面中的任意一种实现方式中的方法。
可选地,作为一种实现方式,所述芯片还可以包括存储器,所述存储器中存储有指令,所述处理器用于执行所述存储器上存储的指令,当所述指令被执行时,所述处理器用于执行第一方面或第二方面中的任意一种实现方式中的方法。
上述芯片具体可以是现场可编程门阵列(field-programmable gate array,FPGA)或者专用集成电路(application-specific integrated circuit,ASIC)。
附图说明
图1为本申请实施例提供的一种系统架构的结构示意图。
图2为本申请实施例提供的一种卷积神经网络的结构示意图。
图3为本申请实施例提供的另一种卷积神经网络的结构示意图。
图4为本申请实施例提供的一种芯片的硬件结构示意图。
图5为本申请实施例提供的一种系统架构的示意图。
图6是一种图像处理系统的示意性结构图。
图7是本申请实施例提供的一种图像处理系统的示意性结构图。
图8是本申请实施例提供的一种神经网络训练方法的示意性结构图。
图9是本申请实施例提供的一种图像处理方法的示意性流程图。
图10是本申请实施例提高的图像处理方法的压缩性能的示意图。
图11是本申请实施例提供的一种图像处理装置的示意性结构图。
图12是本申请实施例提供的一种神经网络训练装置的示意性结构图。
图13是本申请实施例的图像处理装置的示意性结构图。
图14是本申请实施例的神经网络训练装置的示意性结构图。
具体实施方式
下面将结合本申请实施例中的附图,对本申请实施例中的技术方案进行描述,显然,所描述的实施例仅仅是本申请一部分实施例,而不是全部的实施例。基于本申请中的实施例,本领域普通技术人员在没有做出创造性劳动前提下所获得的所有其他实施例,都属于本申请保护的范围。
由于本申请实施例涉及大量神经网络的应用,为了便于理解,下面先对本申请实施例可能涉及的神经网络的相关术语和概念进行介绍。
(1)神经网络
神经网络可以是由神经单元组成的,神经单元可以是指以xs和截距1为输入的运算单元,该运算单元的输出可以为:
Figure BDA0002610949950000051
其中,s=1、2、……n,n为大于1的自然数,Ws为xs的权重,b为神经单元的偏置。f为神经单元的激活函数(activation functions),用于将非线性特性引入神经网络中,来将神经单元中的输入信号转换为输出信号。该激活函数的输出信号可以作为下一层卷积层的输入,激活函数可以是sigmoid函数。神经网络是将多个上述单一的神经单元联结在一起形成的网络,即一个神经单元的输出可以是另一个神经单元的输入。每个神经单元的输入可以与前一层的局部接受域相连,来提取局部接受域的特征,局部接受域可以是由若干个神经单元组成的区域。
(2)深度神经网络
深度神经网络(deep neural network,DNN),也称多层神经网络,可以理解为具有多层隐含层的神经网络。按照不同层的位置对DNN进行划分,DNN内部的神经网络可以分为三类:输入层,隐含层,输出层。一般来说第一层是输入层,最后一层是输出层,中间的层数都是隐含层。层与层之间是全连接的,也就是说,第i层的任意一个神经元一定与第i+1层的任意一个神经元相连。
虽然DNN看起来很复杂,但是就每一层的工作来说,其实并不复杂,简单来说就是如下线性关系表达式:
Figure BDA0002610949950000052
其中,
Figure BDA0002610949950000053
是输入向量,
Figure BDA0002610949950000054
是输出向量,
Figure BDA0002610949950000055
是偏移向量,W是权重矩阵(也称系数),α()是激活函数。每一层仅仅是对输入向量
Figure BDA0002610949950000056
经过如此简单的操作得到输出向量
Figure BDA0002610949950000057
由于DNN层数多,系数W和偏移向量
Figure BDA0002610949950000058
的数量也比较多。这些参数在DNN中的定义如下所述:以系数W为例:假设在一个三层的DNN中,第二层的第4个神经元到第三层的第2个神经元的线性系数定义为
Figure BDA0002610949950000059
上标3代表系数W所在的层数,而下标对应的是输出的第三层索引2和输入的第二层索引4。
综上,第L-1层的第k个神经元到第L层的第j个神经元的系数定义为
Figure BDA0002610949950000061
需要注意的是,输入层是没有W参数的。在深度神经网络中,更多的隐含层让网络更能够刻画现实世界中的复杂情形。理论上而言,参数越多的模型复杂度越高,“容量”也就越大,也就意味着它能完成更复杂的学习任务。训练深度神经网络的也就是学习权重矩阵的过程,其最终目的是得到训练好的深度神经网络的所有层的权重矩阵(由很多层的向量W形成的权重矩阵)。
(3)卷积神经网络
卷积神经网络(convolutional neuron network,CNN)是一种带有卷积结构的深度神经网络。卷积神经网络包含了一个由卷积层和子采样层构成的特征抽取器,该特征抽取器可以看作是滤波器。卷积层是指卷积神经网络中对输入信号进行卷积处理的神经元层。在卷积神经网络的卷积层中,一个神经元可以只与部分邻层神经元连接。一个卷积层中,通常包含若干个特征平面,每个特征平面可以由一些矩形排列的神经单元组成。同一特征平面的神经单元共享权重,这里共享的权重就是卷积核。共享权重可以理解为提取图像信息的方式与位置无关。卷积核可以以随机大小的矩阵的形式初始化,在卷积神经网络的训练过程中卷积核可以通过学习得到合理的权重。另外,共享权重带来的直接好处是减少卷积神经网络各层之间的连接,同时又降低了过拟合的风险。
(4)循环神经网络(recurrent neural networks,RNN)是用来处理序列数据的。在传统的神经网络模型中,是从输入层到隐含层再到输出层,层与层之间是全连接的,而对于每一层层内之间的各个节点是无连接的。这种普通的神经网络虽然解决了很多难题,但是却仍然对很多问题无能无力。例如,你要预测句子的下一个单词是什么,一般需要用到前面的单词,因为一个句子中前后单词并不是独立的。RNN之所以称为循环神经网路,即一个序列当前的输出与前面的输出也有关。具体的表现形式为网络会对前面的信息进行记忆并应用于当前输出的计算中,即隐含层本层之间的节点不再无连接而是有连接的,并且隐含层的输入不仅包括输入层的输出还包括上一时刻隐含层的输出。理论上,RNN能够对任何长度的序列数据进行处理。对于RNN的训练和对传统的CNN或DNN的训练一样。
既然已经有了卷积神经网络,为什么还要循环神经网络?原因很简单,在卷积神经网络中,有一个前提假设是:元素之间是相互独立的,输入与输出也是独立的,比如猫和狗。但现实世界中,很多元素都是相互连接的,比如股票随时间的变化,再比如一个人说了:我喜欢旅游,其中最喜欢的地方是云南,以后有机会一定要去。这里填空,人类应该都知道是填“云南”。因为人类会根据上下文的内容进行推断,但如何让机器做到这一步?RNN就应运而生了。RNN旨在让机器像人一样拥有记忆的能力。因此,RNN的输出就需要依赖当前的输入信息和历史的记忆信息。
(5)损失函数
在训练深度神经网络的过程中,因为希望深度神经网络的输出尽可能的接近真正想要预测的值,所以可以通过比较当前网络的预测值和真正想要的目标值,再根据两者之间的差异情况来更新每一层神经网络的权重向量(当然,在第一次更新之前通常会有初始化的过程,即为深度神经网络中的各层预先配置参数),比如,如果网络的预测值高了,就调整权重向量让它预测低一些,不断地调整,直到深度神经网络能够预测出真正想要的目标值或与真正想要的目标值非常接近的值。因此,就需要预先定义“如何比较预测值和目标值之间的差异”,这便是损失函数(loss function)或目标函数(objective function),它们是用于衡量预测值和目标值的差异的重要方程。其中,以损失函数举例,损失函数的输出值(loss)越高表示差异越大,那么深度神经网络的训练就变成了尽可能缩小这个loss的过程。
(6)反向传播算法
神经网络可以采用误差反向传播(back propagation,BP)算法在训练过程中修正初始的神经网络模型中参数的大小,使得神经网络模型的重建误差损失越来越小。具体地,前向传递输入信号直至输出会产生误差损失,通过反向传播误差损失信息来更新初始的神经网络模型中参数,从而使误差损失收敛。反向传播算法是以误差损失为主导的反向传播运动,旨在得到最优的神经网络模型的参数,例如权重矩阵。
(7)像素值
图像的像素值可以是一个红绿蓝(RGB)颜色值,像素值可以是表示颜色的长整数。例如,像素值为256*Red+100*Green+76Blue,其中,Blue代表蓝色分量,Green代表绿色分量,Red代表红色分量。各个颜色分量中,数值越小,亮度越低,数值越大,亮度越高。对于灰度图像来说,像素值可以是灰度值。
(8)图像压缩
图像压缩,是指以较少的比特有损或无损地表示原来的像素矩阵的技术,也称图像编码。通过图像压缩,对图像内容执行变换,可以减少表示数字图像时需要的数据量,从而可以降低图像存储占用的空间。
图像数据之所以能被压缩,就是因为数据中存在着冗余。图像数据的冗余主要表现为:图像中相邻像素间的相关性引起的空间冗余;图像序列中不同帧之间存在相关性引起的时间冗余;不同彩色平面或频谱带的相关性引起的频谱冗余。数据压缩的目的就是通过去除这些数据冗余来减少表示数据所需的比特数。由于图像数据量的庞大,在存储、传输、处理时非常困难,因此图像数据的压缩就显得非常重要。
(9)图像解压
图像解压是图像压缩的逆过程,也可以称为解压或解码。通过图像解码,可以将输入的紧凑表示的信息格式恢复为图像。
(10)图像失真度
一般采用原始图像与编码重建图像之间的峰值性噪比(peak signal-to-noiseratio,PSNR)衡量图像失真度,这个PSNR可以是亮度的PSNR,也可以是亮度与色度的PSNR的线性组合。一般最简单的情况下,采用亮度的PSNR(Y-PSNR)作为主要衡量依据。其中,峰值信号即图像中像素的最大值(比方说像素亮度的最大值),噪声指原始图像与重建图像中各像素值的均方差(差值的平方取均值);将两者的比值转换成分贝形式,即为PSNR。
(11)码率
码率(rate),也可以称为编码码率,可以是压缩后的数据中每个像素的平均数据量((bit-per-pixel,bpp),用于表示数据压缩的程度。码率可以根据图像压缩后数据量的比例确定。
(12)率失真
率失真用于表示图像失真度与码率之间的关系。
图像失真度越小,意味着图像细节越多,码率越大。图像失真度越大,图像由于压缩所损失的细节越多,那么码率越小。
率失真优化(rate distortion optimization,R-D optimization)是指按照预设规则,尽可能减小图像失真度和码率。也就是说,可以在尽可能小的码率的情况下,使得获取的图像失真度尽可能的少,从而达到较好的压缩效果。通过率失真优化,可以在码率与失真度中找出平衡点,使压缩效果最优。当然,率失真优化的规则也可以是在保证码率不过上限的情况下失真度最小,或者,在保证失真度不过下限的情况下码率最小,等等。
可以通过率失真函数计算率失真。率失真函数例如可以表示为I=D+λR,或I=D·R等,其中,I为率失真,R为码率,R为失真度,λ为预设的拉格朗日系数。
如图1所示,本申请实施例提供了一种系统架构100。在图1中,数据采集设备160用于采集训练数据。针对本申请实施例的图像处理方法来说,训练数据可以包括训练图像。
在采集到训练数据之后,数据采集设备160将这些训练数据存入数据库130,训练设备120基于数据库130中维护的训练数据训练得到目标模型/规则101。
下面对训练设备120基于训练数据得到目标模型/规则101进行描述,训练设备120对输入的原始图像进行处理,将输出的图像与原始图像进行对比,直到根据训练设备120输出的图像与原始图像的差值确定的率失真小于一定的阈值,从而完成目标模型/规则101的训练。
上述目标模型/规则101能够用于实现本申请实施例的图像处理方法。本申请实施例中的目标模型/规则101具体可以为神经网络。需要说明的是,在实际的应用中,所述数据库130中维护的训练数据不一定都来自于数据采集设备160的采集,也有可能是从其他设备接收得到的。另外需要说明的是,训练设备120也不一定完全基于数据库130维护的训练数据进行目标模型/规则101的训练,也有可能从云端或其他地方获取训练数据进行模型训练,上述描述不应该作为对本申请实施例的限定。
根据训练设备120训练得到的目标模型/规则101可以应用于不同的系统或设备中,如应用于图1所示的执行设备110,所述执行设备110可以是终端,如手机终端,平板电脑,笔记本电脑,增强现实(augmented reality,AR)AR/虚拟现实(virtual reality,VR),车载终端等,还可以是服务器或者云端等。在图1中,执行设备110配置输入/输出(input/output,I/O)接口112,用于与外部设备进行数据交互,用户可以通过客户设备140向I/O接口112输入数据,所述输入数据在本申请实施例中可以包括:客户设备输入的待处理图像。
预处理模块113和预处理模块114用于根据I/O接口112接收到的输入数据(如待处理图像)进行预处理,在本申请实施例中,也可以没有预处理模块113和预处理模块114(也可以只有其中的一个预处理模块),而直接采用计算模块111对输入数据进行处理。
在执行设备110对输入数据进行预处理,或者在执行设备110的计算模块111执行计算等相关的处理过程中,执行设备110可以调用数据存储系统150中的数据、代码等以用于相应的处理,也可以将相应处理得到的数据、指令等存入数据存储系统150中。
最后,I/O接口112将处理结果,如上述得到的图像的分类结果返回给客户设备140,从而提供给用户。
值得说明的是,训练设备120可以针对不同的目标或称不同的任务,基于不同的训练数据生成相应的目标模型/规则101,该相应的目标模型/规则101即可以用于实现上述目标或完成上述任务,从而为用户提供所需的结果。
在图1中所示情况下,用户可以手动给定输入数据,该手动给定可以通过I/O接口112提供的界面进行操作。另一种情况下,客户设备140可以自动地向I/O接口112发送输入数据,如果要求客户设备140自动发送输入数据需要获得用户的授权,则用户可以在客户设备140中设置相应权限。用户可以在客户设备140查看执行设备110输出的结果,具体的呈现形式可以是显示、声音、动作等具体方式。客户设备140也可以作为数据采集端,采集如图所示输入I/O接口112的输入数据及输出I/O接口112的输出结果作为新的样本数据,并存入数据库130。当然,也可以不经过客户设备140进行采集,而是由I/O接口112直接将如图所示输入I/O接口112的输入数据及输出I/O接口112的输出结果,作为新的样本数据存入数据库130。
值得注意的是,图1仅是本申请实施例提供的一种系统架构的示意图,图中所示设备、器件、模块等之间的位置关系不构成任何限制,例如,在图1中,数据存储系统150相对执行设备110是外部存储器,在其它情况下,也可以将数据存储系统150置于执行设备110中。
如图1所示,根据训练设备120训练得到目标模型/规则101,该目标模型/规则101在本申请实施例中可以是本申请中的神经网络,具体的,本申请实施例使用神经网络可以为CNN,深度卷积神经网络(deep convolutional neural networks,DCNN),循环神经网络(recurrent neural network,RNN)等等。
由于CNN是一种非常常见的神经网络,下面结合图2重点对CNN的结构进行详细的介绍。如上文的基础概念介绍所述,卷积神经网络是一种带有卷积结构的深度神经网络,是一种深度学习(deep learning)架构,深度学习架构是指通过机器学习的算法,在不同的抽象层级上进行多个层次的学习。作为一种深度学习架构,CNN是一种前馈(feed-forward)人工神经网络,该前馈人工神经网络中的各个神经元可以对输入其中的图像作出响应。
本申请实施例的图像处理方法具体采用的神经网络的结构可以如图2所示。在图2中,卷积神经网络(CNN)200可以包括输入层210,卷积层/池化层220(其中池化层为可选的),以及神经网络层230。其中,输入层210可以获取待处理图像,并将获取到的待处理图像交由卷积层/池化层220以及后面的神经网络层230进行处理,可以得到图像的处理结果。下面对图2中的CNN 200中内部的层结构进行详细的介绍。
卷积层/池化层220:
卷积层:
如图2所示卷积层/池化层220可以包括如示例221-226层,举例来说:在一种实现中,221层为卷积层,222层为池化层,223层为卷积层,224层为池化层,225为卷积层,226为池化层;在另一种实现方式中,221、222为卷积层,223为池化层,224、225为卷积层,226为池化层。即卷积层的输出可以作为随后的池化层的输入,也可以作为另一个卷积层的输入以继续进行卷积操作。
下面将以卷积层221为例,介绍一层卷积层的内部工作原理。
卷积层221可以包括很多个卷积算子,卷积算子也称为核,其在图像处理中的作用相当于一个从输入图像矩阵中提取特定信息的过滤器,卷积算子本质上可以是一个权重矩阵,这个权重矩阵通常被预先定义,在对图像进行卷积操作的过程中,权重矩阵通常在输入图像上沿着水平方向一个像素接着一个像素(或两个像素接着两个像素……这取决于步长stride的取值)的进行处理,从而完成从图像中提取特定特征的工作。该权重矩阵的大小应该与图像的大小相关,需要注意的是,权重矩阵的纵深维度(depth dimension)和输入图像的纵深维度是相同的,在进行卷积运算的过程中,权重矩阵会延伸到输入图像的整个深度。因此,和一个单一的权重矩阵进行卷积会产生一个单一纵深维度的卷积化输出,但是大多数情况下不使用单一权重矩阵,而是应用多个尺寸(行×列)相同的权重矩阵,即多个同型矩阵。每个权重矩阵的输出被堆叠起来形成卷积图像的纵深维度,这里的维度可以理解为由上面所述的“多个”来决定。不同的权重矩阵可以用来提取图像中不同的特征,例如一个权重矩阵用来提取图像边缘信息,另一个权重矩阵用来提取图像的特定颜色,又一个权重矩阵用来对图像中不需要的噪点进行模糊化等。该多个权重矩阵尺寸(行×列)相同,经过该多个尺寸相同的权重矩阵提取后的卷积特征图的尺寸也相同,再将提取到的多个尺寸相同的卷积特征图合并形成卷积运算的输出。
这些权重矩阵中的权重值在实际应用中需要经过大量的训练得到,通过训练得到的权重值形成的各个权重矩阵可以用来从输入图像中提取信息,从而使得卷积神经网络200进行正确的预测。
当卷积神经网络200有多个卷积层的时候,初始的卷积层(例如221)往往提取较多的一般特征,该一般特征也可以称之为低级别的特征;随着卷积神经网络200深度的加深,越往后的卷积层(例如226)提取到的特征越来越复杂,比如高级别的语义之类的特征,语义越高的特征越适用于待解决的问题。
池化层:
由于常常需要减少训练参数的数量,因此卷积层之后常常需要周期性的引入池化层,在如图2中220所示例的221-226各层,可以是一层卷积层后面跟一层池化层,也可以是多层卷积层后面接一层或多层池化层。在图像处理过程中,池化层的唯一目的就是减少图像的空间大小。池化层可以包括平均池化算子和/或最大池化算子,以用于对输入图像进行采样得到较小尺寸的图像。平均池化算子可以在特定范围内对图像中的像素值进行计算产生平均值作为平均池化的结果。最大池化算子可以在特定范围内取该范围内值最大的像素作为最大池化的结果。另外,就像卷积层中用权重矩阵的大小应该与图像尺寸相关一样,池化层中的运算符也应该与图像的大小相关。通过池化层处理后输出的图像尺寸可以小于输入池化层的图像的尺寸,池化层输出的图像中每个像素点表示输入池化层的图像的对应子区域的平均值或最大值。
神经网络层230:
在经过卷积层/池化层220的处理后,卷积神经网络200还不足以输出所需要的输出信息。因为如前所述,卷积层/池化层220只会提取特征,并减少输入图像带来的参数。然而为了生成最终的输出信息(所需要的类信息或其他相关信息),卷积神经网络200需要利用神经网络层230来生成一个或者一组所需要的类的数量的输出。因此,在神经网络层230中可以包括多层隐含层(如图2所示的231、232至23n)以及输出层240,该多层隐含层中所包含的参数可以根据具体的任务类型的相关训练数据进行预先训练得到,例如该任务类型可以包括图像识别,图像分类,图像超分辨率重建等等。
在神经网络层230中的多层隐含层之后,也就是整个卷积神经网络200的最后层为输出层240,该输出层240具有类似分类交叉熵的损失函数,具体用于计算预测误差,一旦整个卷积神经网络200的前向传播(如图2由210至240方向的传播为前向传播)完成,反向传播(如图2由240至210方向的传播为反向传播)就会开始更新前面提到的各层的权重值以及偏差,以减少卷积神经网络200的损失,及卷积神经网络200通过输出层输出的结果和理想结果之间的误差。
本申请实施例的图像处理方法具体采用的神经网络的结构可以如图3所示。在图3中,卷积神经网络(CNN)200可以包括输入层110,卷积层/池化层120(其中池化层为可选的),以及神经网络层130。与图2相比,图3中的卷积层/池化层120中的多个卷积层/池化层并行,将分别提取的特征均输入给全神经网络层130进行处理。
需要说明的是,图2和图3所示的卷积神经网络仅作为一种本申请实施例的图像处理方法的两种可能的卷积神经网络的示例,在具体的应用中,本申请实施例的图像处理方法所采用的卷积神经网络还可以以其他网络模型的形式存在。
图4为本申请实施例提供的一种芯片的硬件结构,该芯片包括神经网络处理器50。该芯片可以被设置在如图1所示的执行设备110中,用以完成计算模块111的计算工作。该芯片也可以被设置在如图1所示的训练设备120中,用以完成训练设备120的训练工作并输出目标模型/规则101。如图2和图3所示的卷积神经网络中各层的算法均可在如图4所示的芯片中得以实现。
神经网络处理器NPU 50作为协处理器挂载到主中央处理器(central processingunit,CPU)(host CPU)上,由主CPU分配任务。NPU的核心部分为运算电路503,控制器504控制运算电路503提取存储器(权重存储器或输入存储器)中的数据并进行运算。
在一些实现中,运算电路503内部包括多个处理单元(process engine,PE)。在一些实现中,运算电路503是二维脉动阵列。运算电路503还可以是一维脉动阵列或者能够执行例如乘法和加法这样的数学运算的其它电子线路。在一些实现中,运算电路503是通用的矩阵处理器。
举例来说,假设有输入矩阵A,权重矩阵B,输出矩阵C。运算电路从权重存储器502中取矩阵B相应的数据,并缓存在运算电路中每一个PE上。运算电路从输入存储器501中取矩阵A数据与矩阵B进行矩阵运算,得到的矩阵的部分结果或最终结果,保存在累加器(accumulator)508中。
向量计算单元507可以对运算电路的输出做进一步处理,如向量乘,向量加,指数运算,对数运算,大小比较等等。例如,向量计算单元507可以用于神经网络中非卷积/非FC层的网络计算,如池化(pooling),批归一化(batch normalization),局部响应归一化(local response normalization)等。
在一些实现种,向量计算单元能507将经处理的输出的向量存储到统一缓存器506。例如,向量计算单元507可以将非线性函数应用到运算电路503的输出,例如累加值的向量,用以生成激活值。在一些实现中,向量计算单元507生成归一化的值、合并值,或二者均有。在一些实现中,处理过的输出的向量能够用作到运算电路503的激活输入,例如用于在神经网络中的后续层中的使用。
统一存储器506用于存放输入数据以及输出数据。
权重数据直接通过存储单元访问控制器505(direct memory accesscontroller,DMAC)将外部存储器中的输入数据搬运到输入存储器501和/或统一存储器506、将外部存储器中的权重数据存入权重存储器502,以及将统一存储器506中的数据存入外部存储器。
总线接口单元(bus interface unit,BIU)510,用于通过总线实现主CPU、DMAC和取指存储器509之间进行交互。
与控制器504连接的取指存储器(instruction fetch buffer)509,用于存储控制器504使用的指令;
控制器504,用于调用指存储器509中缓存的指令,实现控制该运算加速器的工作过程。
一般地,统一存储器506,输入存储器501,权重存储器502以及取指存储器509均为片上(On-Chip)存储器,外部存储器为该NPU外部的存储器,该外部存储器可以为双倍数据率同步动态随机存储器(double data rate synchronous dynamic random accessmemory,简称DDR SDRAM)、高带宽存储器(high bandwidth memory,HBM)或其他可读可写的存储器。
其中,图2和图3所示的卷积神经网络中各层的运算可以由运算电路503或向量计算单元507执行。
上文中介绍的图1中的执行设备110能够执行本申请实施例的图像处理方法的各个步骤,图2和图3所示的CNN模型和图4所示的芯片也可以用于执行本申请实施例的图像处理方法的各个步骤。下面结合附图对本申请实施例的神经网络训练的方法和本申请实施例的图像处理方法进行详细的介绍。
如图5所示,本申请实施例提供了一种系统架构300。该系统架构包括本地设备301、本地设备302以及执行设备210和数据存储系统250,其中,本地设备301和本地设备302通过通信网络与执行设备210连接。
执行设备210可以由一个或多个服务器实现。可选的,执行设备210可以与其它计算设备配合使用,例如:数据存储器、路由器、负载均衡器等设备。执行设备210可以布置在一个物理站点上,或者分布在多个物理站点上。执行设备210可以使用数据存储系统250中的数据,或者调用数据存储系统250中的程序代码来实现本申请实施例的图像处理的方法。
具体地,执行设备210可以执行以下过程:确定待处理图像中的多个区域图像中每个区域图像对应的纹理复杂度信息;根据所述每个区域图像对应的纹理复杂度信息,确定所述每个区域图像对应的图像压缩模型,其中,不同的纹理复杂度信息对应不同的图像压缩模型;利用所述每个区域图像对应的图像压缩模型对所述每个区域图像进行压缩。
通过上述过程执行设备210能够通过对于不同纹理复杂度的区域图像,使用与该区域图像纹理复杂度相对应的压缩模型进行图像的压缩,提高对待处理图像的图像压缩的效果。
用户可以操作各自的用户设备(例如本地设备301和本地设备302)与执行设备210进行交互。每个本地设备可以表示任何计算设备,例如个人计算机、计算机工作站、智能手机、平板电脑、智能摄像头、智能汽车或其他类型蜂窝电话、媒体消费设备、可穿戴设备、机顶盒、游戏机等。
每个用户的本地设备可以通过任何通信机制/通信标准的通信网络与执行设备210进行交互,通信网络可以是广域网、局域网、点对点连接等方式,或它们的任意组合。
在一种实现方式中,本地设备301、本地设备302从执行设备210获取到目标神经网络的相关参数,将目标神经网络部署在本地设备301、本地设备302上,利用该目标神经网络进行图像分类或者图像处理等等。
在另一种实现中,执行设备210上可以直接部署目标神经网络,执行设备210通过从本地设备301和本地设备302获取待处理图像,并根据目标神经网络对待处理图像进行分类或者其他类型的图像处理。
上述执行设备210也可以为云端设备,此时,执行设备210可以部署在云端;或者,上述执行设备210也可以为终端设备,此时,执行设备210可以部署在用户终端侧,本申请实施例对此并不限定。
图6示出一种图像处理系统的示意性结构图。
图像处理系统600包括编码端的编码器610和量化模块620,以及解码端的解码器630。其中,编码器610和解码器630为神经网络。
图像处理系统600可以应用在对图像进行传输和存储的场景中。
编码端的编码器610和量化模块620可以设置在云端的服务器中。图像数据在云端执行图像编码过程,得到紧凑表示的压缩数据。对压缩数据进行存储,可以减少保存图像占用的存储空间。对压缩数据进行传输,能够降低图像传输过程的对传输资源的占用,降低对带宽的需求。
解码端的解码器630可以设置在作为客户端的终端设备中。解码端对压缩数据执行解码操作,得到重构图像。终端设备可以通过显示器显示该重构图像。
编码器610用于对待处理图像进行特征的提取,从而得到图像特征。
量化模块620用于对图像特征进行量化,以得到压缩数据。量化,即数字信号处理领域中将信号的连续取值(或者大量可能的离散取值)近似为有限多个(或较少的)离散值的过程。
云端可以将压缩数据传输至客户端。
解码器630用于对压缩数据进行解压缩,以得到重构图像。
图像处理系统600使用整张图像作为输入,通过对图像执行非线性变换,降低码字之间的相关性,提升神经网络的压缩性能。
自然图像本身包含了非常丰富的纹理信息。纹理在计算机图形学中既包括通常意义上物体表面的纹理即使物体表面呈现凹凸不平的沟纹,同时也包括在物体的光滑表面上的彩色图案。纹理复杂度可以用于反映图像中像素值的变换剧烈程度。不同的类别的图像之间纹理细节等特征差别显著,纹理复杂度不同的图像内容特性相差较大。
图像处理系统600对于纹理复杂度不同的图像,利用相同的编码器进行相同的处理,阻碍了压缩性能的进一步提升。
为了解决上述问题,本申请实施例提供了一种图像处理系统,以提高图像压缩性能。
图7是本申请实施例提供的一种图像处理系统的示意性结构图。根据图像的纹理复杂度,在多个神经网络中选择与该图像纹理复杂度对应的神经网络,结构实现了神经网络的选择根据图像内容的自适应调整,根据不同的纹理特性对图像进行压缩,实现了图像压缩性能的进一步提升。
图像处理系统700包括压缩系统710和解压系统720。其中,压缩系统710包括分割模型711、分类模型712、压缩模型713,解压系统720包括解压模块721、融合模块722。
压缩系统710和解压系统720可以位于相同或不同的设备中。
将待处理图像输入压缩系统710,压缩系统710用于对待处理图像进行压缩处理。
分割模型711可以将待处理图像进行分割,以得到多个区域图像。
区域图像也可以称为图像块。
该多个区域图像的尺寸可以相同或不相同。为了降低图像分割的难度,可以按照目标尺寸对图像进行分割,以得到尺寸相同的多个区域图像。
该多个区域图像之间可以存在或不存在重叠。为了提高压缩性能,该多个区域图像不重叠。
例如,可以将待处理图像划分为多个128×128的区域图像。通过多待处理图像的无重叠划分,形成了多个内容无重复的区域图像。
将区域图像输入分类模型712。分类模型712用于计算输入图像的纹理复杂度。
对于数字图像的简单一阶微分运算,由于其具有固定的方向性,只能检测特定的某一方向的边缘。为了检测克服一阶导数的缺点,可以计算图像中各个像素的梯度,从而实现对图像的一阶微分运算,并考虑方向性。
图像的梯度的方向是在图像灰度最大变化率上,可以反映出图像边缘上的灰度变化。梯度算子总是指向变换最剧烈的方向,在图像处理中,梯度算子的方向与图像中的边缘正交。梯度算子的大小表示图像灰度的变化率。
分类模型712可以对输入的图像计算每个像素在水平方向和竖直方向的亮度的差分值。根据每个像素在水平方向和竖直方向的亮度的差分值,可以计算该像素的梯度大小。根据每个像素的梯度大小,可以确定平均梯度大小或该图像中各个像素的梯度大小的中位数。该图像的平均梯度大小或该中位数可以指示图像的平滑程度,反映图像的纹理复杂度。
压缩模块713用于根据图像的纹理复杂度,利用与该纹理复杂度对应的压缩模型对图像进行图像特征的提取,以实现对图像的压缩。压缩模块713中可以包括用于对特征进行提取的AI模型,称为压缩模型(也可以称为图像压缩模型),或者,压缩模块713可以通过接口调用压缩模型,以实现对图像特征的提取。压缩模型可以是预训练完成的一种神经网络模型。可以将区域图像输入压缩模型,以得到该区域图像的图像特征。压缩模型例如可以是CNN、RNN等。
压缩模块713可以保存有纹理复杂度与压缩模型的对应关系。从而,根据区域图像的纹理复杂度,压缩模块713可以从多个与压缩模型中确定与该纹理复杂度对应的压缩模型,对该区域图像进行处理。
根据压缩模块713对区域图像的处理,可以得到每个区域图像的图像特征。
压缩系统710还可以包括量化模型等。量化模型可以对图像特征进行量化处理。
解压系统720用于对于压缩系统710的处理结果进行解压缩。
将压缩系统710处理得到的图像特征输入解压模块721。压缩系统710处理得到的图像特征可以是压缩模块713输出的图像特征。在压缩系统710还可以包括量化模型的情况下,压缩系统710处理得到的图像特征可以是量化后的图像特征。
解压模块721用于对图像特征进行图像解压,以得到解压后的图像。
解压模块721可以用于根据图像的纹理复杂度,利用与该纹理复杂度对应的解压模型对图像特征进行图像解压,以得到解压后的图像。
或者,解压模块721可以接收指示信息,指示信息用于指示每个图像特征对应的解压模型。解压模块721可以利用指示信息指示的解压模型,对该图像特征进行解压。
解压模块721中可以包括用于对图像特征进行解压的AI模型,称为解压模型或图像解压模型,或者,解压模块721可以通过接口调用解压模型,以实现对图像的解压。解压模型可以是预训练完成的一种神经网络模型。可以将图像特征和该图像特征对应的图像纹理复杂度输入解压模型,以得到解压后的该区域图像。解压模型例如可以是CNN、RNN等。
解压模块721可以保存有纹理复杂度与解压模型的对应关系。从而,根据区域图像的纹理复杂度,解压模块721可以从多个与压缩模型中确定与该纹理复杂度对应的解压模型,对该图像特征进行处理。
通过解压模块721的处理,可以得到恢复后的各个区域图像。
融合模块722用于对恢复后的各个区域图像进行融合。融合模块722中可以包括用于图像融合的AI模型,称为融合模型,或者,融合模块722可以通过接口调用融合模型,以实现对区域图像的融合。融合模型可以是预训练完成的一种神经网络模型。可以将恢复后的各个区域图像输入融合模型,以得到融合后的图像。该融合后的图像也可以称为重构图像或压缩重构图像。融合模型例如可以是CNN等。
区域图像的融合,可以是将区域图像进行拼接。
进一步地,区域图像的融合还可以包括对区域图像边缘像素的调整,以使得重构图像与待处理图像之间的误差更小,降低失真度。
图像处理系统700通过对区域图像纹理复杂度的计算,在区域图像的纹理复杂程度不同的情况下,利用不同的压缩模型和解压模型进行数据处理,提高了图像压缩性能。
图像处理系统700通过对待处理图像进行分割,对不同的区域图像分别进行纹理复杂程度的计算,可以对待处理图像的前景和背景采用不同的压缩模型和解压模型进行数据处理,提高了图像压缩性能。
图像处理系统700中的解压系统720通过进行区域图像融合时,对区域图像边缘像素的调整,降低图像失真度,提高了图像压缩性能。
图像处理系统700中采用的各个AI模型可以通过端到端的训练得到;或者,也可以先对压缩模型和解压模型进行训练,之后,训练融合模型。图像处理系统700中采用的AI模型的训练方法可以参见图8的说明。
端到端的训练是一种机器学习范式,整个学习的流程并不进行人为的子问题划分,而是完全交给深度学习模型直接学习从原始数据到期望输出的映射。
图8是本申请实施例提供的一种神经网络模型训练方法的示意性结构图。
在S810,确定训练图像中的多个训练区域图像中每个训练区域图像对应的纹理复杂度信息。
可以获取待处理训练图像。可以对完整的待处理训练图像进行划分,以得到多个训练区域图像。
所述多个训练区域图像不重叠,且所述多个训练区域图像包括所述训练图像中的全部像素。
在S820,根据所述每个训练区域图像对应的训练纹理复杂度信息,确定所述每个训练区域图像对应的编解码模型。
不同的纹理复杂度信息对应不同的编解码模型。
每个所述编解码模型用于对输入的所述训练区域图像进行压缩,并对压缩结果进行解压,从而得到多个解压训练区域图像。
也就是说,编解码模型包括压缩模型和解压模型。其中,压缩模型用于对图像进行压缩,解压模型用于对压缩模型的处理结果进行解压。
将每个训练区域图像输入与该训练区域图像对应的编解码模型,编解码模型对输入的训练区域图像进行处理,得到训练区域图像对应的解码训练区域图像。
编解码模型中,压缩模型对训练区域图像进行压缩处理,得到训练区域图像的训练特征。解压模型对训练区域图像的训练特征进行解码,以得到与该训练区域图像对应的解压训练区域图像。
在S830,根据率失真调整所述编解码模型的参数,所述率失真根据所述解压训练区域图像和所述训练区域图像得到。
也就是说,可以利用所述训练图像,训练编解码模型。
在执行S830时,每次使用参数调整后的编解码模型对训练区域图像进行处理,直到率失真逐渐收敛,从而得到训练完成的编解码模型。
对于一个待处理训练图像,可以利用融合模型对输入该融合模型的各个训练区域图像对应的解压训练区域图像进行拼接处理和优化处理,以得到训练恢复图像。本申请实施例对拼接处理和优化处理的先后顺序不作限定。
优化处理包括对解压训练图像的边沿区域进行调整。
优化处理还可以包括对解压训练图像边沿区域之外的其他区域的调整。
应当理解,优化处理中的调整是对颜色的调整,例如可以调整亮度和色度等。
可以根据训练恢复图像与待处理训练图像的图像失真度,调整融合模型的参数,完成对融合模型的训练。
还可以根据训练恢复图像与待处理训练图像的图像失真度,确定率失真,调整编解码模型的参数。完成编解码模型的训练。
融合模型可以对各个训练区域图像对应的解压训练图像进行拼接,并修改并对位于解压后的训练图像的边沿区域的像素进行调整,以得到训练恢复图像。训练恢复图像也就是恢复后的待处理训练图像。
根据训练恢复图像与待处理训练图像的差异,可以确定图像失真度。可以根据压缩结果中每个像素的平均数据量,确定码率。
根据训练恢复图像与待处理训练图像的图像失真度,对融合模型和每个训练区域图像对应的编解码模型的参数进行调整,以使得码率满足预设条件的情况下,降低图像失真度。
或者,也可以根据码率对融合模型和每个训练区域图像对应的编解码模型的参数进行调整。之后,对融合模型和每个训练区域图像对应的编解码模型的参数进行调整,以使得图像失真度满足预设条件的情况下,降低码率。
或者,可以通过率失真从整体上反映压缩性能。可以对融合模型和每个训练区域图像对应的编解码模型的参数进行调整,以使得率失真最小。
对每个待处理训练图像,每次使用参数调整后的编解码模型、融合模型进行处理,直到率失真逐渐收敛,从而得到训练完成的编解码模型和融合模型。
因此,可以通过“端到端”的方式实现多图像处理系统700中AI模型的训练。
为了提高对图像处理系统700中AI模型的训练的速度,也可以先对编解码模型进行预训练,利用预训练完成的编解码模型训练融合模型。
具体地,可以获取待处理训练图像。可以对待处理训练图像进行划分,以得到多个训练区域图像。
对于每个训练区域图像,可以利用与该训练区域图像对应的编解码模型进行压缩,并对压缩结果进行解压,以确定码率和图像失真率,从而确定率失真。通过调整该编解码模型的参数,以优化率失真。
对大量待处理训练图像进行处理,从而获得大量的训练区域图像,以覆盖每种纹理复杂度的图像。对于每种纹理复杂度,每次使用参数调整后的编解码模型对该纹理复杂度的训练区域图像进行处理,直到率失真逐渐收敛,从而得到预训练完成的各个编解码模型。
利用预训练后的编解码模型对待处理训练图像进行处理,以得到待处理训练图像中各个训练区域图像对应的解压训练区域图像。
利用融合模型,对一个待处理训练图像中各个训练区域图像对应的解压训练图像进行融合,以得到训练恢复图像。
调整融合模型的参数,以最小化训练恢复图像与压缩之前的待处理训练图像之间的差异。
利用调整后的融合模型,对预训练得到的编解码模型对每个待处理训练图像的处理结果进行融合,直到训练恢复图像与压缩之前的待处理训练图像之间的误差逐渐收敛,即得到训练完成的融合模型。
在一些实施例中,在利用预训练得到的编解码模型对融合模型进行训练的过程中,也可以调整编解码模型的参数,从而得到图像处理系统700中的训练完成的各个AI模型。
通过S810至S830,在训练过程中,利用不同纹理复杂度的训练区域图像,对每种纹理复杂度对应的编解码模型进行训练。使用训练完成的编解码模型,能够实现对不同纹理复杂度的区域图像的差异化处理,从而提升对整体图像的压缩性能。
利用训练完成的编解码模型,可以进行图像处理。参见图9的说明。
也就是说,在S830,可以根据解压训练区域图像和训练区域图像,计算码率和图像失真度,从而确定率失真,并根据率失真调整编解码模型的参数,以完成对编解码模型的训练或预训练。之后,采用训练或预训练完成的编解码模型,训练融合模型。
或者,在S830,也可以根据解压训练区域图像,利用融合模型的处理,确定训练恢复图像。根据解压训练区域图像,计算码率。根据训练恢复图像与训练图像,计算图像失真度,从而,根据码率和图像失真度确定率失真。之后,可以根据率失真调整编解码模型和融合模型的参数,以完成对编解码模型和融合模型的训练。
图9是本申请实施例提供的一种图像处理方法的示意性流程图。图9所示的方法900可以由图像处理装置来执行,该图像处理装置可以是移动终端,电脑、服务器等运算能力足以用来图像处理装置。
方法900可以具体应用在图像传输、图形存储等需要对图像进行压缩处理的领域。方法包括S910至S930,下面分别对这些步骤进行详细的描述。
在S910,确定待处理图像中的多个区域图像中每个区域图像对应的纹理复杂度信息。
在S920,根据所述每个区域图像对应的纹理复杂度信息,确定所述每个区域图像对应的图像压缩模型。
其中,不同的纹理复杂度信息对应不同的图像压缩模型。
为了确定区域图像的纹理复杂度,可以计算区域图像每个像素在两个不同方向的差分值。以区域图像的水平方向为x轴,竖直方向为y轴,建立平面直角坐标系。区域图像以一个二维数组的形式存储,则像素(i,j)在x方向的差分值为:
dx(i,j)=p(i,j)-p(i-1,j)
像素(i,j)在x方向的差分值为:
dy(i,j)=p(i,j)-p(i,j-1)
其中,p(i,j)可以是像素(i,j)的亮度,也可以是其他用于表示该像素颜色的参数。
根据这两个方向的差分值计算区域图像中每个像素(i,j)的梯度(也可以称为梯度算子)可以通过向量表示:(dx(i,j),dy(i,j))。
像素(i,j)的梯度大小Grad(i,j)为:
Figure BDA0002610949950000181
从而,可以得到区域图像的梯度大小的平均值:
Figure BDA0002610949950000182
其中,W表示区域图像在x方向的像素数量,H表示区域图像在y方向的像素数量。区域图像的梯度平均值G可以用于表示和评价区域图像的纹理复杂度。
在一些实施例中,每个像素(i,j)的梯度大小(也可以称为梯度长度)Grad(i,j)也可以表示为:
Grad(i,j)=|dx(i,j)|+|dy(i,j)|
上述梯度计算方式中,对x方向、y方向的差分值取绝对值或计算平方值,可以防止出现两个方向的正负相反发生抵消,使得梯度计算结果更准确。
可以根据纹理复杂度与压缩模型的对应关系,确定与每个区域图像的第一纹理复杂度信息对应的图像压缩模型。
纹理复杂度信息与图像压缩模型的对应关系,可以包括两种或两种以上的纹理复杂度信息,以及每种纹理复杂度信息对应的压缩模型。
例如,可以在图像的梯度大小的平均值大于或等于预设值时,确定图像的纹理复杂度为复杂;在图像的梯度大小的平均值小于预设值时,确定图像的纹理复杂度为简单。从而,可以根据图像的梯度大小的平均值,确定与该图像对应的压缩模型。
应当理解,该纹理复杂度与压缩模型的对应关系,与训练方法900中所使用的第一压缩模型时所使用的纹理复杂度与压缩模型的对应关系是相同的。
在S930,利用所述每个区域图像对应的图像压缩模型对所述每个区域图像进行压缩。
通过S910至S930,对于不同纹理复杂度的图像,可以使用与该图像纹理复杂度相对应的图像压缩模型进行对待处理图像中的各个区域图像进行压缩,从而提高对待处理图像的图像压缩效果。
待处理图像可以是一张完整的图像,例如可以是摄像头采集的一张照片,或者视频中的一帧图像。
对待处理图像进行划分,以得到多个区域图像。为了降低划分的复杂程度,该多个区域图像的尺寸可以相同。为了降低压缩后的数据量,该多个区域图像之间不存在重叠的区域。多个区域图像可以包括所述待处理图像中的全部像素,从而降低图像失真度,提高压缩性能。
一张完整的图像的不同区域中,图像的纹理复杂度可能并不相同。例如,在天空、海滩等背景区域,图像的纹理复杂度较低;在包括人物等目标的兴趣区域或前景区域,图像复杂度较高。
通过将待处理图像划分为多个区域图像,对每个区域图像分别使用与该图像纹理复杂度相对应的压缩模型进行图像的压缩,可以对待处理图像中不同纹理复杂度的区域进行与该纹理复杂度相适应的压缩处理,以提高对待处理图像整体的压缩效果。本申请实施例提供了一种更加灵活的图像处理方式。
对待处理图像划分为多个区域图像,并对每个区域图像进行了压缩处理,得到了压缩后的数据。在对压缩后的数据进行解压,以得到待处理图像时,可以分别对每个区域图像对应的压缩后的数据进行解压。
分别使用与每个区域图像对应的图像压缩模型对该区域图像进行压缩,以得到该区域图像的图像特征。对于压缩后得到的图像特征,应当使用于与压缩时的图像压缩模型对应的解压模型进行解压,以得到区域解压图像,区域解压图像也可以理解为进行解压得到的恢复后的区域图像。
对待处理图像中各个区域图像对应的区域解压图像进行拼接,从而可以对待处理图像进行恢复。
进一步地,可以对各个区域解压图像进行优化处理。
优化处理可以包括对一个或多个区域解压图像的边沿区域进行调整。
由于各个区域图像对应的第二图像可以是通过不同的解压模型得到的,在两个相邻的第二图像的边沿区域,在拼接之后可能出现线条不连续或颜色差异等现象。为了使得解压得到的图像与待处理图像之间的图像失真度更小,可以对一个或多个第二图像的边沿区域的像素进行调整。
优化处理还可以包括对区域解压图像的边沿区域之外的其他区域的调整。
可以在拼接之前或之后,进行优化处理,本申请实施例对此不作限定。
可以使用融合模型对第二图像的边沿区域的像素进行拼接处理和优化处理。
通过优化处理,对一个或多个第二图像的边沿区域的像素进行调整,可以进一步降低图像失真度,提高图像压缩效果。
图10是本申请实施例提高的图像处理方法的压缩性能的示意图。
采用图像处理方法900对图像进行压缩和解压,能够实现更好的图像压缩性能。
对待处理图像划分为多个大小相等的区域图像。该多个区域图像包括待处理图像的全部像素。
对于每个区域图像,在图像的梯度大小的平均值大于或等于预设值时,确定图像的纹理复杂度为复杂;在图像的梯度大小的平均值小于预设值时,确定图像的纹理复杂度为简单。从而,可以根据图像的梯度大小的平均值,确定与该图像对应的压缩模型和解压模型,对图像进行压缩,对压缩结果进行解压。
之后,利用融合模型,调整位于解压得到的图像边沿区域的像素,并进行拼接,得到解压后的图像。
使用不同的压缩算法,在Kodak数据集上进行测试。如图10所示,相比于不区分图像复杂度,对待处理图像采用单一的压缩模型和解压模型进行处理的方式,本申请实施例提供的图像处理方法,采用多模型处理的方式,能够在相同码率的情况下,有效提高PSNR,具有更低的图像失真度。
上文结合图1至图10的描述了本申请实施例提供的图像处理系统、图像处理系统所需的AI模型的训练方法以及图像处理方法,下面结合图11至图14,描述本申请实施例的装置实施例。应理解,图像处理系统、图像处理系统所需的AI模型的训练方法以及图像处理方法的描述与装置实施例的描述相互对应,因此,未详细描述的部分可以参见上文的描述。
图11是本申请实施例提供的一种图像处理装置的示意性结构图。
图像处理装置2000包括存储模块2010、处理模块2020。
存储模块2010用于存储程序指令。
当所述程序指令在所述处理器中执行时,处理模块2020用于:
确定待处理图像中的多个区域图像中每个区域图像对应的纹理复杂度信息;
根据所述每个区域图像对应的纹理复杂度信息,确定所述每个区域图像对应的图像压缩模型,其中,不同的纹理复杂度信息对应不同的图像压缩模型;
利用所述每个区域图像对应的图像压缩模型对所述每个区域图像进行压缩。
可选地,处理模块2020还用于,利用与压缩所述每个区域图像的图像压缩模型对应的图像解压模型,对所述每个区域图像压缩后得到的图像特征进行解压缩,以得到所述每个区域图像对应的区域解压图像。
处理模块2020还用于,对所述多个区域解压图像进行拼接处理和优化处理,得到恢复后的待处理图像,所述优化处理包括对所述多个区域解压图像的边沿进行像素调整。
可选地,处理模块2020还用于,计算每个区域图像中每个像素的梯度大小。
处理模块2020还用于,根据所述每个像素的梯度大小,确定每个区域图像的纹理复杂度信息。
可选地,处理模块2020还用于,将所述待处理图像划分为所述多个区域图像,所述多个区域图像不重叠,且所述多个区域图像包括所述待处理图像中的全部像素。
图12是本申请实施例提供的一种神经网络训练装置的示意性结构图。
神经网络训练装置3000包括存储模块3010、处理模块3020。
存储模块3010用于存储程序指令。
当所述程序指令在所述处理器中执行时,处理模块3020用于:
确定训练图像中的多个训练区域图像中每个训练区域图像对应的纹理复杂度信息;
根据所述每个训练区域图像对应的训练纹理复杂度信息,确定所述每个训练区域图像对应的编解码模型,其中,不同的纹理复杂度信息对应不同的编解码模型,每个所述编解码模型用于对输入的所述训练区域图像进行压缩,并对压缩结果进行解压得到多个解压训练区域图像;
根据率失真调整所述编解码模型的参数,所述率失真根据所述解压训练区域图像和所述训练区域图像得到。
可选地,处理模块3020还用于,通过融合模型对所述多个解压训练区域图像进行拼接处理和优化处理,所述优化处理包括对所述多个解压训练区域图像的边沿进行像素调整。
处理模块3020还用于,根据所述训练恢复图像与所述训练图像之间的图像失真度,调整所述融合模型的参数。
可选地,处理模块3020还用于,根据所述训练恢复图像与所述训练图像之间的图像失真度,调整所述编解码模型的参数。
可选地,处理模块3020还用于,计算每个训练区域图像中每个像素的梯度大小。
处理模块3020还用于,根据所述每个像素的梯度大小,确定每个训练区域图像的纹理复杂度信息。
可选地,处理模块3020还用于,将所述训练图像划分为所述多个训练区域图像,所述多个训练区域图像不重叠,且所述多个训练区域图像包括所述训练图像中的全部像素。
图13是本申请实施例的图像处理装置的硬件结构示意图。图13所示的图像处理装置4000包括存储器4001、处理器4002、通信接口4003以及总线4004。其中,存储器4001、处理器4002、通信接口4003通过总线4004实现彼此之间的通信连接。
存储器4001可以是ROM,静态存储设备和RAM。存储器4001可以存储程序,当存储器4001中存储的程序被处理器4002执行时,处理器4002和通信接口4003用于执行本申请实施例的图像处理方法的各个步骤。
处理器4002可以采用通用的,CPU,微处理器,ASIC,GPU或者一个或多个集成电路,用于执行相关程序,以实现本申请实施例的图像处理装置中的单元所需执行的功能,或者执行本申请方法实施例的图像处理方法。
处理器4002还可以是一种集成电路芯片,具有信号的处理能力,例如,可以是图4所示的芯片。在实现过程中,本申请实施例的图像处理方法的各个步骤可以通过处理器4002中的硬件的集成逻辑电路或者软件形式的指令完成。
上述处理器4002还可以是通用处理器、DSP、ASIC、FPGA或者其他可编程逻辑器件、分立门或者晶体管逻辑器件、分立硬件组件。可以实现或者执行本申请实施例中的公开的各方法、步骤及逻辑框图。通用处理器可以是微处理器或者该处理器也可以是任何常规的处理器等。结合本申请实施例所公开的方法的步骤可以直接体现为硬件译码处理器执行完成,或者用译码处理器中的硬件及软件模块组合执行完成。软件模块可以位于随机存储器,闪存、只读存储器,可编程只读存储器或者电可擦写可编程存储器、寄存器等本领域成熟的存储介质中。该存储介质位于存储器4001,处理器4002读取存储器4001中的信息,结合其硬件完成本申请实施例的图像处理装置中包括的单元所需执行的功能,或者执行本申请方法实施例的图像处理方法。
通信接口4003使用例如但不限于收发器一类的收发装置,来实现装置4000与其他设备或通信网络之间的通信。例如,可以通过通信接口4003获取待处理图像。
总线4004可包括在装置4000各个部件(例如,存储器4001、处理器4002、通信接口4003)之间传送信息的通路。
图14是本申请实施例的神经网络训练装置的硬件结构示意图。与上述装置3000和装置4000类似,图14所示的神经网络训练装置5000包括存储器5001、处理器5002、通信接口5003以及总线5004。其中,存储器5001、处理器5002、通信接口5003通过总线5004实现彼此之间的通信连接。
可以通过图14所示的神经网络训练装置5000对该神经网络进行训练,训练得到的神经网络就可以用于执行本申请实施例的图像处理方法了。
具体地,图14所示的装置可以通过通信接口5003从外界获取训练数据以及待训练的神经网络,然后由处理器根据训练数据对待训练的神经网络进行训练。
应注意,尽管上述装置4000和装置5000仅仅示出了存储器、处理器、通信接口,但是在具体实现过程中,本领域的技术人员应当理解,装置4000和装置5000还可以包括实现正常运行所必须的其他器件。同时,根据具体需要,本领域的技术人员应当理解,装置4000和装置5000还可包括实现其他附加功能的硬件器件。此外,本领域的技术人员应当理解,装置4000和装置5000也可仅仅包括实现本申请实施例所必须的器件,而不必包括图13和图14中所示的全部器件。
应理解,本申请实施例中的处理器可以为中央处理单元(central processingunit,CPU),该处理器还可以是其他通用处理器、数字信号处理器(digital signalprocessor,DSP)、专用集成电路(application specific integrated circuit,ASIC)、现成可编程门阵列(field programmable gate array,FPGA)或者其他可编程逻辑器件、分立门或者晶体管逻辑器件、分立硬件组件等。通用处理器可以是微处理器或者该处理器也可以是任何常规的处理器等。
还应理解,本申请实施例中的存储器可以是易失性存储器或非易失性存储器,或可包括易失性和非易失性存储器两者。其中,非易失性存储器可以是只读存储器(read-only memory,ROM)、可编程只读存储器(programmable ROM,PROM)、可擦除可编程只读存储器(erasable PROM,EPROM)、电可擦除可编程只读存储器(electrically EPROM,EEPROM)或闪存。易失性存储器可以是随机存取存储器(random access memory,RAM),其用作外部高速缓存。通过示例性但不是限制性说明,许多形式的随机存取存储器(random accessmemory,RAM)可用,例如静态随机存取存储器(static RAM,SRAM)、动态随机存取存储器(DRAM)、同步动态随机存取存储器(synchronous DRAM,SDRAM)、双倍数据速率同步动态随机存取存储器(double data rate SDRAM,DDR SDRAM)、增强型同步动态随机存取存储器(enhanced SDRAM,ESDRAM)、同步连接动态随机存取存储器(synchlink DRAM,SLDRAM)和直接内存总线随机存取存储器(direct rambus RAM,DR RAM)。
上述实施例,可以全部或部分地通过软件、硬件、固件或其他任意组合来实现。当使用软件实现时,上述实施例可以全部或部分地以计算机程序产品的形式实现。所述计算机程序产品包括一个或多个计算机指令或计算机程序。在计算机上加载或执行所述计算机指令或计算机程序时,全部或部分地产生按照本申请实施例所述的流程或功能。所述计算机可以为通用计算机、专用计算机、计算机网络、或者其他可编程装置。所述计算机指令可以存储在计算机可读存储介质中,或者从一个计算机可读存储介质向另一个计算机可读存储介质传输,例如,所述计算机指令可以从一个网站站点、计算机、服务器或数据中心通过有线(例如红外、无线、微波等)方式向另一个网站站点、计算机、服务器或数据中心进行传输。所述计算机可读存储介质可以是计算机能够存取的任何可用介质或者是包含一个或多个可用介质集合的服务器、数据中心等数据存储设备。所述可用介质可以是磁性介质(例如,软盘、硬盘、磁带)、光介质(例如,DVD)、或者半导体介质。半导体介质可以是固态硬盘。
应理解,本文中术语“和/或”,仅仅是一种描述关联对象的关联关系,表示可以存在三种关系,例如,A和/或B,可以表示:单独存在A,同时存在A和B,单独存在B这三种情况,其中A,B可以是单数或者复数。另外,本文中字符“/”,一般表示前后关联对象是一种“或”的关系,但也可能表示的是一种“和/或”的关系,具体可参考前后文进行理解。
本申请中,“至少一个”是指一个或者多个,“多个”是指两个或两个以上。“以下至少一项(个)”或其类似表达,是指的这些项中的任意组合,包括单项(个)或复数项(个)的任意组合。例如,a,b,或c中的至少一项(个),可以表示:a,b,c,a-b,a-c,b-c,或a-b-c,其中a,b,c可以是单个,也可以是多个。
应理解,在本申请的各种实施例中,上述各过程的序号的大小并不意味着执行顺序的先后,各过程的执行顺序应以其功能和内在逻辑确定,而不应对本申请实施例的实施过程构成任何限定。
本领域普通技术人员可以意识到,结合本文中所公开的实施例描述的各示例的单元及算法步骤,能够以电子硬件、或者计算机软件和电子硬件的结合来实现。这些功能究竟以硬件还是软件方式来执行,取决于技术方案的特定应用和设计约束条件。专业技术人员可以对每个特定的应用来使用不同方法来实现所描述的功能,但是这种实现不应认为超出本申请的范围。
所属领域的技术人员可以清楚地了解到,为描述的方便和简洁,上述描述的系统、装置和单元的具体工作过程,可以参考前述方法实施例中的对应过程,在此不再赘述。
在本申请所提供的几个实施例中,应该理解到,所揭露的系统、装置和方法,可以通过其它的方式实现。例如,以上所描述的装置实施例仅仅是示意性的,例如,所述单元的划分,仅仅为一种逻辑功能划分,实际实现时可以有另外的划分方式,例如多个单元或组件可以结合或者可以集成到另一个系统,或一些特征可以忽略,或不执行。另一点,所显示或讨论的相互之间的耦合或直接耦合或通信连接可以是通过一些接口,装置或单元的间接耦合或通信连接,可以是电性,机械或其它的形式。
所述作为分离部件说明的单元可以是或者也可以不是物理上分开的,作为单元显示的部件可以是或者也可以不是物理单元,即可以位于一个地方,或者也可以分布到多个网络单元上。可以根据实际的需要选择其中的部分或者全部单元来实现本实施例方案的目的。
另外,在本申请各个实施例中的各功能单元可以集成在一个处理单元中,也可以是各个单元单独物理存在,也可以两个或两个以上单元集成在一个单元中。
所述功能如果以软件功能单元的形式实现并作为独立的产品销售或使用时,可以存储在一个计算机可读取存储介质中。基于这样的理解,本申请的技术方案本质上或者说对现有技术做出贡献的部分或者该技术方案的部分可以以软件产品的形式体现出来,该计算机软件产品存储在一个存储介质中,包括若干指令用以使得一台计算机设备(可以是个人计算机,服务器,或者网络设备等)执行本申请各个实施例所述方法的全部或部分步骤。而前述的存储介质包括:U盘、移动硬盘、只读存储器(Read-Only Memory,ROM)、随机存取存储器(Random Access Memory,RAM)、磁碟或者光盘等各种可以存储程序代码的介质。
以上所述,仅为本申请的具体实施方式,但本申请的保护范围并不局限于此,任何熟悉本技术领域的技术人员在本申请揭露的技术范围内,可轻易想到变化或替换,都应涵盖在本申请的保护范围之内。因此,本申请的保护范围应以所述权利要求的保护范围为准。

Claims (18)

1.一种图像处理方法,其特征在于,包括:
确定待处理图像中的多个区域图像中每个区域图像对应的纹理复杂度信息;
根据所述每个区域图像对应的纹理复杂度信息,确定所述每个区域图像对应的图像压缩模型,其中,不同的纹理复杂度信息对应不同的图像压缩模型;
利用所述每个区域图像对应的图像压缩模型对所述每个区域图像进行压缩。
2.根据权利要求1所述的方法,其特征在于,所述方法还包括:
利用与压缩所述每个区域图像的图像压缩模型对应的图像解压模型,对所述每个区域图像压缩后得到的图像特征进行解压缩,得到所述每个区域图像对应的区域解压图像;
对所述多个区域解压图像进行拼接处理和优化处理,得到恢复后的待处理图像,所述优化处理包括对至少一个所述区域解压图像的边沿进行调整。
3.根据权利要求1或2所述的方法,其特征在于,所述确定待处理图像中的多个区域图像中每个区域图像对应的纹理复杂度信息,包括:
计算每个区域图像中每个像素的梯度大小;
根据所述每个像素的梯度大小,确定每个区域图像的纹理复杂度信息。
4.根据权利要求1-3中任一项所述的方法,其特征在于,所述方法还包括:
将所述待处理图像划分为所述多个区域图像,所述多个区域图像不重叠,且所述多个区域图像包括所述待处理图像中的全部像素。
5.一种神经网络训练方法,其特征在于,所述方法包括:
确定训练图像中的多个训练区域图像中每个训练区域图像对应的纹理复杂度信息;
根据所述每个训练区域图像对应的训练纹理复杂度信息,确定所述每个训练区域图像对应的编解码模型,其中,不同的纹理复杂度信息对应不同的编解码模型,每个所述编解码模型用于对输入的所述训练区域图像进行压缩,并对压缩结果进行解压得到多个解压训练区域图像;
根据率失真调整所述编解码模型的参数,所述率失真根据所述解压训练区域图像和所述训练区域图像得到。
6.如权利要求5所述的神经网络训练方法,其特征在于,所述方法还包括:
通过融合模型对所述多个解压训练区域图像进行拼接处理和优化处理,得到训练恢复图像,所述优化处理包括对至少一个所述解压训练区域图像的边沿进行调整;
根据所述训练恢复图像与所述训练图像之间的图像失真度,调整所述编解码模型和所述融合模型的参数。
7.根据权利要求5或6所述的方法,其特征在于,所述确定训练图像中的多个训练区域图像中每个训练区域图像对应的纹理复杂度信息,包括:
计算每个训练区域图像中每个像素的梯度大小;
根据所述每个像素的梯度大小,确定每个训练区域图像的纹理复杂度信息。
8.根据权利要求5-7中任一项所述的方法,其特征在于,所述方法还包括:
将所述训练图像划分为所述多个训练区域图像,所述多个训练区域图像不重叠,且所述多个训练区域图像包括所述训练图像中的全部像素。
9.一种图像处理装置,其特征在于,包括存储模块和处理模块;
所述存储模块用于存储程序指令;
当所述程序指令在所述处理器中执行时,所述处理模块用于:
确定待处理图像中的多个区域图像中每个区域图像对应的纹理复杂度信息;
根据所述每个区域图像对应的纹理复杂度信息,确定所述每个区域图像对应的图像压缩模型,其中,不同的纹理复杂度信息对应不同的图像压缩模型;
利用所述每个区域图像对应的图像压缩模型对所述每个区域图像进行压缩。
10.根据权利要求9所述的装置,其特征在于,所述处理模块还用于:
利用与压缩所述每个区域图像的图像压缩模型对应的图像解压模型,对所述每个区域图像压缩后得到的图像特征进行解压缩,得到所述每个区域图像对应的区域解压图像;
对所述多个区域解压图像进行拼接处理和优化处理,得到恢复后的待处理图像,所述优化处理包括对所述多个区域解压图像的边沿进行像素调整。
11.根据权利要求9或10所述的装置,其特征在于,所述处理模块还用于:
计算每个区域图像中每个像素的梯度大小;
根据所述每个像素的梯度大小,确定每个区域图像的纹理复杂度信息。
12.根据权利要求9-11中任一项所述的装置,其特征在于,所述处理模块还用于:
将所述待处理图像划分为所述多个区域图像,所述多个区域图像不重叠,且所述多个区域图像包括所述待处理图像中的全部像素。
13.一种神经网络训练装置,其特征在于,包括存储模块和处理模块;
所述存储模块用于存储程序指令,
当所述程序指令在所述处理器中执行时,所述处理模块用于:
确定训练图像中的多个训练区域图像中每个训练区域图像对应的纹理复杂度信息;
根据所述每个训练区域图像对应的训练纹理复杂度信息,确定所述每个训练区域图像对应的编解码模型,其中,不同的纹理复杂度信息对应不同的编解码模型,每个所述编解码模型用于对输入的所述训练区域图像进行压缩,并对压缩结果进行解压得到多个解压训练区域图像;
根据率失真调整所述编解码模型的参数,所述率失真根据所述解压训练区域图像和所述训练区域图像得到。
14.根据权利要求13所述的装置,其特征在于,所述处理模块还用于:
通过融合模型对所述多个解压训练区域图像进行拼接处理和优化处理,得到训练恢复图像,所述优化处理包括对至少一个所述解压训练区域图像的边沿进行像素调整;
根据所述训练恢复图像与所述训练图像的图像失真度,调整所述编解码模型和所述融合模型的参数。
15.根据权利要求13或14所述的装置,其特征在于,所述处理模块还用于:
计算每个训练区域图像中每个像素的梯度大小;
根据所述每个像素的梯度大小,确定每个训练区域图像的纹理复杂度信息。
16.根据权利要求13-15中任一项所述的装置,其特征在于,所述处理模块还用于:
将所述训练图像划分为所述多个训练区域图像,所述多个训练区域图像不重叠,且所述多个训练区域图像包括所述训练图像中的全部像素。
17.一种计算机可读存储介质,其特征在于,所述计算机可读介质存储用于设备执行的程序代码,该程序代码包括用于执行如权利要求1至8中任一项所述的方法。
18.一种芯片,其特征在于,所述芯片包括处理器与数据接口,所述处理器通过所述数据接口读取存储器上存储的指令,以执行如权利要求1至8中任一项所述的方法。
CN202010754067.XA 2020-07-30 2020-07-30 图像处理方法与装置、神经网络训练的方法与装置 Pending CN114067007A (zh)

Priority Applications (2)

Application Number Priority Date Filing Date Title
CN202010754067.XA CN114067007A (zh) 2020-07-30 2020-07-30 图像处理方法与装置、神经网络训练的方法与装置
PCT/CN2021/086836 WO2022021938A1 (zh) 2020-07-30 2021-04-13 图像处理方法与装置、神经网络训练的方法与装置

Applications Claiming Priority (1)

Application Number Priority Date Filing Date Title
CN202010754067.XA CN114067007A (zh) 2020-07-30 2020-07-30 图像处理方法与装置、神经网络训练的方法与装置

Publications (1)

Publication Number Publication Date
CN114067007A true CN114067007A (zh) 2022-02-18

Family

ID=80037134

Family Applications (1)

Application Number Title Priority Date Filing Date
CN202010754067.XA Pending CN114067007A (zh) 2020-07-30 2020-07-30 图像处理方法与装置、神经网络训练的方法与装置

Country Status (2)

Country Link
CN (1) CN114067007A (zh)
WO (1) WO2022021938A1 (zh)

Cited By (2)

* Cited by examiner, † Cited by third party
Publication number Priority date Publication date Assignee Title
CN115147501A (zh) * 2022-09-05 2022-10-04 深圳市明源云科技有限公司 图片解压方法、装置、终端设备以及存储介质
CN117793352A (zh) * 2024-02-22 2024-03-29 北京铁力山科技股份有限公司 基于语义理解的视频压缩方法、装置、设备及存储介质

Families Citing this family (3)

* Cited by examiner, † Cited by third party
Publication number Priority date Publication date Assignee Title
CN114491272B (zh) * 2022-02-14 2023-09-12 北京有竹居网络技术有限公司 一种多媒体内容推荐方法及装置
CN115278246B (zh) * 2022-08-01 2024-04-16 天津大学 一种深度图端到端智能压缩编码方法及装置
CN116684607B (zh) * 2023-07-26 2023-11-14 腾讯科技(深圳)有限公司 图像压缩和解压缩方法、装置、电子设备及存储介质

Family Cites Families (4)

* Cited by examiner, † Cited by third party
Publication number Priority date Publication date Assignee Title
CN100576921C (zh) * 2008-01-14 2009-12-30 浙江大学 基于块分类的混合图像压缩方法
CN103700121A (zh) * 2013-12-30 2014-04-02 Tcl集团股份有限公司 一种复合图像的压缩方法及装置
CN108062780B (zh) * 2017-12-29 2019-08-09 百度在线网络技术(北京)有限公司 图像压缩方法和装置
CN109996078B (zh) * 2019-02-25 2021-06-29 创新先进技术有限公司 一种图像压缩方法、装置及电子设备

Cited By (3)

* Cited by examiner, † Cited by third party
Publication number Priority date Publication date Assignee Title
CN115147501A (zh) * 2022-09-05 2022-10-04 深圳市明源云科技有限公司 图片解压方法、装置、终端设备以及存储介质
CN117793352A (zh) * 2024-02-22 2024-03-29 北京铁力山科技股份有限公司 基于语义理解的视频压缩方法、装置、设备及存储介质
CN117793352B (zh) * 2024-02-22 2024-05-10 北京铁力山科技股份有限公司 基于语义理解的视频压缩方法、装置、设备及存储介质

Also Published As

Publication number Publication date
WO2022021938A1 (zh) 2022-02-03

Similar Documents

Publication Publication Date Title
CN110532871B (zh) 图像处理的方法和装置
CN110188795B (zh) 图像分类方法、数据处理方法和装置
WO2020177651A1 (zh) 图像分割方法和图像处理装置
CN110222717B (zh) 图像处理方法和装置
CN114067007A (zh) 图像处理方法与装置、神经网络训练的方法与装置
CN112308200B (zh) 神经网络的搜索方法及装置
CN109993707B (zh) 图像去噪方法和装置
CN111914997B (zh) 训练神经网络的方法、图像处理方法及装置
CN113259665B (zh) 一种图像处理方法以及相关设备
CN113011575A (zh) 神经网络模型更新方法、图像处理方法及装置
CN113066017B (zh) 一种图像增强方法、模型训练方法及设备
CN112529150A (zh) 一种模型结构、模型训练方法、图像增强方法及设备
CN113011562A (zh) 一种模型训练方法及装置
CN112215332A (zh) 神经网络结构的搜索方法、图像处理方法和装置
CN111695673A (zh) 训练神经网络预测器的方法、图像处理方法及装置
CN114698395A (zh) 神经网络模型的量化方法和装置、数据处理的方法和装置
CN113066018A (zh) 一种图像增强方法及相关装置
CN112446835A (zh) 图像恢复方法、图像恢复网络训练方法、装置和存储介质
CN113284055A (zh) 一种图像处理的方法以及装置
CN114915783A (zh) 编码方法和装置
CN114071141A (zh) 一种图像处理方法及其设备
WO2022022176A1 (zh) 一种图像处理方法以及相关设备
CN115409697A (zh) 一种图像处理方法及相关装置
WO2021057091A1 (zh) 视点图像处理方法及相关设备
TWI826160B (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