SAM3-LoRA-Breast-Lesion / scripts /run_single_prompt_eval.sh
pengzeli's picture
Add files using upload-large-folder tool
590304f verified
Raw
History Blame Contribute Delete
768 Bytes
#!/usr/bin/env bash
set -euo pipefail
INDEX_CSV="${1:?Usage: $0 INDEX_CSV OUTPUT_DIR [GPU]}"
OUTPUT_DIR="${2:?Usage: $0 INDEX_CSV OUTPUT_DIR [GPU]}"
GPU="${3:-0}"
SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
BUNDLE="$(cd "${SCRIPT_DIR}/.." && pwd)"
PYTHON="${PYTHON:-python}"
SAM3_CKPT="${SAM3_CHECKPOINT:-${BUNDLE}/model/sam3_base.pt}"
cd "${BUNDLE}"
CUDA_VISIBLE_DEVICES="${GPU}" PYTHONNOUSERSITE=1 "${PYTHON}" -u \
"${SCRIPT_DIR}/evaluate_sam3_test_text_prompt_segmentation.py" \
--index_csv "${INDEX_CSV}" \
--sam3_model_path "${SAM3_CKPT}" \
--checkpoint_path "${BUNDLE}/model/best_model.pt" \
--encoder_trainable lora \
--lora_rank 8 \
--lora_alpha 16 \
--prompts "breast lesion" \
--output_dir "${OUTPUT_DIR}" \
--overwrite