Update readme.md

This commit is contained in:
Gordon Guocheng Qian 钱国成
2023-09-06 12:31:02 -07:00
committed by GitHub
parent 03471a99b3
commit 7f051875e4

View File

@@ -96,9 +96,9 @@ $init_token is a single token to describe the image using natural language
For example: For example:
```bash ```bash
bash scripts/textual_inversion/textual_inversion.sh runwayml/stable-diffusion-v1-5 data/demo/ironman/rgba.png out/textual_inversion/ironman _ironman_ ironman --max_train_steps 3000 bash scripts/textual_inversion/textual_inversion.sh runwayml/stable-diffusion-v1-5 data/demo/a-full-body-ironman/rgba.png out/textual_inversion/ironman _ironman_ ironman --max_train_steps 3000
``` ```
Don't forget to move the final `learned_embeds.bin` under data/demo/ironman/ Don't forget to move the final `learned_embeds.bin` under data/demo/a-full-body-ironman/
## Run ## Run
@@ -128,13 +128,13 @@ textual inversion is tedious (requires ~2.5 hours optimization), if you want to
- first, foreground and depth estimation - first, foreground and depth estimation
``` ```
python preprocess_image.py --path data/demo/ironman/main.png python preprocess_image.py --path data/demo/a-full-body-ironman/main.png
``` ```
- Run Magic123 coarse stage without textual inversion, takes ~40 mins - Run Magic123 coarse stage without textual inversion, takes ~40 mins
``` ```
export RUN_ID='default-a-full-body-ironman' export RUN_ID='default-a-full-body-ironman'
export DATA_DIR='data/demo/ironman' export DATA_DIR='data/demo/a-full-body-ironman'
export IMAGE_NAME='rgba.png' export IMAGE_NAME='rgba.png'
export FILENAME=$(basename $DATA_DIR) export FILENAME=$(basename $DATA_DIR)
export dataset=$(basename $(dirname $DATA_DIR)) export dataset=$(basename $(dirname $DATA_DIR))
@@ -159,7 +159,7 @@ textual inversion is tedious (requires ~2.5 hours optimization), if you want to
``` ```
export RUN_ID='default-a-full-body-ironman' export RUN_ID='default-a-full-body-ironman'
export RUN_ID2='dmtet' export RUN_ID2='dmtet'
export DATA_DIR='data/demo/ironman' export DATA_DIR='data/demo/a-full-body-ironman'
export IMAGE_NAME='rgba.png' export IMAGE_NAME='rgba.png'
export FILENAME=$(basename $DATA_DIR) export FILENAME=$(basename $DATA_DIR)
export dataset=$(basename $(dirname $DATA_DIR)) export dataset=$(basename $(dirname $DATA_DIR))
@@ -195,7 +195,7 @@ textual inversion is tedious (requires ~2.5 hours optimization), if you want to
- Run Magic123 with only 3D prior (Like Zero-1-to-3 but we achieve much better performance through training stragies and the coarse-to-fine pipeline) - Run Magic123 with only 3D prior (Like Zero-1-to-3 but we achieve much better performance through training stragies and the coarse-to-fine pipeline)
``` ```
bash scripts/magic123/run_3dprior.sh 0 nerf dmtet data/demo/ironman 1 1 bash scripts/magic123/run_3dprior.sh 0 nerf dmtet data/demo/a-full-body-ironman 1 1
``` ```