scripts/run_pipeline_wc14.sh

#!/bin/bash

# Set parameters
ROOT_DIR="datasets/WC-2014/"
SPLIT="test"
CFG="config/hrnetv2_w48.yaml"
CFG_L="config/hrnetv2_w48_l.yaml"
WEIGHTS_KP="weights/SV_FT_WC14_kp"
WEIGHTS_L="weights/SV_FT_WC14_lines"
SAVE_DIR="inference/inference_2D/inference_wc14/"
KP_TH=0.1274
LINE_TH=0.1439
DEVICE="cuda:0"
PRED_FILE="${SAVE_DIR}${SPLIT}_pred.zip"

# Run inference script
python scripts/inference_wc14.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 --cuda $DEVICE

# Run evaluation script
python scripts/eval_wc14.py --root_dir $ROOT_DIR --split $SPLIT --pred_file $PRED_FILE