codeformer

README.md

@@ -98,17 +98,17 @@ Tips：以上dtk驱动、python、torch等DCU相关工具版本需要严格一
 
     dataset
     |—— ffhq_512
-        |—— xxx.png
+        ├── 00522.png
+        ├── 01459.png
+        ├── 02090.png
+        ├── xxx.png
 
-注意：原始数据为`1024x1024`需要处理为`512x512`。
+注意：原始数据为`1024x1024`需要处理为`512x512`，可运行`data_process.py`对数据进行处理。
 
 ## 推理
 
 ### 模型下载
 
-百度网盘链接：https://pan.baidu.com/s/1CuntFQYr9gnkH7zopWdYtQ 
-（提取码：kwai）
-
 Github：https://github.com/sczhou/CodeFormer/releases/tag/v0.1.0
 
     weights
@@ -190,6 +190,9 @@ Github：https://github.com/sczhou/CodeFormer/releases/tag/v0.1.0
 
     python -m torch.distributed.launch --nproc_per_node=8 --master_port=4323 basicsr/train.py -opt options/CodeFormer_stage3.yml --launcher pytorch
 
+
+注意：`vqgan`为`net_g_xxx.pth`，`net_d`为第一阶段生成。
+
 ## result
 
 ![Alt text](readme_images/image-3.png)