Skip to content

GitLab

Commit 561123c7 authored by dongchy920's avatar dongchy920
Browse files

Update README.md

parent 4c8c290f
Hide whitespace changes
Inline Side-by-side
Showing with 5 additions and 3 deletions
README.md
View file @ 561123c7
...@@ -109,9 +109,6 @@ HIP_VISIBLE_DEVICES=2 python train.py LMDB --batch 4 ...@@ -109,9 +109,6 @@ HIP_VISIBLE_DEVICES=2 python train.py LMDB --batch 4
``` ```
HIP_VISIBLE_DEVICES=2,3 python -m torch.distributed.launch --nproc_per_node=2 --master_port=29500 train.py --batch 4 LMDB HIP_VISIBLE_DEVICES=2,3 python -m torch.distributed.launch --nproc_per_node=2 --master_port=29500 train.py --batch 4 LMDB
``` ```
<div align=center>
<img src="./assets/loss.png"/>
</div>
## 推理 ## 推理
下载权重文件[stylegan2-ffhq-config-f.pt](https://pan.baidu.com/s/1IOD4DjjjrMZBF_TfL7pGJQ?pwd=1234) 下载权重文件[stylegan2-ffhq-config-f.pt](https://pan.baidu.com/s/1IOD4DjjjrMZBF_TfL7pGJQ?pwd=1234)
...@@ -121,11 +118,16 @@ SCNet快速下载链接：[stylegan2-ffhq-config-f.pt](http://113.200.138.88:180 ...@@ -121,11 +118,16 @@ SCNet快速下载链接：[stylegan2-ffhq-config-f.pt](http://113.200.138.88:180
``` ```
python generate.py --sample 6 --pics 6 --ckpt stylegan2-ffhq-config-f.pt python generate.py --sample 6 --pics 6 --ckpt stylegan2-ffhq-config-f.pt
``` ```
## result
推理结果保存为sample/000006.png： 推理结果保存为sample/000006.png：
<div align=center> <div align=center>
<img src="./assets/000000.png"/> <img src="./assets/000000.png"/>
</div> </div>
### 精度
<div align=center>
<img src="./assets/loss.png"/>
</div>
## 应用场景 ## 应用场景
### 算法类别 ### 算法类别
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment