一种图像去噪模型的在线训练装置及方法
技术领域
本发明涉及图像处理领域,具体涉及一种图像去噪模型的在线训练装置及方法。
背景技术
图像传感器在获取图像时,由于拍摄场景不同,其获取的图像中包含的噪声信息不同;尤其在低光照环境下,图像中的噪声水平很高,甚至可以和图像信息相比拟,如何在低光照拍摄环境下对图像进行去噪并提取图像信息,是一项很有挑战的工作。
近年来基于卷积神经网络的去噪模型逐渐应用在图像去噪领域,通常来讲,神经网络的训练是采用线下训练的方法,先搜集大量场景,每一场景均采集不同曝光条件下的图像,然后在服务器上完成训练。一般来讲,低光照工作环境有其特定的应用场景,比如工业环境、地下车库等,在不同的拍摄环境中拍摄的图像包含的噪声不同,为了确保去噪的准确性,不同拍摄环境需要配备不同的去噪模型,在训练不同的去噪模型时需要大量的该拍摄环境下不同曝光条件所拍摄的图像。
传统的图像采集方式不利于对具体拍摄环境中训练图像的大量采集,现有的线下训练方式也不能确保训练之后去噪模型的精准度;当验证之后的去噪模型不能满足要求时,不能及时快捷地寻找出该拍摄环境下的训练图像。因此,如何快速准确地训练得到特定拍摄环境下的去噪模型是图像处理领域亟待解决的问题。
发明内容
本发明的目的是提供一种图像去噪模型的在线训练装置及方法,克服了线下训练方法存在的采样图像不能覆盖全空间的问题,确保了去噪模型的精准性。
为了实现上述目的,本发明采用如下技术方案:一种图像去噪模型的在线训练装置,包括摄像头、存储模块和训练模块,所述存储模块分别连接所述摄像头和训练模块,所述摄像头用于拍摄图像,并将拍摄的图像及对应的曝光条件按照不同拍摄环境分类存储在所述存储模块;所述曝光条件包括正常曝光条件和其他曝光条件;
所述训练模块中包括去噪模型,所述存储模块中拍摄环境A中正常曝光条件及对应图像、其他曝光条件及对应图像传输至所述训练模块中对所述去噪模型进行训练,获得拍摄环境A对应的最优去噪模型。
进一步地,所述训练模块包括FPGA和外围接口电路。
进一步地,所述训练模块还包括图像预处理单元和训练调度单元。
进一步地,所述去噪模型为神经网络去噪模型。
进一步地,所述在线训练装置还包括移动载体,所述摄像头位于所述移动载体上,并随着移动载体的移动进行不同拍摄环境下的拍摄。
本发明提供的一种图像去噪模型的在线训练方法,包括如下步骤:
S01:摄像头拍摄图像,并将拍摄的图像及对应的曝光条件按照不同拍摄环境分类存储在存储模块;
S02:拍摄环境A中正常曝光条件及对应图像、其他曝光条件及对应图像传输至训练模块中对去噪模型进行训练;
S03:所述训练模块获得拍摄环境A对应的最优去噪模型。
进一步地,所述训练模块还包括图像预处理单元和训练调度单元,所述步骤S02中拍摄环境A正常曝光条件及对应图像、其他曝光条件及对应图像在所述训练调度单元的调配下进入所述图像预处理单元进行预处理,预处理之后的图像对所述去噪模型进行训练。
进一步地,所述去噪模型为神经网络去噪模型。
进一步地,所述步骤S02中将拍摄环境A中其他曝光条件及对应图像、正常曝光条件作为神经网络去噪模型的输入,正常曝光条件下的图像作为神经网络去噪模型的输出,对神经网络去噪模型进行训练。
进一步地,所述步骤S03中当训练之后的神经网络去噪模型收敛之后,获得拍摄环境A对应的最优去噪模型;否则,继续对神经网络去噪模型进行训练直至收敛。
本发明具有如下有益效果:本发明对特定拍摄环境中不同曝光条件下拍摄的大量图像进行神经网络去噪模型的在线训练;同时去噪模型的训练基于FPGA板卡实现,从而获得特定拍摄环境对应的神经网路去噪模型。本发明克服了线下训练方法可能存在的采样图像不能覆盖全空间的问题,采集的图像充分,保证了训练之后去噪模型的精准性。
附图说明
附图1为本发明一种图像去噪模型的在线训练装置的结构示意图;
附图2为本发明一种图像去噪模型的在线训练方法的流程图。
具体实施方式
为使本发明的目的、技术方案和优点更加清楚,下面结合附图对本发明的具体实施方式做进一步的详细说明。
如附图1所示,本发明提供的一种图像去噪模型的在线训练装置,包括摄像头、存储模块和训练模块,存储模块分别连接摄像头和训练模块,摄像头用于拍摄图像,并将拍摄的图像及对应的曝光条件按照不同拍摄环境分类存储在存储模块,训练模块中包括去噪模型,存储模块中拍摄环境A中正常曝光条件及对应图像、其他曝光条件及对应图像传输至训练模块中对去噪模型进行训练,获得拍摄环境A对应的最优去噪模型。
本发明中摄像头所用的图像传感器即为目标图像传感器,用于采集特定拍摄环境下的图像,不同的拍摄环境对应不同的应用场景。本发明中训练之后的去噪模型可用于对摄像头所在的图像传感器拍摄的图像进行去噪。对于同一拍摄环境,会在不同曝光条件下拍摄多帧图像,作为后续去噪模型的训练图像集。如果该图像传感器会应用于多种拍摄环境,则此设备可以依次工作于相应的拍摄环境,针对同一拍摄环境,在不同曝光条件下拍摄多张图像,供后续去噪模型训练使用。本发明每一种拍摄环境所对应的去噪模型对应的参数并不相同,因此,后续不同拍摄环境中的图像用于训练对应拍摄环境下的去噪模型。本发明中拍摄环境可以为任意的拍摄环境,例如在低光照拍摄环境中拍摄图像的噪声较为明显,可以采用本发明装置和方法训练低光照拍摄环境下对应的去噪模型。
本发明中存储模块可以分为多个存储单元,每一个存储单元用于存储一种拍摄环境中的图像,其中,在同一种拍摄环境下,采用不同的曝光条件进行拍摄,存储单元中同时存储该拍摄环境下的曝光条件以及对应的图像;每一种曝光条件下拍摄多帧图像用于后续去噪模型的训练。每一种拍摄环境的曝光条件包括正常曝光条件和其他曝光条件,其中,具体的曝光条件包括曝光时间、焦距、光照、摄像头光圈大小等等参数。本发明中正常的曝光条件指的是该拍摄环境下,能够得到图像信号强而噪声弱的图像时对应的曝光条件,且正常的曝光条件可以根据图像质量以及图像的应用领域自行设定;在该拍摄环境下,不同于正常曝光条件的其曝光条件统称为其他曝光条件。
本发明中训练模块具体包括FPGA和外围接口电路,外围接口电路用于对去噪模型进行后续的传输以及处理。训练模块还包括图像预处理单元和训练调度单元,在图像预处理单元写入图像预处理算法,在训练调度单元中写入训练调度算法,同时在训练模块中写入去噪模型。这里写入的去噪模型具体可以为神经网络去噪模型,且此时的神经网络去噪模型为初始的模型框架,经过训练之后才能形成具体对应的模型参数。上述图像预处理算法对拍摄得到的图像按需要进行预处理;训练调度算法调度图像和FPGA相关资源,确保训练过程顺利完成。
在线训练装置还包括移动载体,摄像头位于移动载体上,并随着移动载体的移动进行不同拍摄环境的拍摄。在特定的拍摄环境中,在摄像头采集图像数据的同时进行去噪模型的训练。并且如果该图像传感器需要工作在多个低光环境或者场景变化较大的同一低光环境中,可以将本装置放于移动载体上,在所要求的拍摄环境中不断变换所在位置,在采集不同位置图像数据的同时,完成去噪模型的训练,使算法能够在不同的拍摄环境中都取得好的去噪效果。
如附图2所示,本发明提供的一种图像去噪模型的在线训练方法,包括如下步骤:
S01:摄像头拍摄图像,并将拍摄的图像及对应的曝光条件按照不同拍摄环境分类存储在存储模块。存储模块可以分为多个存储单元,每一个存储单元用于存储一种拍摄环境中的图像,其中,在同一种拍摄环境下,采用不同的曝光条件进行拍摄,存储单元中同时存储该拍摄环境下的曝光条件以及对应的图像;每一种曝光条件下拍摄多帧图像用于后续去噪模型的训练。每一种拍摄环境的曝光条件包括正常曝光条件和其他曝光条件。
S02:拍摄环境A中正常曝光条件及对应图像、其他曝光条件及对应图像传输至训练模块中对去噪模型进行训练。
训练模块还包括图像预处理单元和训练调度单元,在图像预处理单元写入图像预处理算法,在训练调度单元中写入训练调度算法;图像预处理算法对拍摄得到的图像按需要进行预处理;训练调度算法调度图像和FPGA相关资源,确保训练过程顺利完成。用于训练的图像在训练调度单元的调配下进入图像预处理单元进行预处理,预处理之后的图像再对去噪模型进行训练。
本发明中去噪模型为神经网络去噪模型。将拍摄环境A中其他曝光条件EX2、其他曝光条件拍摄的图像P2、正常曝光条件EX1作为神经网络去噪模型的输入,正常曝光条件下的图像P1作为神经网络去噪模型的输出,对神经网络去噪模型进行训练。上述训练过程为循环的多次训练,选取的其他曝光条件及其对应的图像越多,最终得到的神经网络去噪模型越精确。
本发明中拍摄环境为低光照场景时,其他曝光条件选取为低光照曝光条件。
S03:训练模块获得拍摄环境A对应的最优去噪模型。
当训练之后的神经网络去噪模型收敛之后,获得拍摄环境A对应的最优去噪模型;否则,继续对神经网络去噪模型进行训练直至收敛。神经网络去噪模型的收敛标准可以采用神经网络去噪模型的损失函数结果来确定,神经网络去噪模型的损失函数结果趋于稳定时,且通过人工查看图像去躁结果满足需求时,即可判断神经网络去噪模型收敛,训练结束;否则,需要进一步获取更多的训练图像进行训练。
本发明对特定拍摄环境中不同曝光条件下拍摄的大量图像进行神经网络去噪模型的在线训练;同时去噪模型的训练基于FPGA板卡实现,从而获得特定拍摄环境对应的神经网路去噪模型。本发明克服了线下训练方法可能存在的采样图像不能覆盖全空间的问题,采集的图像充分,保证了训练之后去噪模型的精准性。
以上所述仅为本发明的优选实施例,所述实施例并非用于限制本发明的专利保护范围,因此凡是运用本发明的说明书及附图内容所作的等同结构变化,同理均应包含在本发明所附权利要求的保护范围内。