duanyiqun / DiffusionDepth

PyTorch Implementation of introducing diffusion approach to 3D depth perception ECCV 2024
https://arxiv.org/abs/2303.05021
Apache License 2.0
306 stars 17 forks source link

Training example command for NYU Depth v2 #10

Closed ganyz closed 1 year ago

ganyz commented 1 year ago

Dear author: Could you provide the example commond to train the model on NYU Depth v2 like KITTI.

like :

$ python main.py --dir_data datta_path --data_name KITTIDC --split_json ../data_json/kitti_dp.json \ --patch_height 352 --patch_width 906 --gpus 0,1,2,3 --loss 1.0L1+1.0L2+1.0*DDIM --epochs 30 \ --batch_size 8 --max_depth 88.0 --save NAME_TO_SAVE \ --model_name DiffusionDCbase --backbone_module swin --backbone_name swin_large_naive_l4w722422k --head_specify DDIMDepthEstimate_Swin_ADDHAHI

Thank you!

duanyiqun commented 1 year ago

Hi, sorry for the late reply due to busy days. You might be interested in https://github.com/duanyiqun/DiffusionDepth/issues/9 for this issue I think the command is ok. But still, I recommend implementing it on Kitti first as it is more stable.