【文件属性】:
文件名称:GMA:学习使用全局运动聚合(ArXiv)估计隐藏运动
文件大小:22.24MB
文件格式:ZIP
更新时间:2021-04-07 18:36:50
Python
学习使用全局运动聚合(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。 请参阅文件。
致谢
整个代码框架改编自 。 我们感谢作者的贡献。 我们还要感谢开源实现变压器。
【文件预览】:
GMA-main
----checkpoints()
--------gma-sintel.pth(22.7MB)
----imgs()
--------frame_0047.png(697KB)
--------frame_0048.png(677KB)
----core()
--------gma.py(3KB)
--------utils()
--------corr.py(3KB)
--------update.py(5KB)
--------network.py(4KB)
--------extractor.py(6KB)
----LICENSE(484B)
----evaluate_single.py(2KB)
----README.md(798B)
----.gitignore(13B)
----demo.sh(85B)