上传者: 42125770
|
上传时间: 2021-10-09 21:43:42
|
文件大小: 87.53MB
|
文件类型: -
使用Tensorflow进行光流预测
此存储库提供了基于DesSun的精彩论文“ PWC-Net:使用金字塔,扭曲和成本量进行光流的CNN”的基于TensorFlow的实现。 (CVPR 2018)。
已经有一些尝试使用TensorFlow实施PWC-Net。 但是,它们要么使用本文CNN网络的过时架构,要么仅提供TF推断(不提供TF训练),要么仅在Linux平台上工作,并且不支持多GPU训练。
此实现同时提供基于TF的训练和推理。 它具有可移植性:因为它不使用任何基于CUDA的动态加载的TensorFlow用户操作,因此可以在Linux和Windows上运行。 它还支持多GPU训练(此处