摘要
本申请提供了一种光声图像重建模型训练方法、装置、设备及介质。包括:获取模型训练样本;将模型训练样本输入至待训练的光声图像重建模型;调用离散变换层对稀疏光声图像进行离散二维小波变换,得到低频特征图和高频特征图;调用高频通道注意力层对高频特征图进行垂直方向和水平方向上的特征捕获,得到高频注意力特征图;调用低频通道注意力层对低频特征图进行处理,得到低频注意力特征图;对高频注意力特征图、低频注意力特征图和稀疏光声图像进行图像重建得到重建图像;基于重建图像和作为真值的密集采样光声图像计算得到待训练的光声图像重建模型的损失值;在损失值处于预设范围内的情况下,得到最终的光声图像重建模型。
技术关键词
图像重建
高频特征
分量特征
水平高频分量
模型训练方法
注意力机制
变换特征
卷积特征
通道
模型训练装置
网络结构
电子设备
图像获取模块
处理器
输入模块
可读存储介质
存储器