Pytorch implementation of FlowNet by Dosovitskiy et al.
This repository is a torch implementation of FlowNet, by Alexey Dosovitskiy et al. in PyTorch. This code is mainly inspired from here
Deformable Convolutions as seen in this paper, code inspired by
Following neural network models are currently provided :
- FlowNetS
- FlowNetSBN
- FlowNetC
- FlowNetCBN
- Def_FlowNetS
- Def_FlowNetSBN
- Def_FlowNetC
- Def_FlowNetCBN
pytorch >= 0.4.1
tensorboard-pytorch
tensorboardX >= 1.4
spatial-correlation-sampler>=0.0.8
imageio
argparse
- Example usage for FlowNetS :
python main.py /path/to/training_set/ -b8 -j8 -a flownets -s path/to/split/file