利索能及
我要发布
收藏
专利号: 2019100989929
申请人: 深圳市商汤科技有限公司
专利类型:发明专利
专利状态:已下证
更新日期:2026-05-14
缴费截止日期: 暂无
联系人

摘要:

权利要求书:

1.一种图像超分辨率重建方法,其特征在于,所述方法包括:根据待重建原始图像进行数据生成处理,得到中间图像和第一彩色图像;

基于预设超分辨率神经网络对所述中间图像和所述第一彩色图像进行重构,得到重建后的图像;

其中,所述预设超分辨率神经网络包括预设图像恢复模型和预设颜色校正模型,所述基于预设超分辨率神经网络对所述中间图像和所述第一彩色图像进行重构,得到所述重建后的图像,包括:将所述中间图像输入至所述预设图像恢复模型中,输出第一线性彩色图像;其中,所述第一线性彩色图像用于表征线性彩色图像预测;

将所述第一彩色图像输入至所述预设颜色校正模型中,输出全局校正参数和单像素校正参数;

根据所述第一线性彩色图像、所述全局校正参数以及所述单像素校正参数,获得所述重建后的图像。

2.根据权利要求1所述的方法,其特征在于,所述根据待重建原始图像进行数据生成处理,得到中间图像和第一彩色图像,包括:对所述待重建原始图像进行预处理,获得所述中间图像;

对所述中间图像进行转换处理,获得所述第一彩色图像。

3.根据权利要求1所述的方法,其特征在于,所述根据待重建原始图像进行数据生成处理,得到中间图像和第一彩色图像之前,所述方法还包括:获取所述待重建原始图像。

4.根据权利要求3所述的方法,其特征在于,所述获取所述待重建原始图像之后,所述方法还包括:根据所述待重建原始图像进行数据生成处理,得到第二彩色图像和第二线性彩色图像;其中,所述第二彩色图像的分辨率高于所述第一彩色图像的分辨率。

5.根据权利要求4所述的方法,其特征在于,所述根据所述待重建原始图像进行数据生成处理,得到第二彩色图像,包括:对所述待重建原始图像进行转换处理,获得所述第二彩色图像。

6.根据权利要求4所述的方法,其特征在于,所述根据所述中间图像和所述第一彩色图像,对所述待重建原始图像进行图像超分辨率重建,获得重建后的图像之后,所述方法还包括:对所述第二彩色图像和所述重建后的图像进行对比,获得第一对比结果;

根据所述第一对比结果对所述预设超分辨率神经网络进行校正。

7.根据权利要求4所述的方法,其特征在于,所述将所述中间图像输入至所述预设图像恢复模型中,输出第一线性彩色图像之后,所述方法还包括:对所述第一线性彩色图像和所述第二线性彩色图像进行对比,获得第二对比结果;

根据所述第二对比结果对所述预设超分辨率神经网络进行校正。

8.根据权利要求2所述的方法,其特征在于,所述对所述中间图像进行转换处理,获得所述第一彩色图像之后,所述方法还包括:根据所述中间图像和所述预设图像恢复模型,获得第一图像特征;

根据所述第一彩色图像和所述预设颜色校正模型,获得第二图像特征;

根据所述第一图像特征和所述第二图像特征进行特征融合处理。

9.一种图像超分辨率重建装置,其特征在于,所述图像超分辨率重建装置包括:获取单元和重建单元,所述获取单元,用于根据待重建原始图像进行数据生成处理,得到中间图像和第一彩色图像;

所述重建单元,用于基于预设超分辨率神经网络对所述中间图像和所述第一彩色图像进行重构,得到重建后的图像;

其中,所述预设超分辨率神经网络包括预设图像恢复模型和预设颜色校正模型;所述重建单元,还用于将将所述中间图像输入至所述预设图像恢复模型中,输出第一线性彩色图像;其中,所述第一线性彩色图像用于表征线性彩色图像预测;以及将所述第一彩色图像输入至所述预设颜色校正模型中,输出全局校正参数和单像素校正参数;以及根据所述第一线性彩色图像、所述全局校正参数以及所述单像素校正参数,获得所述重建后的图像。

10.根据权利要求9所述的图像超分辨率重建装置,其特征在于,所述获取单元,具体用于对所述待重建原始图像进行预处理,获得所述中间图像;以及对所述中间图像进行转换处理,获得所述第一彩色图像。

11.根据权利要求9所述的图像超分辨率重建装置,其特征在于,所述获取单元,还用于根据待重建原始图像进行数据生成处理,得到中间图像和第一彩色图像之前,获取所述待重建原始图像。

12.根据权利要求11所述的图像超分辨率重建装置,其特征在于,所述获取单元,还用于获取所述待重建原始图像之后,根据所述待重建原始图像进行数据生成处理,得到第二彩色图像和第二线性彩色图像;其中,所述第二彩色图像的分辨率高于所述第一彩色图像的分辨率。

13.根据权利要求12所述的图像超分辨率重建装置,其特征在于,所述获取单元,具体用于对所述待重建原始图像进行转换处理,获得所述第二彩色图像。

14.根据权利要求12所述的图像超分辨率重建装置,其特征在于,所述图像超分辨率重建装置还包括:对比单元和校正单元,所述对比单元,用于根据所述中间图像和所述第一彩色图像,对所述待重建原始图像进行图像超分辨率重建,获得重建后的图像之后,对所述第二彩色图像和所述重建后的图像进行对比,获得第一对比结果;

所述校正单元,用于根据所述第一对比结果对所述预设超分辨率神经网络进行校正。

15.根据权利要求14所述的图像超分辨率重建装置,其特征在于,所述对比单元,还用于将所述中间图像输入至所述预设图像恢复模型中,输出第一线性彩色图像之后,对所述第一线性彩色图像和所述第二线性彩色图像进行对比,获得第二对比结果;

所述校正单元,还用于根据所述第二对比结果对所述预设超分辨率神经网络进行校正。

16.根据权利要求10所述的图像超分辨率重建装置,其特征在于,所述图像超分辨率重建装置还包括:融合单元,所述获取单元,还用于对所述中间图像进行转换处理,获得所述第一彩色图像之后,根据所述中间图像和所述预设图像恢复模型,获得第一图像特征;以及根据所述第一彩色图像和所述预设颜色校正模型,获得第二图像特征;

所述融合单元,用于根据所述第一图像特征和所述第二图像特征进行特征融合处理。

17.一种图像超分辨率重建装置,其特征在于,所述图像超分辨率重建装置包括:处理器、存储器及通信总线,所述处理器执行所述存储器中存储的运行程序时实现如权利要求

1‑8任一项所述的方法。

18.一种计算机可读存储介质,其上存储有计算机程序,应用于图像超分辨率重建装置,其特征在于,该计算机程序被处理器执行时实现如权利要求1‑8任一项所述的方法。