SegVol / model /script /inference_demo.sh
BoyaWu10's picture
init the space (#2)
a950ee6
raw
history blame
No virus
260 Bytes
export segvol_ckpt="path/to/SegVol_v1.pth"
export work_dir="path/to/work_dir"
export demo_config_path="./config/config_demo.json"
CUDA_VISIBLE_DEVICES=0 python inference_demo.py \
--resume $segvol_ckpt \
-work_dir $work_dir \
--demo_config $demo_config_path