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

摘要:

权利要求书:

1.一种图像去噪方法,其特征在于,包括:

通过梯度信息预测网络对带噪声图像进行图像处理,得到所述带噪声图像对应的第一预测梯度信息,其中,所述第一预测梯度信息用于表示所述带噪声图像对应的清晰图像中像素点的梯度变化信息;

根据所述第一预测梯度信息,对所述带噪声图像中的像素点进行分类,得到所述带噪声图像中各像素点对应的第一分类结果;

根据所述第一分类结果,对所述带噪声图像进行去噪,得到所述带噪声图像对应的去噪后图像。

2.根据权利要求1所述的方法,其特征在于,所述根据所述第一预测梯度信息,对所述带噪声图像中的像素点进行分类,得到所述带噪声图像中各像素点对应的第一分类结果,包括:针对所述带噪声图像中的任一像素点,根据所述第一预测梯度信息,确定该像素点的梯度特征值;

根据该像素点的梯度特征值所处的梯度特征值区间,确定该像素点对应的梯度类别,其中,不同梯度类别对应不同的梯度特征值区间;

根据各像素点对应的梯度类别,确定所述第一分类结果。

3.根据权利要求1或2所述的方法,其特征在于,所述根据所述第一分类结果,对所述带噪声图像进行去噪,得到所述带噪声图像对应的去噪后图像,包括:基于所述第一分类结果,通过分类去噪网络对不同类别的像素点进行分类卷积去噪,得到所述带噪声图像对应的去噪后图像。

4.根据权利要求3所述的方法,其特征在于,所述分类去噪网络包括多个卷积层;

所述基于所述第一分类结果,通过分类去噪网络对不同类别的像素点进行分类卷积去噪,得到所述带噪声图像对应的去噪后图像,包括:基于所述第一分类结果,在所述分类去噪网络的至少一个卷积层中对不同类别的像素点进行分类卷积去噪,得到所述带噪声图像对应的去噪后图像。

5.根据权利要求1至4任一项所述的方法,其特征在于,所述方法还包括:通过所述梯度信息预测网络对训练集中包括的带噪声样本图像进行图像处理,得到所述带噪声样本图像对应的第二预测梯度信息,其中,所述第二预测梯度信息用于表示所述带噪声样本图像对应的清晰图像中像素点的梯度变化信息,所述训练集中还包括所述带噪声样本图像对应的清晰图像的参考梯度信息;

基于所述第二预测梯度信息和所述参考梯度信息之间的差异,调整所述梯度信息预测网络的网络参数。

6.根据权利要求5所述的方法,其特征在于,所述训练集中还包括所述带噪声样本图像对应的清晰图像;

所述方法还包括:

通过对所述带噪声样本图像对应的清晰图像进行结构张量分析,确定所述参考梯度信息。

7.根据权利要求5所述的方法,其特征在于,所述训练集中还包括所述带噪声样本图像对应的清晰图像;

所述方法还包括:

根据所述第二预测梯度信息,对所述带噪声样本图像中的像素点进行分类,得到所述带噪声样本图像中各像素点对应的第二分类结果;

基于所述第二分类结果,通过所述分类去噪网络对不同类别的像素点进行分类卷积去噪,得到所述带噪声样本图像对应的去噪后图像;

基于所述带噪声样本图像对应的去噪后图像和所述带噪声样本图像对应的清晰图像之间的差异,调整所述分类去噪网络的网络参数。

8.根据权利要求1-7任一项所述的方法,其特征在于,梯度信息包括下述至少一种:梯度方向、梯度强度和梯度一致度。

9.一种图像去噪装置,其特征在于,包括:

第一梯度信息预测模块,用于通过梯度信息预测网络对带噪声图像进行图像处理,得到所述带噪声图像对应的第一预测梯度信息,其中,所述第一预测梯度信息用于表示所述带噪声图像对应的清晰图像中像素点的梯度变化信息;

第一分类模块,用于根据所述第一预测梯度信息,对所述带噪声图像中的像素点进行分类,得到所述带噪声图像中各像素点对应的第一分类结果;

第一去噪模块,用于根据所述第一分类结果,对所述带噪声图像进行去噪,得到所述带噪声图像对应的去噪后图像。

10.根据权利要求9所述的装置,其特征在于,所述第一分类模块,包括:第一确定子模块,用于针对所述带噪声图像中的任一像素点,根据所述第一预测梯度信息,确定该像素点的梯度特征值;

第二确定子模块,用于根据该像素点的梯度特征值所处的梯度特征值区间,确定该像素点对应的梯度类别,其中,不同梯度类别对应不同的梯度特征值区间;

第三确定子模块,用于根据各像素点对应的梯度类别,确定所述第一分类结果。

11.根据权利要求9或10所述的装置,其特征在于,所述第一去噪模块具体用于:基于所述第一分类结果,通过分类去噪网络对不同类别的像素点进行分类卷积去噪,得到所述带噪声图像对应的去噪后图像。

12.根据权利要求11所述的装置,其特征在于,所述分类去噪网络包括多个卷积层;

所述第一去噪模块具体用于:

基于所述第一分类结果,在所述分类去噪网络的至少一个卷积层中对不同类别的像素点进行分类卷积去噪,得到所述带噪声图像对应的去噪后图像。

13.根据权利要求9至12任一项所述的装置,其特征在于,所述装置还包括:第二梯度信息预测模块,用于通过所述梯度信息预测网络对训练集中包括的带噪声样本图像进行图像处理,得到所述带噪声样本图像对应的第二预测梯度信息,其中,所述第二预测梯度信息用于表示所述带噪声样本图像对应的清晰图像中像素点的梯度变化信息,所述训练集中还包括所述带噪声样本图像对应的清晰图像的参考梯度信息;

第一参数调整模块,用于基于所述第二预测梯度信息和所述参考梯度信息之间的差异,调整所述梯度信息预测网络的网络参数。

14.根据权利要求13所述的装置,其特征在于,所述训练集中还包括所述带噪声样本图像对应的清晰图像;

所述装置还包括:

结构张量分析模块,用于通过对所述带噪声样本图像对应的清晰图像进行结构张量分析,确定所述参考梯度信息。

15.根据权利要求13所述的装置,其特征在于,所述训练集中还包括所述带噪声样本图像对应的清晰图像;

所述装置还包括:

第二分类模块,用于根据所述第二预测梯度信息,对所述带噪声样本图像中的像素点进行分类,得到所述带噪声样本图像中各像素点对应的第二分类结果;

第二去噪模块,用于基于所述第二分类结果,通过所述分类去噪网络对不同类别的像素点进行分类卷积去噪,得到所述带噪声样本图像对应的去噪后图像;

第二网络参数调整模块,用于基于所述带噪声样本图像对应的去噪后图像和所述带噪声样本图像对应的清晰图像之间的差异,调整所述分类去噪网络的网络参数。

16.根据权利要求9-15任一项所述的装置,其特征在于,梯度信息包括下述至少一种:梯度方向、梯度强度和梯度一致度。

17.一种电子设备,其特征在于,包括:

处理器;

用于存储处理器可执行指令的存储器;

其中,所述处理器被配置为调用所述存储器存储的指令,以执行权利要求1至8中任意一项所述的方法。

18.一种计算机可读存储介质,其上存储有计算机程序指令,其特征在于,所述计算机程序指令被处理器执行时实现权利要求1至8中任意一项所述的方法。