Update README.md

MingtaoGuo · Oct 12, 2022 · a7eacf0 · a7eacf0
1 parent 7d1e1c0
commit a7eacf0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/README.md b/README.md
@@ -29,7 +29,7 @@ All dependencies for defining the environment are provided in `environment.yaml`
 
 - Generate face mask by faceparsing
 ``` 
-python face_parsing/face_parsing.py --img_dir face_dataset --res_dir face_mask_dataset --resolution 256
+python face_parsing/face_parsing.py --img_dir face_dataset --res_dir face_mask_dataset --resolution 256 --model_path saved_models/79999_iter.pth 
 ```
 - Train StyleSwap 
 ```