一种针对弱光图像增强的结构-纹理感知方法
摘要:
本发明提出一种针对弱光图像增强的结构‑纹理感知方法,包括如下步骤:步骤S1:准备训练数据集,从LOL数据集和DPED数据集中提取弱光/正常图像对作为训练集;步骤S2:建立网络模型,利用轮廓图引导滤波器得到输入图像的结构图,再将输入图像减去结构图得到纹理图,将得到的结构图和纹理图分别输入结构关注子网和纹理关注子网;步骤S3:训练网络模型,将步骤S1中得到的数据集按照批量尺寸组合一起作为输入,设置网络模型学习参数,利用混和损失函数采用Adam优化器对网络参数进行优化;步骤S4:图像增强,将弱光场景下拍摄的图像输入构建好的网络方法,加载预训练好的网络参数,进行图像增强处理,得到增强后的图像。
公开/授权文献
0/0