文件名称:PWC-Net_pytorch:pytorch实施的“ PWC-Net
文件大小:29.07MB
文件格式:ZIP
更新时间:2024-05-30 09:26:37
cnn pytorch optical-flow pwc-net Python
正式版(Caffe&PyTorch)在 ,谢谢大家的关注。 消息 修正我对“相关层”的用法,我一直在使用19 * 19邻域进行匹配。 NVIDIA非常善良地使用其出色的CUDA来使我的错误看起来不那么愚蠢,但我不想为了代码多样性之类而删除我那缓慢的缓慢的Cost Volume Layer。 致谢 :框架,数据转换器,损失函数以及有关流量估计的许多详细信息。 :Tensorboard记录器 :与model.summary()类似的模型摘要 普华永道 这是CVPR2018论文的非官方pytorch实现:孙德清等。 “ PWC-Net:使用金字塔,翘曲和成本量的光流CNN” 。 资源 | (官方) (流量输出从上到下,最右边是groundtruth) 它开始输出合理的流量。 但是,时间和性能都需要改善。 希望您能喜欢此代码,并随时分享有关网络及其超参数的想法。 用法 要求 Pyth
【文件预览】:
PWC-Net_pytorch-master
----.gitignore(72B)
----summary.py(3KB)
----LICENSE(1KB)
----correlation_package()
--------functions()
--------src()
--------__init__.py(0B)
--------build.py(719B)
--------_ext()
--------make.sh(346B)
--------modules()
----utils.py(357B)
----README.md(5KB)
----example()
--------2.png(472KB)
--------flow.png(134KB)
--------1.png(472KB)
--------SintelFinal-200K-noBN_SintelFinal-148K-BN.pkl(33.09MB)
----dataset.py(8KB)
----modules.py(5KB)
----main.py(17KB)
----flow_utils.py(4KB)
----logger.py(2KB)
----model.py(4KB)
----losses.py(3KB)