diff --git a/cv/super_resolution/esrgan/pytorch/README.md b/cv/super_resolution/esrgan/pytorch/README.md index 1569c0094b0e72ae1807d51882490c2d9cbb462e..438c2e997056526cddbc7eca19578ccd705e0cbb 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 806fb64b4fb0da774d2df31c275341050cb1e54c..bf53fe07bc7d12b5a49c7a1ab0fe2936f83a8aa3 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 7c4684644aa176453978a04e833db5c64d071076..45665d7d6ccf4552ad09da4ae4ed5c6d07f04bfb 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)))