一种基于扩散模型的传像束光纤图像质量优化方法及系统

    公开(公告)号:CN117455803A

    公开(公告)日:2024-01-26

    申请号:CN202311225526.5

    申请日:2023-09-21

    IPC分类号: G06T5/73 G06T5/50 G06T7/30

    摘要: 本发明提供一种基于扩散模型的传像束光纤图像质量优化方法及系统,所述方法的步骤包括:获取预设的训练数据集,将低质量图输入到优化模型的粗糙去模糊网络中,得到粗糙去模糊图像,基于所述粗糙去模糊图像与高清图计算得到残差图像,将所述残差图像输入到精细去模糊网络中,得到高频解码图像、低频解码图像并得到精细优化图像,基于精细优化图像和粗糙去模糊图像得到输出图像;基于粗糙去模糊图像和高清图计算第一损失函数值,基于高频解码图像、低频解码图像和高清图计算第二损失函数值;基于第一损失函数值和第二损失函数值对优化模型进行训练;获取光纤输出的传像束光纤图像,将传像束光纤图像输入到完成训练的优化模型中,得到优化图像。