From 716f239469c2578d1f8d67247d16fb08fcd64dc8 Mon Sep 17 00:00:00 2001 From: "yili.li" Date: Wed, 7 Dec 2022 06:39:54 +0000 Subject: [PATCH] fixbug and pr link #I64SPT fix issue_164SPT Signed-off-by: yili.li --- cv/super_resolution/esrgan/pytorch/README.md | 6 ++++-- .../configs/esrgan_psnr_x4c64b23g32_g1_1000k_div2k.py | 2 +- cv/super_resolution/esrgan/pytorch/get_div2k_anno.py | 2 +- 3 files changed, 6 insertions(+), 4 deletions(-) diff --git a/cv/super_resolution/esrgan/pytorch/README.md b/cv/super_resolution/esrgan/pytorch/README.md index 1569c0094..438c2e997 100755 --- a/cv/super_resolution/esrgan/pytorch/README.md +++ b/cv/super_resolution/esrgan/pytorch/README.md @@ -26,6 +26,8 @@ $ mkdir -p data/test $ cd data/test # Home page of Set5: http://people.rennes.inria.fr/Aline.Roumy/results/SR_BMVC12.html # Home page of Set14: https://github.com/jbhuang0604/SelfExSR + +$ python3 get_div2k_anno.py ``` ## Step 3: Training @@ -39,8 +41,8 @@ $ python3 train.py [training args] # config file can be found in ### Mutiple GPUs on one machine ```shell -$ bash train_dist.sh [training args] # config file can be found in the configs directory +$ bash dist_train.sh [training args] # config file can be found in the configs directory ``` ## Reference -https://github.com/open-mmlab/mmediting \ No newline at end of file +https://github.com/open-mmlab/mmediting diff --git a/cv/super_resolution/esrgan/pytorch/configs/esrgan_psnr_x4c64b23g32_g1_1000k_div2k.py b/cv/super_resolution/esrgan/pytorch/configs/esrgan_psnr_x4c64b23g32_g1_1000k_div2k.py index 806fb64b4..bf53fe07b 100755 --- a/cv/super_resolution/esrgan/pytorch/configs/esrgan_psnr_x4c64b23g32_g1_1000k_div2k.py +++ b/cv/super_resolution/esrgan/pytorch/configs/esrgan_psnr_x4c64b23g32_g1_1000k_div2k.py @@ -1,5 +1,5 @@ # Copyright (c) 2022, Shanghai Iluvatar CoreX Semiconductor Co., Ltd. - exp_name = 'esrgan_psnr_x4c64b23g32_g1_1000k_div2k' +exp_name = 'esrgan_psnr_x4c64b23g32_g1_1000k_div2k' scale = 4 # model settings diff --git a/cv/super_resolution/esrgan/pytorch/get_div2k_anno.py b/cv/super_resolution/esrgan/pytorch/get_div2k_anno.py index 7c4684644..45665d7d6 100755 --- a/cv/super_resolution/esrgan/pytorch/get_div2k_anno.py +++ b/cv/super_resolution/esrgan/pytorch/get_div2k_anno.py @@ -9,7 +9,7 @@ def generate_meta_info_div2k(): """ gt_folder = 'data/DIV2K/DIV2K_train_HR' - meta_info_txt = 'meta_info_DIV2K800sub_GT.txt' + meta_info_txt = 'data/DIV2K/meta_info_DIV2K800sub_GT.txt' img_list = sorted(list(mmcv.scandir(gt_folder))) -- Gitee