基于改进LiteFlowNet3网络的轻量化光流估计方法

现有的光流估计方法参数量大,计算耗时,难以满足实时性需求。为此,提出一种基于改进LiteFlowNet3网络的轻量化光流估计方法。使用池化和深度可分离卷积替换LiteFlowNet3网络中的常规卷积层,大幅减少网络模型参数量。在损失函数中增加了光流梯度的损失,强调训练中对光流边界的监督,在不增加参数的前提下提升性能。改进的LiteFlowNet3网络参数量仅为0.78 M。实验结果表明,改进的LiteFlowNet3光流估计方法在Sintel数据集的Clean和Final序列上的端点误差分别为2.69和4.12,单次推理时间仅为25 ms,性能优于其他的轻量级光流方法。