Cross-spectrum Depth Estimation

Cross-Spectrum Unsupervised Depth Estimation by Visible-light and Thermal Cameras

[papre link]

Training

requriment:

pytorch >= 1.7.0

cuda >= 10.0

python >= 3.6
Prepare dataset

Download VTD dataset.
Edit train.sh to add your dataset path.

CUDA_VISIBLE_DEVICES=0 python train.py --mscroot "yourpath"

then, run train.sh.

Name		Name	Last commit message	Last commit date
Latest commit History 5 Commits
data		data
.gitignore		.gitignore
datasets.py		datasets.py
evaluation.py		evaluation.py
evaluation_utils.py		evaluation_utils.py
loss.py		loss.py
model_gan.py		model_gan.py
model_mscdepth.py		model_mscdepth.py
optimize.py		optimize.py
readme.md		readme.md
reprojection.py		reprojection.py
test.py		test.py
test.sh		test.sh
train.py		train.py
train.sh		train.sh
utils.py		utils.py