LesionSegmenter / inference.sh
ChrisXzZ's picture
Upload folder using huggingface_hub
0dc829a verified
export nnUNet_N_proc_DA=36
export nnUNet_results="./nnUNet_results"
export nnUNet_predictions="./nnUNet_predictions"
export nnUNet_eval="./nnUNet_eval"
export PYTHONWARNINGS="ignore"
GPU_ID=0
DATASET=1357
TRAINER=nnUNetTrainer
CUDA_VISIBLE_DEVICES=$GPU_ID nnUNetv2_predict -d $DATASET -i $nnUNet_eval/Dataset1357/imagesTs/ -o $nnUNet_predictions/ -tr $TRAINER -d $DATASET -c 3d_fullres -f all -p nnUNetPlans --continue_prediction