├── .gitignore ├── README.md ├── basic_train.py ├── checkpoints ├── KITTI_flow.pth.tar ├── KITTI_flow_joint.pth.tar └── KITTI_stereo_depth.pth.tar ├── configs ├── KITTI_depth_stereo.json ├── KITTI_flow.json └── KITTI_rigid_flow_stereo.json ├── data └── filenames │ ├── FlyingChairs_train.txt │ ├── FlyingChairs_val.txt │ └── kitti_train_files_png_4frames.txt ├── datasets ├── KITTI.py └── get_dataset.py ├── demo ├── cityscapes.gif └── kitti.gif ├── logger.py ├── losses ├── depth_loss.py ├── flow_loss.py ├── get_loss.py ├── loss_blocks.py └── rigid_loss.py ├── models ├── get_model.py ├── net_blocks.py └── pwcnet.py ├── requirements.txt ├── train.py ├── trainer ├── base_trainer.py ├── get_trainer.py ├── kitti_depth_trainer.py ├── kitti_flow_trainer.py └── kitti_rigid_trainer.py ├── transforms ├── co_transforms.py └── sep_transforms.py └── utils ├── depth_utils.py ├── flow_utils.py ├── misc_utils.py ├── rigid_utils.py ├── torch_utils.py └── warp_utils.py /.gitignore: -------------------------------------------------------------------------------- 1 | *.py[cod] 2 | __pycache__/ 3 | .idea/ -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lliuz/UnRigidFlow/HEAD/README.md -------------------------------------------------------------------------------- /basic_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lliuz/UnRigidFlow/HEAD/basic_train.py -------------------------------------------------------------------------------- /checkpoints/KITTI_flow.pth.tar: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lliuz/UnRigidFlow/HEAD/checkpoints/KITTI_flow.pth.tar -------------------------------------------------------------------------------- /checkpoints/KITTI_flow_joint.pth.tar: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lliuz/UnRigidFlow/HEAD/checkpoints/KITTI_flow_joint.pth.tar -------------------------------------------------------------------------------- /checkpoints/KITTI_stereo_depth.pth.tar: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lliuz/UnRigidFlow/HEAD/checkpoints/KITTI_stereo_depth.pth.tar -------------------------------------------------------------------------------- /configs/KITTI_depth_stereo.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lliuz/UnRigidFlow/HEAD/configs/KITTI_depth_stereo.json -------------------------------------------------------------------------------- /configs/KITTI_flow.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lliuz/UnRigidFlow/HEAD/configs/KITTI_flow.json -------------------------------------------------------------------------------- /configs/KITTI_rigid_flow_stereo.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lliuz/UnRigidFlow/HEAD/configs/KITTI_rigid_flow_stereo.json -------------------------------------------------------------------------------- /data/filenames/FlyingChairs_train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lliuz/UnRigidFlow/HEAD/data/filenames/FlyingChairs_train.txt -------------------------------------------------------------------------------- /data/filenames/FlyingChairs_val.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lliuz/UnRigidFlow/HEAD/data/filenames/FlyingChairs_val.txt -------------------------------------------------------------------------------- /data/filenames/kitti_train_files_png_4frames.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lliuz/UnRigidFlow/HEAD/data/filenames/kitti_train_files_png_4frames.txt -------------------------------------------------------------------------------- /datasets/KITTI.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lliuz/UnRigidFlow/HEAD/datasets/KITTI.py -------------------------------------------------------------------------------- /datasets/get_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lliuz/UnRigidFlow/HEAD/datasets/get_dataset.py -------------------------------------------------------------------------------- /demo/cityscapes.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lliuz/UnRigidFlow/HEAD/demo/cityscapes.gif -------------------------------------------------------------------------------- /demo/kitti.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lliuz/UnRigidFlow/HEAD/demo/kitti.gif -------------------------------------------------------------------------------- /logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lliuz/UnRigidFlow/HEAD/logger.py -------------------------------------------------------------------------------- /losses/depth_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lliuz/UnRigidFlow/HEAD/losses/depth_loss.py -------------------------------------------------------------------------------- /losses/flow_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lliuz/UnRigidFlow/HEAD/losses/flow_loss.py -------------------------------------------------------------------------------- /losses/get_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lliuz/UnRigidFlow/HEAD/losses/get_loss.py -------------------------------------------------------------------------------- /losses/loss_blocks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lliuz/UnRigidFlow/HEAD/losses/loss_blocks.py -------------------------------------------------------------------------------- /losses/rigid_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lliuz/UnRigidFlow/HEAD/losses/rigid_loss.py -------------------------------------------------------------------------------- /models/get_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lliuz/UnRigidFlow/HEAD/models/get_model.py -------------------------------------------------------------------------------- /models/net_blocks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lliuz/UnRigidFlow/HEAD/models/net_blocks.py -------------------------------------------------------------------------------- /models/pwcnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lliuz/UnRigidFlow/HEAD/models/pwcnet.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lliuz/UnRigidFlow/HEAD/requirements.txt -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lliuz/UnRigidFlow/HEAD/train.py -------------------------------------------------------------------------------- /trainer/base_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lliuz/UnRigidFlow/HEAD/trainer/base_trainer.py -------------------------------------------------------------------------------- /trainer/get_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lliuz/UnRigidFlow/HEAD/trainer/get_trainer.py -------------------------------------------------------------------------------- /trainer/kitti_depth_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lliuz/UnRigidFlow/HEAD/trainer/kitti_depth_trainer.py -------------------------------------------------------------------------------- /trainer/kitti_flow_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lliuz/UnRigidFlow/HEAD/trainer/kitti_flow_trainer.py -------------------------------------------------------------------------------- /trainer/kitti_rigid_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lliuz/UnRigidFlow/HEAD/trainer/kitti_rigid_trainer.py -------------------------------------------------------------------------------- /transforms/co_transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lliuz/UnRigidFlow/HEAD/transforms/co_transforms.py -------------------------------------------------------------------------------- /transforms/sep_transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lliuz/UnRigidFlow/HEAD/transforms/sep_transforms.py -------------------------------------------------------------------------------- /utils/depth_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lliuz/UnRigidFlow/HEAD/utils/depth_utils.py -------------------------------------------------------------------------------- /utils/flow_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lliuz/UnRigidFlow/HEAD/utils/flow_utils.py -------------------------------------------------------------------------------- /utils/misc_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lliuz/UnRigidFlow/HEAD/utils/misc_utils.py -------------------------------------------------------------------------------- /utils/rigid_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lliuz/UnRigidFlow/HEAD/utils/rigid_utils.py -------------------------------------------------------------------------------- /utils/torch_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lliuz/UnRigidFlow/HEAD/utils/torch_utils.py -------------------------------------------------------------------------------- /utils/warp_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lliuz/UnRigidFlow/HEAD/utils/warp_utils.py --------------------------------------------------------------------------------