flownet2-pytorch:FlowNet 2.0的Pytorch实施

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