本发明涉及计算机成像,特别涉及一种图像重建模型的训练方法、主控设备及图像重建方法。
背景技术:
1、离散余弦单像素成像是一种正交变换单像素成像方法,基于二维离散余弦变换,采集成像物体图像的离散余弦变换频谱系数,并通过二维离散余弦反变换实现图像重建。在离散余弦单像素成像中,需要将一组不同频率的余弦条纹结构图案加载到空间光调制器中,对成像物体的光空间信息进行调制,并使用单像素探测器探测条纹图案对应的信号值,根据信号值构造物体图像的离散余弦变换系数,并组成离散余弦变换谱,再对采集所得的离散余弦变换谱执行反变换,完成图像重建。一般而言,图像的离散余弦变换谱的能量,即数值绝对值较大的系数值,更多集中在低频部分。因此,离散余弦单像素成像通常只采集一部分绝对值较大的系数值,减少采集次数,以此缩短采集图像所需的时间,实现图像的欠采样方案。
2、然而,通过欠采样方案实现离散余弦单像素成像时,成像的图像频谱是残缺的。且由于大量频谱系数的缺失,会造成图像出现图像模糊和振铃效应等问题,导致图像质量较差。
技术实现思路
1、本发明的主要目的是提出一种图像重建模型的训练方法、主控设备及图像重建方法,旨在解决极低采样率条件下离散余弦单像素成像方法的图像质量较差的问题。
2、为实现上述目的,本发明提出一种图像重建模型的训练方法,所述图像重建模型的训练方法包括:
3、构建训练集,所述训练集包括欠采样图像和与所述欠采样图像相对应的图像频谱;
4、将所述欠采样图像输入初始网络模型以得到目标图像;
5、根据所述目标图像和与所述目标图像相对应的图像频谱计算损失值;
6、根据所述损失值优化所述初始网络模型;以及
7、当所述初始网络模型的优化次数达到目标值时,输出优化后的初始网络模型作为所述图像重建模型。
8、优选地,根据所述目标图像和与所述目标图像相对应的图像频谱计算损失值包括:
9、根据所述目标图像的离散余弦变换频谱和相对应的图像频谱计算所述损失值。
10、优选地,将所述欠采样图像输入初始网络模型以得到目标图像之前,所述图像重建模型的训练方法还包括:
11、构建所述初始网络模型,所述初始网络模型包括特征提取模块、残差模块和重建模块,所述残差模块包括若干依次连接的残差块。
12、优选地,将所述欠采样图像输入初始网络模型以得到目标图像包括:
13、将所述欠采样图像输入所述特征提取模块以得到低层特征;
14、将所述低层特征输入所述残差模块以得到残差特征;
15、将所述残差特征输入所述重建模块以得到所述目标图像。
16、优选地,将所述低层特征输入所述残差模块以得到残差特征包括:
17、将所述低层特征输入第一个所述残差块以得到输出特征;
18、将所述低层特征和所述输出特征作为输入特征共同输入第二个所述残差块,并从第二个所述残差块开始,将前一所述残差块的输出特征和输入特征共同输入后一所述残差块,直至最后一个所述残差块根据所述输入特征输出高层特征;
19、根据所述低层特征和所述高层特征计算所述残差特征。
20、优选地,构建训练集包括:
21、对彩色图像进行预处理以得到目标灰度图像;
22、对所述目标灰度图像进行离散余弦变换处理以得到所述图像频谱;
23、将所述图像频谱中的部分频谱系数设置为预设值以得到欠采样频谱;
24、对所述欠采样频谱进行离散余弦反变换处理以得到所述欠采样图像。
25、优选地,对彩色图像进行预处理以得到目标灰度图像包括:
26、将所述彩色图像转换为初始灰度图像;
27、将所述初始灰度图像进行归一化处理以得到归一化灰度图像;以及
28、对所述归一化灰度图像进行裁切以得到所述目标灰度图像。
29、优选地,对所述欠采样频谱进行离散余弦反变换处理以得到所述欠采样图像之后,所述图像重建模型的训练方法还包括:
30、将所述欠采样图像的灰度值进行归一化。
31、本发明进一步提出一种主控设备,所述主控设备包括:
32、存储器,用于存储程序指令;以及
33、处理器,用于执行所述程序指令以实现如上所述的图像重建模型的训练方法。
34、本发明进一步还提出一种图像重建方法,所述图像重建方法包括:
35、接收单像素成像装置发送的离散余弦变换谱;
36、对所述离散余弦变换谱进行离散余弦反变换处理以得到待重建图像;
37、将所述待重建图像输入图像重建模型以得到目标重建图像,所述图像重建模型由如上所述的图像重建模型的训练方法训练得到。
38、本发明技术方案的有益效果在于:将初始网络模型构建为卷积神经网络,并构建具有欠采样图像的训练集,通过训练集对初始网络模型进行迭代训练,以恢复欠采样图像的完整变换频谱为目标,基于深度学习方法,将训练完成的图像重建模型作为后处理器,使得图像重建模型具有恢复图像频谱的能力,能够尽可能地恢复离散余弦单像素成像的欠采样图像的频谱,明显去除欠采样图像的质量问题,显著提高离散余弦单像素成像的成像质量。
1.一种图像重建模型的训练方法,其特征在于,所述图像重建模型的训练方法包括:
2.根据权利要求1所述的图像重建模型的训练方法,其特征在于,将所述欠采样图像输入初始网络模型以得到目标图像之前,所述图像重建模型的训练方法还包括:
3.根据权利要求2所述的图像重建模型的训练方法,其特征在于,将所述欠采样图像输入初始网络模型以得到目标图像包括:
4.根据权利要求3所述的图像重建模型的训练方法,其特征在于,将所述低层特征输入所述残差模块以得到残差特征包括:
5.根据权利要求1所述的图像重建模型的训练方法,其特征在于,构建训练集包括:
6.根据权利要求5所述的图像重建模型的训练方法,其特征在于,对彩色图像进行预处理以得到目标灰度图像包括:
7.根据权利要求5所述的图像重建模型的训练方法,其特征在于,对所述欠采样频谱进行离散余弦反变换处理以得到所述欠采样图像之后,所述图像重建模型的训练方法还包括:
8.一种主控设备,其特征在于,所述主控设备包括:
9.一种图像重建方法,其特征在于,所述图像重建方法包括: