Skip to content

Commit

Permalink
update readme, add testing steps
Browse files Browse the repository at this point in the history
oneThousand1000 committed Apr 25, 2022

Verified

This commit was created on GitHub.com and signed with GitHub’s verified signature. The key has expired.
1 parent 6418347 commit d86ac42
Showing 1 changed file with 7 additions and 1 deletion.
8 changes: 7 additions & 1 deletion README.md
Original file line number Diff line number Diff line change
@@ -116,10 +116,14 @@ Then download and put the pre-trained models to **path**:

Directly use our pre-trained model for hair removal.

Real images should be extracted and aligned using DLib and a function from the original FFHQ dataset preparation step, you can use the [image align code](https://github.com/Puzer/stylegan-encoder/blob/master/align_images.py) provided by [stylegan-encoder](https://github.com/Puzer/stylegan-encoder).
**step1:**

Real images **should be extracted and aligned using DLib and a function from the original FFHQ dataset preparation step**, you can use the [image align code](https://github.com/Puzer/stylegan-encoder/blob/master/align_images.py) provided by [stylegan-encoder](https://github.com/Puzer/stylegan-encoder).

Please put the real images to **./test_data/origin** (examplar data can be found in ./data/test_data/final/origin).

**step2:**

Then using encoder4editing to get the corresponding latent codes:

```
@@ -129,6 +133,8 @@ python encode.py --data_dir ../test_data

latent codes will be saved to `./test_data/code`.

**step3:**

Then run HairMapper:

```python

0 comments on commit d86ac42

Please sign in to comment.