├── LICENSE ├── README.md ├── checkpoints └── place_checkpoints_here.txt ├── docs ├── badge-website.svg ├── teaser_new.png └── teaser_new.webp ├── environment.yaml ├── infer.py ├── inference └── depthlab_pipeline.py ├── output └── in-the-wild_example │ ├── depth_colored │ └── RGB_pred_colored.png │ └── depth_npy │ └── RGB_pred.npy ├── scripts └── infer.sh ├── src └── models │ ├── attention.py │ ├── mutual_self_attention.py │ ├── projection.py │ ├── transformer_2d.py │ ├── unet_2d_blocks.py │ ├── unet_2d_condition.py │ └── unet_2d_condition_main.py ├── test_cases ├── RGB.JPG ├── know_depth.npy ├── mask.jpg └── mask.npy └── utils ├── image_util.py └── seed_all.py /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ant-research/DepthLab/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ant-research/DepthLab/HEAD/README.md -------------------------------------------------------------------------------- /checkpoints/place_checkpoints_here.txt: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /docs/badge-website.svg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ant-research/DepthLab/HEAD/docs/badge-website.svg -------------------------------------------------------------------------------- /docs/teaser_new.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ant-research/DepthLab/HEAD/docs/teaser_new.png -------------------------------------------------------------------------------- /docs/teaser_new.webp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ant-research/DepthLab/HEAD/docs/teaser_new.webp -------------------------------------------------------------------------------- /environment.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ant-research/DepthLab/HEAD/environment.yaml -------------------------------------------------------------------------------- /infer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ant-research/DepthLab/HEAD/infer.py -------------------------------------------------------------------------------- /inference/depthlab_pipeline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ant-research/DepthLab/HEAD/inference/depthlab_pipeline.py -------------------------------------------------------------------------------- /output/in-the-wild_example/depth_colored/RGB_pred_colored.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ant-research/DepthLab/HEAD/output/in-the-wild_example/depth_colored/RGB_pred_colored.png -------------------------------------------------------------------------------- /output/in-the-wild_example/depth_npy/RGB_pred.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ant-research/DepthLab/HEAD/output/in-the-wild_example/depth_npy/RGB_pred.npy -------------------------------------------------------------------------------- /scripts/infer.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ant-research/DepthLab/HEAD/scripts/infer.sh -------------------------------------------------------------------------------- /src/models/attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ant-research/DepthLab/HEAD/src/models/attention.py -------------------------------------------------------------------------------- /src/models/mutual_self_attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ant-research/DepthLab/HEAD/src/models/mutual_self_attention.py -------------------------------------------------------------------------------- /src/models/projection.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ant-research/DepthLab/HEAD/src/models/projection.py -------------------------------------------------------------------------------- /src/models/transformer_2d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ant-research/DepthLab/HEAD/src/models/transformer_2d.py -------------------------------------------------------------------------------- /src/models/unet_2d_blocks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ant-research/DepthLab/HEAD/src/models/unet_2d_blocks.py -------------------------------------------------------------------------------- /src/models/unet_2d_condition.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ant-research/DepthLab/HEAD/src/models/unet_2d_condition.py -------------------------------------------------------------------------------- /src/models/unet_2d_condition_main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ant-research/DepthLab/HEAD/src/models/unet_2d_condition_main.py -------------------------------------------------------------------------------- /test_cases/RGB.JPG: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ant-research/DepthLab/HEAD/test_cases/RGB.JPG -------------------------------------------------------------------------------- /test_cases/know_depth.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ant-research/DepthLab/HEAD/test_cases/know_depth.npy -------------------------------------------------------------------------------- /test_cases/mask.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ant-research/DepthLab/HEAD/test_cases/mask.jpg -------------------------------------------------------------------------------- /test_cases/mask.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ant-research/DepthLab/HEAD/test_cases/mask.npy -------------------------------------------------------------------------------- /utils/image_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ant-research/DepthLab/HEAD/utils/image_util.py -------------------------------------------------------------------------------- /utils/seed_all.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ant-research/DepthLab/HEAD/utils/seed_all.py --------------------------------------------------------------------------------