File size: 439 Bytes
29aaa12 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 | #!/usr/bin/env bash
set -euo pipefail
if [[ $# -lt 2 ]]; then
echo "Usage: $0 <input_ct.nii.gz> <output_mask.nii.gz> [cuda_device]" >&2
exit 2
fi
INPUT_NII="$1"
OUTPUT_NII="$2"
DEVICE="${3:-cuda:0}"
/data/wxh/miniconda3/envs/flowmatching/bin/python \
/data/wxh/Medical/samihs_ich_segmentation_package/segment_ich.py \
--input "$INPUT_NII" \
--output "$OUTPUT_NII" \
--device "$DEVICE" \
--batch-size 8 \
--threshold 0.5
|