truongkhang / depth-completion-seq-net

Online 3D modeling by depth completion (RA-L 2021)

Sequential Depth Completion with Confidence Estimation for 3D Model Reconstruction

Introduction

This is a deep neural network for depth completion using sequence information.

The full demo video can be found here. Link to our paper here

Installation

conda create -n venv python=3.7
conda activate venv
conda install pytorch==1.1.0 torchvision cudatoolkit=9.0 -c pytorch
conda install -c conda-forge opencv
pip install pycuda Cython pandas pykitti scikit-image scipy

Using pip for installing the other dependence packages

Next, compiling the library for using warping function. We upgraded this library from the original here.

Make sure that the CUDA version on your machine is 9.0. You can check whether the folder /usr/local/cuda-9.0 exists or not.

cd depth-completion-seq-net/MYTH
python setup.py install

Preparation

You need to prepare your dataset followed our template except for KITTI dataset (It should be organized as same as the original).

Dataset structure should look like this:

|--train # for training model
    |-- sequence 1
        |--depth*.png  # ground truth depth, '*' here the index 
        |--image*.png  # RGB image
        |--sparse*.png # sparse depth input
        |--pose*.txt   # camera-to-world, 4×4 matrix in homogeneous coordinates
    |-- sequence 2
    ...
|--val # for evaluating our model based on some metrics: RMSE, MAE...
    |-- sequence 1
        |--depth*.png  # ground truth depth, '*' here the index 
        |--image*.png  # RGB image
        |--sparse*.png # sparse depth input
        |--pose*.txt   # camera-to-world, 4×4 matrix in homogeneous coordinates
    |-- sequence 2
    ...
|--test # for depth prediction without ground truth depth
    |-- sequence 1
        |--image*.png  # RGB image
        |--sparse*.png # sparse depth input
        |--pose*.txt   # camera-to-world, 4×4 matrix in homogeneous coordinates
    |-- sequence 2
    ...

For kitti, we use two datasets (raw and depth-completion), the split.txt file is inside data_loaders folder test.txt and train.txt

for the raw use this. (look here - https://github.com/nianticlabs/monodepth2)

use wget https://github.com/mrharicot/monodepth/blob/master/utils/kitti_archives_to_download.txt / wget -i kitti_archives_to_download.txt / unzip "*.zip"

|--raw_dataset # 
    |-- 2011_09_26
        |-- calib_velo_to_cam.txt
        |-- calib_imu_to_velo.txt
        |-- calib_cam_to_cam.txt
        |-- 2011_09_26_drive_0020_sync   # camera-to-world, 4×4 matrix in homogeneous coordinates
            |-- image_00
            |-- image_01

    
    ...
|--depth_dataset # 
    |-- data_depth_velodyne # wget https://s3.eu-central-1.amazonaws.com/avg-kitti/data_depth_velodyne.zip
        
    |-- data_depth_annotated # https://s3.eu-central-1.amazonaws.com/avg-kitti/data_depth_annotated.zip
    ...

Training, evaluation end prediction

Make sure activate your virtual environment before running code:

conda activate venv

Train model:

python train.py --config config_kitti.json
# all parameters of model are configured in this file
# file config_kitti.json is used for the kitti dataset
# file config_aerial.json is used for the aerial dataset

Evaluate model:

We need a pretrained model for evaluation. The pretrained model is put in a folder along with file config. See the pretrains in the project for illustration. The evaluated dataset is the data in the folder val

python test.py --resume <path to pretrained model> --save_folder <folder to save the results>

Prediction:

Predict the depth map from the image and sparse depth input. There's no need ground truth depth (see the structure of test folder)

python predict.py --resume <path to pretrained model> --save_folder <folder to save the results>

Citation

If you find this source code helpful for your research, please cite this

@ARTICLE{9286507,
  author={K. {Truong Giang} and S. {Song} and D. {Kim} and S. {Choi}},
  journal={IEEE Robotics and Automation Letters}, 
  title={Sequential Depth Completion With Confidence Estimation for 3D Model Reconstruction}, 
  year={2021},
  volume={6},
  number={2},
  pages={327-334},
  doi={10.1109/LRA.2020.3043172}}

depth-completion-seq-net's Issues

cuda10.1

hi, i have a question, how should i modify the MYTH folder when i using cuda10.1 or 11.1? thanks. ​:smile:​