File size: 779 Bytes
3d1f2c9 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 |
#!/bin/bash
# Set parameters
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"
# Run inference script
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
# Run evaluation script
python sn_calibration/src/evalai_camera.py -s $GT_FILE -p $PRED_FILE
|