From 3bf03c81361a5db5a7531e37e84bd5d6498b0d7d Mon Sep 17 00:00:00 2001 From: chenxuanhong Date: Wed, 20 Apr 2022 18:57:56 +0800 Subject: [PATCH] Update README.md --- README.md | 3 +++ 1 file changed, 3 insertions(+) diff --git a/README.md b/README.md index 2a5ba2b..00d4543 100644 --- a/README.md +++ b/README.md @@ -62,6 +62,8 @@ If you find this project useful, please star it. It is the greatest appreciation Download the dataset from [VGGFace2-HQ](https://github.com/NNNNAI/VGGFace2-HQ). +The training script is slightly different from the original version, e.g., we replace the patch discriminator with the projected discriminator, which saves a lot of hardware overhead and achieves slightly better results. + - Train 256 models ``` python train.py --name simswap256_test --gpu_ids 0 --dataset /path/to/VGGFace2HQ --train_simswap True --Gdeep False @@ -73,6 +75,7 @@ python train.py --name simswap512_test --gpu_ids 0 --dataset /path/to/VGGFace2H ``` + ## Inference with a pretrained SimSwap model [Preparation](./docs/guidance/preparation.md)