本申请属于图像处理,具体涉及一种图像处理方法、装置及电子设备。
背景技术:
1、水下摄影具有重要的实用价值和艺术价值,可以用于科学研究、商业应用、生态保护和艺术创作等领域。
2、但是,水下复杂的成像环境会导致水下采集的图像质量较差。例如,水下光线吸收和光线折射会导致图像偏绿,水中存在漂浮物等会导致图像发灰、对比度太低。因此,目前水下采集的图像的质量不尽人意,与人眼直接看到的景象相差较大。
技术实现思路
1、本申请实施例的目的是提供一种图像处理方法、装置及电子设备,能够有效地对水下采集的图像进行色彩和对比度的全面矫正,提高图像质量,使图像更接近人眼直接看到的效果。
2、第一方面,本申请实施例提供了一种图像处理方法,该方法包括:
3、获取第一图像,第一图像为水下采集的图像;
4、基于色彩矫正模型对第一图像进行色彩矫正,得到第二图像;
5、基于亮度矫正模型对第二图像进行亮度矫正,得到第三图像;
6、基于细节矫正模型对第三图像的色彩细节和亮度细节进行矫正,得到第四图像;
7、其中,色彩矫正模型、亮度矫正模型和细节矫正模型均是基于水下采集的图像训练得到的。
8、第二方面,本申请实施例提供了一种图像处理装置,该装置包括:
9、图像获取模块,用于获取第一图像,第一图像为水下采集的图像;
10、色彩矫正模块,用于基于色彩矫正模型对第一图像进行色彩矫正,得到第二图像;
11、亮度矫正模块,用于基于亮度矫正模型对第二图像进行亮度矫正,得到第三图像;
12、细节矫正模块,用于基于细节矫正模型对第三图像的色彩细节和亮度细节进行矫正,得到第四图像;
13、其中,色彩矫正模型、亮度矫正模型和细节矫正模型均是基于水下采集的图像训练得到的。
14、第三方面,本申请实施例提供了一种电子设备,该电子设备包括处理器和存储器,所述存储器存储可在所述处理器上运行的程序或指令,所述程序或指令被所述处理器执行时实现如第一方面所述的方法的步骤。
15、第四方面,本申请实施例提供了一种可读存储介质,所述可读存储介质上存储程序或指令,所述程序或指令被处理器执行时实现如第一方面所述的方法的步骤。
16、第五方面,本申请实施例提供了一种芯片,所述芯片包括处理器和通信接口,所述通信接口和所述处理器耦合,所述处理器用于运行程序或指令,实现如第一方面所述的方法。
17、第六方面,本申请实施例提供一种计算机程序产品,该程序产品被存储在存储介质中,该程序产品被至少一个处理器执行以实现如第一方面所述的方法。
18、在本申请实施例中,能够基于色彩矫正模型对水下采集的第一图像进行色彩矫正,得到第二图像,色彩矫正模型能对图像的白平衡和色彩进行初步的矫正;然后基于亮度矫正模型对第二图像进行亮度矫正,得到第三图像,亮度矫正模型能改善图像整体的对比度;最后再基于细节矫正模型对第三图像的色彩细节和亮度细节进行矫正,得到第四图像。该色彩矫正模型、亮度矫正模型和细节矫正模型均是基于水下采集的图像训练得到的,因此,本申请实施例可以有效地对水下采集的图像进行色彩和对比度的全面矫正,提高图像质量,使图像更接近人眼直接看到的效果。
1.一种图像处理方法,其特征在于,所述方法包括:
2.根据权利要求1所述的方法,其特征在于,所述基于色彩矫正模型对所述第一图像进行色彩矫正,得到第二图像,包括:
3.根据权利要求2所述的方法,其特征在于,在所述基于所述n个不同水深等级分别对应的预设色彩矫正系数和n个所述第一色彩矫正系数,确定第二色彩矫正系数之前,所述方法还包括:
4.根据权利要求1所述的方法,其特征在于,所述基于亮度矫正模型对所述第二图像进行亮度矫正,得到第三图像,包括:
5.根据权利要求4所述的方法,其特征在于,所述将所述第二图像输入所述亮度矫正模型进行亮度矫正系数提取,得到所述第二图像对应的亮度矫正系数曲线,包括:
6.根据权利要求1所述的方法,其特征在于,所述基于细节矫正模型对所述第三图像的色彩细节和亮度细节进行矫正,得到第四图像,包括:
7.根据权利要求1所述的方法,其特征在于,在所述基于色彩矫正模型对所述第一图像进行色彩矫正,得到第二图像之前,所述方法还包括:
8.根据权利要求1所述的方法,其特征在于,在所述基于亮度矫正模型对所述第二图像进行亮度矫正,得到第三图像之前,所述方法还包括:
9.根据权利要求1所述的方法,其特征在于,在所述基于细节矫正模型对所述第三图像的色彩细节和亮度细节进行矫正,得到第四图像之前,所述方法还包括:
10.一种图像处理装置,其特征在于,所述装置包括:
11.一种电子设备,其特征在于,包括处理器和存储器,所述存储器存储可在所述处理器上运行的程序或指令,所述程序或指令被所述处理器执行时实现如权利要求1-9任一项所述的图像处理方法的步骤。