文件名称:softmax-splatting:PyTorch实现可微正向弯曲的softmax喷溅的实现
文件大小:1.05MB
文件格式:ZIP
更新时间:2024-04-09 00:41:35
python deep-learning cuda pytorch cupy
softmax-splatting 这是softmax splatting运算符的参考实现,已在使用PyTorch的用于视频帧插值的Softmax splatting [1]中提出。 Softmax喷溅是一种有积极性的方法,可用于区分前向翘曲。它使用平移不变重要性度量来消除多个源像素映射到同一目标像素的情况的歧义。如果您要利用我们的工作,请引用我们的论文[1]。 设置 softmax喷溅是使用CuPy在CUDA中实现的,这就是为什么CuPy是必需的依赖项的原因。可以使用pip install cupy安装它,也可以使用CuPy存储库中概述的提供的二进制程序包之一进行pip install cupy 。 提供的示例脚本正在使用OpenCV加载和显示图像,以及读取提供的光流文件。安装适用于Python的OpenCV一种简单方法是使用pip install opencv-contrib-pyth
【文件预览】:
softmax-splatting-master
----images()
--------second.png(180KB)
--------README.md(81B)
--------first.png(178KB)
--------flow.flo(800KB)
----benchmark.py(5KB)
----requirements.txt(67B)
----run.py(4KB)
----README.md(2KB)
----softsplat.py(14KB)