├── .gitignore ├── Datasets ├── __init__.py ├── cowmask.py ├── flowlib.py ├── segmask_gt.py ├── tartanTrajFlowDataset.py ├── util_flow.py └── utils.py ├── DytanVO.py ├── LICENSE ├── Network ├── PWC │ ├── PWCNet.py │ ├── __init__.py │ └── correlation.py ├── VOFlowNet.py ├── VONet.py ├── __init__.py └── rigidmask │ ├── .gitignore │ ├── VCNplus.py │ ├── __init__.py │ ├── conv4d.py │ ├── det.py │ ├── det_losses.py │ ├── det_utils.py │ ├── networks │ ├── DCNv2 │ │ ├── .gitignore │ │ ├── DCN │ │ │ ├── __init__.py │ │ │ ├── dcn_v2.py │ │ │ ├── src │ │ │ │ ├── cpu │ │ │ │ │ ├── dcn_v2_cpu.cpp │ │ │ │ │ ├── dcn_v2_im2col_cpu.cpp │ │ │ │ │ ├── dcn_v2_im2col_cpu.h │ │ │ │ │ ├── dcn_v2_psroi_pooling_cpu.cpp │ │ │ │ │ └── vision.h │ │ │ │ ├── cuda │ │ │ │ │ ├── dcn_v2_cuda.cu │ │ │ │ │ ├── dcn_v2_im2col_cuda.cu │ │ │ │ │ ├── dcn_v2_im2col_cuda.h │ │ │ │ │ ├── dcn_v2_psroi_pooling_cuda.cu │ │ │ │ │ └── vision.h │ │ │ │ ├── dcn_v2.h │ │ │ │ └── vision.cpp │ │ │ ├── testcpu.py │ │ │ └── testcuda.py │ │ ├── LICENSE │ │ ├── README.md │ │ ├── make.sh │ │ └── setup.py │ ├── dlav0.py │ ├── large_hourglass.py │ ├── msra_resnet.py │ ├── pose_dla_dcn.py │ └── resnet_dcn.py │ └── submodule.py ├── README.md ├── environment.yml ├── evaluator ├── __init__.py ├── evaluate_ate_scale.py ├── evaluate_kitti.py ├── evaluate_rpe.py ├── evaluator_base.py ├── tartanair_evaluator.py ├── trajectory_transform.py └── transformation.py └── vo_trajectory_from_folder.py /.gitignore: -------------------------------------------------------------------------------- 1 | *.pyc 2 | models/ 3 | data/ 4 | __pycache__/ 5 | .DS_Store -------------------------------------------------------------------------------- /Datasets/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /Datasets/cowmask.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castacks/DytanVO/HEAD/Datasets/cowmask.py -------------------------------------------------------------------------------- /Datasets/flowlib.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castacks/DytanVO/HEAD/Datasets/flowlib.py -------------------------------------------------------------------------------- /Datasets/segmask_gt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castacks/DytanVO/HEAD/Datasets/segmask_gt.py -------------------------------------------------------------------------------- /Datasets/tartanTrajFlowDataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castacks/DytanVO/HEAD/Datasets/tartanTrajFlowDataset.py -------------------------------------------------------------------------------- /Datasets/util_flow.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castacks/DytanVO/HEAD/Datasets/util_flow.py -------------------------------------------------------------------------------- /Datasets/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castacks/DytanVO/HEAD/Datasets/utils.py -------------------------------------------------------------------------------- /DytanVO.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castacks/DytanVO/HEAD/DytanVO.py -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castacks/DytanVO/HEAD/LICENSE -------------------------------------------------------------------------------- /Network/PWC/PWCNet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castacks/DytanVO/HEAD/Network/PWC/PWCNet.py -------------------------------------------------------------------------------- /Network/PWC/__init__.py: -------------------------------------------------------------------------------- 1 | from .PWCNet import * 2 | -------------------------------------------------------------------------------- /Network/PWC/correlation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castacks/DytanVO/HEAD/Network/PWC/correlation.py -------------------------------------------------------------------------------- /Network/VOFlowNet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castacks/DytanVO/HEAD/Network/VOFlowNet.py -------------------------------------------------------------------------------- /Network/VONet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castacks/DytanVO/HEAD/Network/VONet.py -------------------------------------------------------------------------------- /Network/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /Network/rigidmask/.gitignore: -------------------------------------------------------------------------------- 1 | __pycache__ 2 | -------------------------------------------------------------------------------- /Network/rigidmask/VCNplus.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castacks/DytanVO/HEAD/Network/rigidmask/VCNplus.py -------------------------------------------------------------------------------- /Network/rigidmask/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /Network/rigidmask/conv4d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castacks/DytanVO/HEAD/Network/rigidmask/conv4d.py -------------------------------------------------------------------------------- /Network/rigidmask/det.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castacks/DytanVO/HEAD/Network/rigidmask/det.py -------------------------------------------------------------------------------- /Network/rigidmask/det_losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castacks/DytanVO/HEAD/Network/rigidmask/det_losses.py -------------------------------------------------------------------------------- /Network/rigidmask/det_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castacks/DytanVO/HEAD/Network/rigidmask/det_utils.py -------------------------------------------------------------------------------- /Network/rigidmask/networks/DCNv2/.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castacks/DytanVO/HEAD/Network/rigidmask/networks/DCNv2/.gitignore -------------------------------------------------------------------------------- /Network/rigidmask/networks/DCNv2/DCN/__init__.py: -------------------------------------------------------------------------------- 1 | from .dcn_v2 import * 2 | -------------------------------------------------------------------------------- /Network/rigidmask/networks/DCNv2/DCN/dcn_v2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castacks/DytanVO/HEAD/Network/rigidmask/networks/DCNv2/DCN/dcn_v2.py -------------------------------------------------------------------------------- /Network/rigidmask/networks/DCNv2/DCN/src/cpu/dcn_v2_cpu.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castacks/DytanVO/HEAD/Network/rigidmask/networks/DCNv2/DCN/src/cpu/dcn_v2_cpu.cpp -------------------------------------------------------------------------------- /Network/rigidmask/networks/DCNv2/DCN/src/cpu/dcn_v2_im2col_cpu.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castacks/DytanVO/HEAD/Network/rigidmask/networks/DCNv2/DCN/src/cpu/dcn_v2_im2col_cpu.cpp -------------------------------------------------------------------------------- /Network/rigidmask/networks/DCNv2/DCN/src/cpu/dcn_v2_im2col_cpu.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castacks/DytanVO/HEAD/Network/rigidmask/networks/DCNv2/DCN/src/cpu/dcn_v2_im2col_cpu.h -------------------------------------------------------------------------------- /Network/rigidmask/networks/DCNv2/DCN/src/cpu/dcn_v2_psroi_pooling_cpu.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castacks/DytanVO/HEAD/Network/rigidmask/networks/DCNv2/DCN/src/cpu/dcn_v2_psroi_pooling_cpu.cpp -------------------------------------------------------------------------------- /Network/rigidmask/networks/DCNv2/DCN/src/cpu/vision.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castacks/DytanVO/HEAD/Network/rigidmask/networks/DCNv2/DCN/src/cpu/vision.h -------------------------------------------------------------------------------- /Network/rigidmask/networks/DCNv2/DCN/src/cuda/dcn_v2_cuda.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castacks/DytanVO/HEAD/Network/rigidmask/networks/DCNv2/DCN/src/cuda/dcn_v2_cuda.cu -------------------------------------------------------------------------------- /Network/rigidmask/networks/DCNv2/DCN/src/cuda/dcn_v2_im2col_cuda.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castacks/DytanVO/HEAD/Network/rigidmask/networks/DCNv2/DCN/src/cuda/dcn_v2_im2col_cuda.cu -------------------------------------------------------------------------------- /Network/rigidmask/networks/DCNv2/DCN/src/cuda/dcn_v2_im2col_cuda.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castacks/DytanVO/HEAD/Network/rigidmask/networks/DCNv2/DCN/src/cuda/dcn_v2_im2col_cuda.h -------------------------------------------------------------------------------- /Network/rigidmask/networks/DCNv2/DCN/src/cuda/dcn_v2_psroi_pooling_cuda.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castacks/DytanVO/HEAD/Network/rigidmask/networks/DCNv2/DCN/src/cuda/dcn_v2_psroi_pooling_cuda.cu -------------------------------------------------------------------------------- /Network/rigidmask/networks/DCNv2/DCN/src/cuda/vision.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castacks/DytanVO/HEAD/Network/rigidmask/networks/DCNv2/DCN/src/cuda/vision.h -------------------------------------------------------------------------------- /Network/rigidmask/networks/DCNv2/DCN/src/dcn_v2.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castacks/DytanVO/HEAD/Network/rigidmask/networks/DCNv2/DCN/src/dcn_v2.h -------------------------------------------------------------------------------- /Network/rigidmask/networks/DCNv2/DCN/src/vision.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castacks/DytanVO/HEAD/Network/rigidmask/networks/DCNv2/DCN/src/vision.cpp -------------------------------------------------------------------------------- /Network/rigidmask/networks/DCNv2/DCN/testcpu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castacks/DytanVO/HEAD/Network/rigidmask/networks/DCNv2/DCN/testcpu.py -------------------------------------------------------------------------------- /Network/rigidmask/networks/DCNv2/DCN/testcuda.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castacks/DytanVO/HEAD/Network/rigidmask/networks/DCNv2/DCN/testcuda.py -------------------------------------------------------------------------------- /Network/rigidmask/networks/DCNv2/LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castacks/DytanVO/HEAD/Network/rigidmask/networks/DCNv2/LICENSE -------------------------------------------------------------------------------- /Network/rigidmask/networks/DCNv2/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castacks/DytanVO/HEAD/Network/rigidmask/networks/DCNv2/README.md -------------------------------------------------------------------------------- /Network/rigidmask/networks/DCNv2/make.sh: -------------------------------------------------------------------------------- 1 | #!/usr/bin/env bash 2 | python setup.py build develop 3 | -------------------------------------------------------------------------------- /Network/rigidmask/networks/DCNv2/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castacks/DytanVO/HEAD/Network/rigidmask/networks/DCNv2/setup.py -------------------------------------------------------------------------------- /Network/rigidmask/networks/dlav0.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castacks/DytanVO/HEAD/Network/rigidmask/networks/dlav0.py -------------------------------------------------------------------------------- /Network/rigidmask/networks/large_hourglass.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castacks/DytanVO/HEAD/Network/rigidmask/networks/large_hourglass.py -------------------------------------------------------------------------------- /Network/rigidmask/networks/msra_resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castacks/DytanVO/HEAD/Network/rigidmask/networks/msra_resnet.py -------------------------------------------------------------------------------- /Network/rigidmask/networks/pose_dla_dcn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castacks/DytanVO/HEAD/Network/rigidmask/networks/pose_dla_dcn.py -------------------------------------------------------------------------------- /Network/rigidmask/networks/resnet_dcn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castacks/DytanVO/HEAD/Network/rigidmask/networks/resnet_dcn.py -------------------------------------------------------------------------------- /Network/rigidmask/submodule.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castacks/DytanVO/HEAD/Network/rigidmask/submodule.py -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castacks/DytanVO/HEAD/README.md -------------------------------------------------------------------------------- /environment.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castacks/DytanVO/HEAD/environment.yml -------------------------------------------------------------------------------- /evaluator/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /evaluator/evaluate_ate_scale.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castacks/DytanVO/HEAD/evaluator/evaluate_ate_scale.py -------------------------------------------------------------------------------- /evaluator/evaluate_kitti.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castacks/DytanVO/HEAD/evaluator/evaluate_kitti.py -------------------------------------------------------------------------------- /evaluator/evaluate_rpe.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castacks/DytanVO/HEAD/evaluator/evaluate_rpe.py -------------------------------------------------------------------------------- /evaluator/evaluator_base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castacks/DytanVO/HEAD/evaluator/evaluator_base.py -------------------------------------------------------------------------------- /evaluator/tartanair_evaluator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castacks/DytanVO/HEAD/evaluator/tartanair_evaluator.py -------------------------------------------------------------------------------- /evaluator/trajectory_transform.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castacks/DytanVO/HEAD/evaluator/trajectory_transform.py -------------------------------------------------------------------------------- /evaluator/transformation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castacks/DytanVO/HEAD/evaluator/transformation.py -------------------------------------------------------------------------------- /vo_trajectory_from_folder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/castacks/DytanVO/HEAD/vo_trajectory_from_folder.py --------------------------------------------------------------------------------