diff --git a/README.md b/README.md index 2c63c04..1b75b24 100644 --- a/README.md +++ b/README.md @@ -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)