gitextract_5wcn44k3/ ├── .gitignore ├── LICENSE ├── README.md ├── configs/ │ ├── backbone.yaml │ └── stereo/ │ ├── cfg_coex.yaml │ └── cfg_psm.yaml ├── dataloaders/ │ ├── __init__.py │ └── stereo/ │ ├── KITTILoader.py │ ├── KITTIRawLoader.py │ ├── KITTI_submission_loader.py │ ├── KITTIloader2012.py │ ├── KITTIloader2015.py │ ├── SceneFlowLoader.py │ ├── __init__.py │ ├── listflowfile.py │ ├── lists/ │ │ ├── kitti2012_test.list │ │ ├── kitti2012_train.list │ │ ├── kitti2012_train170.list │ │ ├── kitti2012_val24.list │ │ ├── kitti2015_test.list │ │ ├── kitti2015_train.list │ │ ├── kitti2015_train180.list │ │ ├── kitti2015_val20.list │ │ ├── middeval3_test.list │ │ ├── middeval3_train.list │ │ ├── sceneflow_search_trainA.list │ │ ├── sceneflow_search_trainB.list │ │ ├── sceneflow_search_val.list │ │ ├── sceneflow_test.list │ │ └── sceneflow_train.list │ ├── preprocess.py │ ├── readpfm.py │ ├── stereo_albumentation.py │ └── transforms.py ├── demo.py ├── demo_tensorrt.py ├── demo_torchscript.py ├── environment.yml ├── logs/ │ └── stereo/ │ └── CoEx/ │ └── version_0/ │ └── checkpoints/ │ └── last.ckpt ├── models/ │ ├── __init__.py │ └── stereo/ │ ├── CoEx.py │ ├── CoExTRT.py │ ├── PSMNet.py │ ├── __init__.py │ └── submodules/ │ ├── Submodule.py │ ├── __init__.py │ ├── aggregation.py │ ├── feature.py │ ├── regression.py │ ├── spixel_utils/ │ │ ├── spixel.py │ │ ├── spixel_conv.py │ │ ├── spixel_loss.py │ │ └── spixel_test.py │ ├── util_conv.py │ └── utils.py ├── stereo.py ├── torch_to_tensorrt.py ├── utils/ │ ├── __init__.py │ └── load.py └── zoo/ └── torchscript/ └── CoEx.pt