Yeqing0814/depthsplat

安装MinkowskiEngine

• 因为用到了稀疏3D卷积，需要安装MinkowskiEngine

• 但是对于cuda版本大于11的安装这个库存在很多环境问题：

• 按照以下步骤可以解决

• （1）首先按照这个链接安装环境(其中Step 4与Step 5已完成)： https://github.com/Julie-tang00/Common-envs-issues/blob/main/Cuda12-MinkowskiEngine

• （2）安装依赖：

conda activate depthsplat 
conda install -c conda-forge openblas

• （3）测试：

python
>>> import MinkowskiEngine as ME
>>> print(ME.__version__)
0.5.4

训练指令

conda activate depthsplat
CUDA_VISIBLE_DEVICES=2 python -m src.main +experiment=dl3dv \
data_loader.train.batch_size=1 \
'dataset.roots'='["datasets/dl3dv"]' \
dataset.view_sampler.num_target_views=8 \
dataset.view_sampler.num_context_views=6 \
dataset.min_views=1 \
dataset.max_views=6 \
trainer.max_steps=100000 \
trainer.num_nodes=1 \
model.encoder.num_scales=2 \
model.encoder.upsample_factor=4 \
model.encoder.lowest_feature_resolution=8 \
model.encoder.monodepth_vit_type=vitb \
checkpointing.pretrained_monodepth=pretrained/pretrained_weights/depth_anything_v2_vitb.pth \
checkpointing.pretrained_mvdepth=pretrained/pretrained_weights/gmflow-scale1-things-e9887eda.pth \
wandb.project=VolSplat \
output_dir=checkpoints/dl3dv-256x448-depthsplat-base-randview2-6 \
train.no_viz_video=True