Update run.sh

huangyangyi · Mar 7, 2024 · 3bdfad6 · 3bdfad6
1 parent f84972c
commit 3bdfad6
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/scripts/run.sh b/scripts/run.sh
@@ -14,8 +14,8 @@ python utils/body_utils/preprocess.py --in_path ${INPUT_FILE} --out_dir ${EXP_DI
 # Step 2: Get BLIP prompt and gender, you can also use your own prompt
 python utils/get_prompt_blip.py --img-path ${EXP_DIR}/png/${SUBJECT_NAME}_crop.png --out-path ${EXP_DIR}/prompt.txt
 # python core/get_prompt.py ${EXP_DIR}/png/${SUBJECT_NAME}_crop.png
-export PROMPT=`cat ${EXP_DIR}/prompt.txt| cut -d'|' -f1`
-export GENDER=`cat ${EXP_DIR}/prompt.txt| cut -d'|' -f2`
+export PROMPT="`cat ${EXP_DIR}/prompt.txt| cut -d'|' -f1`"
+export GENDER="`cat ${EXP_DIR}/prompt.txt| cut -d'|' -f2`"
 
 # Step 3: Finetune Dreambooth model (minimal GPU memory requirement: 2x32G)
 rm -rf ${EXP_DIR}/ldm
@@ -33,4 +33,4 @@ python utils/body_utils/postprocess.py --dir $EXP_DIR/obj --name $SUBJECT_NAME
 python core/main.py --config configs/tech_texture.yaml --exp_dir $EXP_DIR --sub_name $SUBJECT_NAME
 
 # [Optional] export textured mesh with UV map, using atlas for UV unwraping.
-python core/main.py --config configs/tech_texture_export.yaml --exp_dir $EXP_DIR --sub_name $SUBJECT_NAME --test
+python core/main.py --config configs/tech_texture_export.yaml --exp_dir $EXP_DIR --sub_name $SUBJECT_NAME --test