GMA:学习使用全局运动聚合（ArXiv）估计隐藏运动

学习使用全局运动聚合（GMA）估计隐藏的运动 环境环境 您将必须选择cudatoolkit版本以匹配您的计算环境。 该代码已在PyTorch 1.8.0上进行了测试，但其他版本也可以使用。 conda create --name gma python==3.7 conda activate gma conda install pytorch=1.8.0 torchvision=0.9.0 cudatoolkit=11.1 -c pytorch -c conda-forge pip install matplotlib imageio einops scipy opencv-python 演示版 sh demo.sh 执照 WTFPL。 请参阅文件。 致谢 整个代码框架改编自 。 我们感谢作者的贡献。 我们还要感谢开源实现变压器。