基于Vision Transformer和U-Net网络的计算鬼成像重建方法

本发明公开了基于Vision Transformer和U‑Net网络的计算鬼成像重建方法，按照以下步骤实施：步骤1，将切蛋糕序列的Hadamard矩阵作为相位掩模依次嵌入到空间光调制器上并按序投影到目标图像上，通过桶探测器来收集总光强序列；步骤2，将切蛋糕序列的Hadamard矩阵及光强序列进行分块，将每个块进行线性投影，得到高维向量token，为每个高维向量token添加一个位置编码后输入到Transformer编码器进行处理，将输出转换为二维矩阵；步骤3，将步骤2得到的二维矩阵输入到U‑Net网络中来增强特征信息的传递与融合，直到ViT‑U‑Net网络的损失函数保持稳定，得到计算鬼成像的结果。本发明方法重建的图像质量高且重建效率高。