slam3r-i2p_demo / scripts /demo_wild.sh
siyan824's picture
init
8bd45de
#!/bin/bash
######################################################################################
# Your input can be a directory of images, a video file, and a webcamera url.
# For directory of images, set the TEST_DATASET to path like: data/test/myimages
# For video, set the TEST_DATASET to path like: data/test/myvideo.mp4
# For webcamera, set the TEST_DATASET to url like: http://rab:12345678@192.168.137.83:8081
######################################################################################
TEST_DATASET="data/wild/Library"
######################################################################################
# set the parameters for whole scene reconstruction below
# for defination of these parameters, please refer to the recon.py
######################################################################################
TEST_NAME="Wild_demo"
KEYFRAME_STRIDE=3 #-1 for auto-adaptive keyframe stride selection
WIN_R=5
MAX_NUM_REGISTER=10
NUM_SCENE_FRAME=10
INITIAL_WINSIZE=5
CONF_THRES_L2W=12
CONF_THRES_I2P=1.5
NUM_POINTS_SAVE=1000000
RETRIEVE_FREQ=1
UPDATE_BUFFER_INTV=1
BUFFER_SIZE=100 # -1 if size is not limited
BUFFER_STRATEGY="reservoir" # or "fifo"
KEYFRAME_ADAPT_MIN=1
KEYFRAME_ADAPT_MAX=20
KEYFRAME_ADAPT_STRIDE=1
GPU_ID=-1
python recon.py \
--test_name $TEST_NAME \
--dataset "${TEST_DATASET}" \
--gpu_id $GPU_ID \
--keyframe_stride $KEYFRAME_STRIDE \
--win_r $WIN_R \
--num_scene_frame $NUM_SCENE_FRAME \
--initial_winsize $INITIAL_WINSIZE \
--conf_thres_l2w $CONF_THRES_L2W \
--conf_thres_i2p $CONF_THRES_I2P \
--num_points_save $NUM_POINTS_SAVE \
--update_buffer_intv $UPDATE_BUFFER_INTV \
--buffer_size $BUFFER_SIZE \
--buffer_strategy "${BUFFER_STRATEGY}" \
--max_num_register $MAX_NUM_REGISTER \
--keyframe_adapt_min $KEYFRAME_ADAPT_MIN \
--keyframe_adapt_max $KEYFRAME_ADAPT_MAX \
--keyframe_adapt_stride $KEYFRAME_ADAPT_STRIDE \
--retrieve_freq $RETRIEVE_FREQ \
--save_preds \
# --online # Uncomment it if you want to reconstruct it online