File size: 1,601 Bytes
a6dd040 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 |
#测试dl3dv模型
conda activate /mnt/pfs/users/chaojun.ni/wangweijie_mnt/wangweijie/miniconda3/envs/depthsplat
CUDA_VISIBLE_DEVICES=5 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 \
mode=test \
test.save_video=false \
test.save_depth_concat_img=True \
test.save_image=True \
test.save_gt_image=True \
checkpointing.pretrained_model=checkpoints/dl3dv-256x448-volsplat_3dunet-randview2-6/checkpoints/epoch_9-step_100000.ckpt \
model.encoder.supervise_intermediate_depth=false \
output_dir=checkpoints/dl3dv-256x448-depthsplat-base-randview2-6 \
train.no_viz_video=True
#测试re10k模型
CUDA_VISIBLE_DEVICES=0 python -m src.main +experiment=re10k \
data_loader.train.batch_size=1 \
'dataset.roots'='["/home/aiscuser/yeqing/dataset/re10k"]' \
dataset.test_chunk_interval=10 \
trainer.max_steps=150000 \
model.encoder.num_scales=2 \
model.encoder.upsample_factor=2 \
model.encoder.lowest_feature_resolution=4 \
model.encoder.monodepth_vit_type=vitb \
mode=test \
test.save_video=false \
test.save_depth_concat_img=false \
test.save_image=false \
test.save_gt_image=false \
checkpointing.pretrained_model=/home/aiscuser/yeqing/Vol-Splat/checkpoints/re10k-256x256-unet_volsplat/checkpoints/epoch_2-step_150000.ckpt |