Thanks to visit codestin.com
Credit goes to github.com

Skip to content

simoneangarano/kdust3r

 
 

Repository files navigation

train_kd.py: train KD model (check args in the file)

python train_kd.py --ckpt log/train_2_v2/checkpoint-best.pth --encoder_only --gauss_std 3 --lmd 10 --batch_size 8 --output_dir logs/train/ --cuda 0

test.py: test KD model (check args in the file)

python test.py --ckpt log/train_2_v2/checkpoint-best.pth --encoder_only --gauss_std 3 --batch_size 8 --cuda 0

point_cloud.py: generate teacher and student point clouds from image pairs (check parameters in the file)

python point_cloud.py --ckpt log/train_2_v2/checkpoint-best.pth --encoder_only --test_pairs co3d_test_1 co3d_test_2 croco dtu --cuda 0

About

KDUSt3R

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 84.9%
  • Jupyter Notebook 14.3%
  • Other 0.8%