flownet2-pytorch
FlowNet Pytorch实现。
支持多种GPU训练,并且代码提供了有关干净数据集和最终数据集的训练或推理示例。 相同的命令可用于训练或推断其他数据集。 有关更多详细信息,请参见下文。
还支持使用fp16(半精度)进行推理。
如需更多帮助,请键入
python main.py --help
网络架构
以下是提供的不同Flownet神经网络架构。 每个网络的batchnorm版本也可用。
FlowNet2S
FlowNet2C
FlowNet2CS
FlowNet2CSS
FlowNet2SD
FlowNet2
自定义图层
FlowNet2或FlowNet2C*结构依赖于自定义层Resample2d或Correlation 。 这些层与CUDA内核的pytorch实现可在 。 注意:当前,半精度内核不适用于这些层。
数据加载器
dat
2021-09-05 22:11:03
222KB
Python
1