|
|
#!/bin/bash
|
|
|
|
|
|
|
|
|
ROOT_DIR="datasets/calibration-2023/"
|
|
|
SPLIT="test"
|
|
|
CFG="config/hrnetv2_w48.yaml"
|
|
|
CFG_L="config/hrnetv2_w48_l.yaml"
|
|
|
WEIGHTS_KP="weights/MV_kp"
|
|
|
WEIGHTS_L="weights/MV_lines"
|
|
|
SAVE_DIR="inference/inference_3D/inference_sn23/"
|
|
|
DEVICE="cuda:0"
|
|
|
KP_TH=0.0712
|
|
|
LINE_TH=0.2571
|
|
|
MAX_REPROJ_ERR=38
|
|
|
GT_FILE="${SAVE_DIR}${SPLIT}.zip"
|
|
|
PRED_FILE="${SAVE_DIR}${SPLIT}_pred.zip"
|
|
|
|
|
|
|
|
|
|
|
|
python scripts/inference_sn.py --cfg $CFG --cfg_l $CFG_L --weights_kp $WEIGHTS_KP --weights_line $WEIGHTS_L --root_dir $ROOT_DIR --split $SPLIT --save_dir $SAVE_DIR --kp_th $KP_TH --line_th $LINE_TH --max_reproj_err $MAX_REPROJ_ERR --cuda $DEVICE
|
|
|
|
|
|
|
|
|
python sn_calibration/src/evalai_camera.py -s $GT_FILE -p $PRED_FILE
|
|
|
|
