Update README (#12)

main
SWHL 2022-12-20 13:04:24 +08:00 committed by GitHub
parent 8f917387b6
commit 3647958e9c
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 4 deletions

View File

@ -26,10 +26,10 @@ The values in the last 3 columns are `AD`, `MS-SSIM`, and `LD`.
1. Download the pretrained model to the `models` directory.
2. Run the `demo.py` by the following code:
```shell
$ python --Enet_ckpt 'models/G_w_checkpoint_13820.pt' \
--Tnet_ckpt 'models/L_w_checkpoint_27640.pt' \
--img_path 'images/1.jpg' \
--out_dir 'output'
$ python demo.py --Enet_ckpt 'models/G_w_checkpoint_13820.pt' \
--Tnet_ckpt 'models/L_w_checkpoint_27640.pt' \
--img_path 'images/1.jpg' \
--out_dir 'output'
```
3. The final result:
![compare](https://user-images.githubusercontent.com/28639377/196933170-81c7e3d8-3661-429b-ae17-efae33366545.png)