一种融入自注意力机制的实时中间流估计算法

本发明公开了一种融入自注意力机制的实时中间流估计算法，涉及计算机视觉技术领域，包括以下步骤：获取相邻的两帧视频图像和；构建图像金字塔，对输入帧进行降采样处理；使用改进的IFNet估计中间流和，其中IFNet采用"粗到细"的光流估计策略，且在IFBlock模块中引入自注意力机制；使用估计的中间流对输入帧进行反向光流扭曲，生成初步中间帧；使用FusionNet融合多源信息，生成最终的中间帧，其中FusionNet在编码器中引入自注意力机制。该融入自注意力机制的实时中间流估计算法，通过引入自注意力机制，模型能够捕捉图像中的全局运动信息，提高光流估计的准确性。