From deccc1dfce9da6896b9a7d09847fcac06d2fbd2c Mon Sep 17 00:00:00 2001 From: ys-li <56712176+Yshuo-Li@users.noreply.github.com> Date: Wed, 1 Dec 2021 17:25:32 +0800 Subject: [PATCH] [Feature] Update metafile and update_model_index.py (#615) * Update metafile * Update * Update * add print * Update * Solve the difference of 'sort by length' between local and CI --- .dev_scripts/github/update_model_index.py | 55 +++++++++++++++++-- configs/inpainting/deepfillv1/metafile.yml | 4 +- configs/inpainting/deepfillv2/metafile.yml | 4 +- configs/inpainting/global_local/metafile.yml | 4 +- configs/inpainting/partial_conv/metafile.yml | 4 +- configs/mattors/dim/metafile.yml | 6 +- configs/mattors/gca/metafile.yml | 8 +-- configs/mattors/indexnet/metafile.yml | 4 +- configs/restorers/basicvsr/metafile.yml | 6 +- .../restorers/basicvsr_plusplus/metafile.yml | 6 +- configs/restorers/dic/metafile.yml | 4 +- configs/restorers/edsr/metafile.yml | 6 +- configs/restorers/edvr/metafile.yml | 4 +- configs/restorers/esrgan/metafile.yml | 4 +- configs/restorers/glean/README.md | 2 +- configs/restorers/glean/metafile.yml | 46 +++++++++++++++- configs/restorers/iconvsr/README.md | 2 +- configs/restorers/iconvsr/metafile.yml | 8 +-- configs/restorers/liif/metafile.yml | 24 ++++---- configs/restorers/rdn/metafile.yml | 6 +- configs/restorers/real_esrgan/metafile.yml | 2 +- configs/restorers/srcnn/metafile.yml | 2 +- configs/restorers/srresnet_srgan/metafile.yml | 4 +- configs/restorers/tdan/metafile.yml | 4 +- configs/restorers/tof/metafile.yml | 2 +- configs/restorers/ttsr/metafile.yml | 4 +- configs/synthesizers/cyclegan/metafile.yml | 20 +++---- configs/synthesizers/pix2pix/metafile.yml | 8 +-- 28 files changed, 171 insertions(+), 82 deletions(-) diff --git a/.dev_scripts/github/update_model_index.py b/.dev_scripts/github/update_model_index.py index d300d06e2b..bf056c9d54 100755 --- a/.dev_scripts/github/update_model_index.py +++ b/.dev_scripts/github/update_model_index.py @@ -7,10 +7,12 @@ # detected before a commit. import glob +import os import os.path as osp import re import sys import warnings +from functools import reduce import mmcv @@ -76,7 +78,7 @@ def get_task_name(md_file): """Get task name from README.md". Args: - md_file: Path to .md file. + md_file (str): Path to .md file. Returns: Str: Task name. @@ -88,14 +90,50 @@ def get_task_name(md_file): return 'Unknown' +def generate_unique_name(md_file): + """Search config files and return the unique name of them. + For Confin.Name. + + Args: + md_file (str): Path to .md file. + Returns: + dict: dict of unique name for each config file. + """ + files = os.listdir(osp.dirname(md_file)) + config_files = [f[:-3] for f in files if f[-3:] == '.py'] + config_files.sort() + config_files.sort(key=lambda x: len(x)) + split_names = [f.split('_') for f in config_files] + config_sets = [set(f.split('_')) for f in config_files] + common_set = reduce(lambda x, y: x & y, config_sets) + unique_lists = [[n for n in name if n not in common_set] + for name in split_names] + + unique_dict = dict() + name_list = [] + for i, f in enumerate(config_files): + base = split_names[i][0] + unique_dict[f] = base + if len(unique_lists[i]) > 0: + for unique in unique_lists[i]: + candidate_name = f'{base}_{unique}' + if candidate_name not in name_list and base != unique: + unique_dict[f] = candidate_name + name_list.append(candidate_name) + break + return unique_dict + + def parse_md(md_file): """Parse .md file and convert it to a .yml file which can be used for MIM. Args: - md_file: Path to .md file. + md_file (str): Path to .md file. Returns: Bool: If the target YAML file is different from the original. """ + unique_dict = generate_unique_name(md_file) + collection_name = osp.splitext(osp.basename(md_file))[0] collection = dict( Name=collection_name, @@ -125,7 +163,8 @@ def parse_md(md_file): # parse table elif lines[i][0] == '|' and i + 1 < len(lines) and \ - (lines[i + 1][:3] == '| :' or lines[i + 1][:2] == '|:'): + (lines[i + 1][:3] == '| :' or lines[i + 1][:2] == '|:' + or lines[i + 1][:2] == '|-'): cols = [col.strip() for col in lines[i].split('|')][1:-1] config_idx = cols.index('Method') checkpoint_idx = cols.index('Download') @@ -157,8 +196,14 @@ def parse_md(md_file): right = line[checkpoint_idx].index(')', left) checkpoint = line[checkpoint_idx][left:right] - model_name = osp.splitext(config)[0].replace( - 'configs/', '', 1).replace('/', '--') + name_key = osp.splitext(osp.basename(config))[0] + if name_key in unique_dict: + model_name = unique_dict[name_key] + else: + model_name = name_key + warnings.warn( + f'Config file of {model_name} is not found,' + 'please check it again.') # find dataset in config file dataset = 'Others' diff --git a/configs/inpainting/deepfillv1/metafile.yml b/configs/inpainting/deepfillv1/metafile.yml index d4dbba205e..ae6765fb5a 100644 --- a/configs/inpainting/deepfillv1/metafile.yml +++ b/configs/inpainting/deepfillv1/metafile.yml @@ -11,7 +11,7 @@ Models: In Collection: DeepFillv1 Metadata: Training Data: PLACES - Name: inpainting--deepfillv1--deepfillv1_256x256_8x2_places + Name: deepfillv1_8x2 Results: - Dataset: PLACES Metrics: @@ -24,7 +24,7 @@ Models: In Collection: DeepFillv1 Metadata: Training Data: CELEBA - Name: inpainting--deepfillv1--deepfillv1_256x256_4x4_celeba + Name: deepfillv1_4x4 Results: - Dataset: CELEBA Metrics: diff --git a/configs/inpainting/deepfillv2/metafile.yml b/configs/inpainting/deepfillv2/metafile.yml index 3e20b169a0..b272a372b8 100644 --- a/configs/inpainting/deepfillv2/metafile.yml +++ b/configs/inpainting/deepfillv2/metafile.yml @@ -11,7 +11,7 @@ Models: In Collection: DeepFillv2 Metadata: Training Data: PLACES - Name: inpainting--deepfillv2--deepfillv2_256x256_8x2_places + Name: deepfillv2_places Results: - Dataset: PLACES Metrics: @@ -24,7 +24,7 @@ Models: In Collection: DeepFillv2 Metadata: Training Data: CELEBA - Name: inpainting--deepfillv2--deepfillv2_256x256_8x2_celeba + Name: deepfillv2_celeba Results: - Dataset: CELEBA Metrics: diff --git a/configs/inpainting/global_local/metafile.yml b/configs/inpainting/global_local/metafile.yml index ef59bde1a9..6d113c9dfe 100644 --- a/configs/inpainting/global_local/metafile.yml +++ b/configs/inpainting/global_local/metafile.yml @@ -11,7 +11,7 @@ Models: In Collection: Global&Local Metadata: Training Data: PLACES - Name: inpainting--global_local--gl_256x256_8x12_places + Name: gl_places Results: - Dataset: PLACES Metrics: @@ -24,7 +24,7 @@ Models: In Collection: Global&Local Metadata: Training Data: CELEBA - Name: inpainting--global_local--gl_256x256_8x12_celeba + Name: gl_celeba Results: - Dataset: CELEBA Metrics: diff --git a/configs/inpainting/partial_conv/metafile.yml b/configs/inpainting/partial_conv/metafile.yml index 01c19cdfa3..97774231da 100644 --- a/configs/inpainting/partial_conv/metafile.yml +++ b/configs/inpainting/partial_conv/metafile.yml @@ -11,7 +11,7 @@ Models: In Collection: PConv Metadata: Training Data: PLACES - Name: inpainting--partial_conv--pconv_256x256_stage2_4x2_places + Name: pconv_4x2 Results: - Dataset: PLACES Metrics: @@ -24,7 +24,7 @@ Models: In Collection: PConv Metadata: Training Data: CELEBA - Name: inpainting--partial_conv--pconv_256x256_stage2_4x2_celeba + Name: pconv_stage2 Results: - Dataset: CELEBA Metrics: diff --git a/configs/mattors/dim/metafile.yml b/configs/mattors/dim/metafile.yml index eecf4caf33..16fa1298d3 100644 --- a/configs/mattors/dim/metafile.yml +++ b/configs/mattors/dim/metafile.yml @@ -11,7 +11,7 @@ Models: In Collection: DIM Metadata: Training Data: COMP1K - Name: mattors--dim--dim_stage1_v16_1x1_1000k_comp1k + Name: dim_stage1 Results: - Dataset: COMP1K Metrics: @@ -25,7 +25,7 @@ Models: In Collection: DIM Metadata: Training Data: COMP1K - Name: mattors--dim--dim_stage2_v16_pln_1x1_1000k_comp1k + Name: dim_stage2 Results: - Dataset: COMP1K Metrics: @@ -39,7 +39,7 @@ Models: In Collection: DIM Metadata: Training Data: COMP1K - Name: mattors--dim--dim_stage3_v16_pln_1x1_1000k_comp1k + Name: dim_stage3 Results: - Dataset: COMP1K Metrics: diff --git a/configs/mattors/gca/metafile.yml b/configs/mattors/gca/metafile.yml index e0c6edab57..f6f41413ba 100644 --- a/configs/mattors/gca/metafile.yml +++ b/configs/mattors/gca/metafile.yml @@ -11,7 +11,7 @@ Models: In Collection: GCA Metadata: Training Data: COMP1K - Name: mattors--gca--baseline_r34_4x10_200k_comp1k + Name: baseline Results: - Dataset: COMP1K Metrics: @@ -25,7 +25,7 @@ Models: In Collection: GCA Metadata: Training Data: COMP1K - Name: mattors--gca--gca_r34_4x10_200k_comp1k + Name: gca Results: - Dataset: COMP1K Metrics: @@ -39,7 +39,7 @@ Models: In Collection: GCA Metadata: Training Data: COMP1K - Name: mattors--gca--baseline_dimaug_r34_4x10_200k_comp1k + Name: baseline_dimaug Results: - Dataset: COMP1K Metrics: @@ -53,7 +53,7 @@ Models: In Collection: GCA Metadata: Training Data: COMP1K - Name: mattors--gca--gca_dimaug_r34_4x10_200k_comp1k + Name: gca_dimaug Results: - Dataset: COMP1K Metrics: diff --git a/configs/mattors/indexnet/metafile.yml b/configs/mattors/indexnet/metafile.yml index 90feed8b1c..3280e744ca 100644 --- a/configs/mattors/indexnet/metafile.yml +++ b/configs/mattors/indexnet/metafile.yml @@ -11,7 +11,7 @@ Models: In Collection: IndexNet Metadata: Training Data: COMP1K - Name: mattors--indexnet--indexnet_mobv2_1x16_78k_comp1k + Name: indexnet Results: - Dataset: COMP1K Metrics: @@ -25,7 +25,7 @@ Models: In Collection: IndexNet Metadata: Training Data: COMP1K - Name: mattors--indexnet--indexnet_dimaug_mobv2_1x16_78k_comp1k + Name: indexnet_dimaug Results: - Dataset: COMP1K Metrics: diff --git a/configs/restorers/basicvsr/metafile.yml b/configs/restorers/basicvsr/metafile.yml index 5b7896a816..58fd950e36 100644 --- a/configs/restorers/basicvsr/metafile.yml +++ b/configs/restorers/basicvsr/metafile.yml @@ -11,7 +11,7 @@ Models: In Collection: BasicVSR Metadata: Training Data: REDS - Name: https:----github.com--open-mmlab--mmediting--blob--master--restorers--basicvsr--basicvsr_reds4 + Name: basicvsr_reds4 Results: - Dataset: REDS Metrics: @@ -39,7 +39,7 @@ Models: In Collection: BasicVSR Metadata: Training Data: VIMEO90K - Name: https:----github.com--open-mmlab--mmediting--blob--master--restorers--basicvsr--basicvsr_vimeo90k_bi + Name: basicvsr_bi Results: - Dataset: VIMEO90K Metrics: @@ -67,7 +67,7 @@ Models: In Collection: BasicVSR Metadata: Training Data: VIMEO90K - Name: https:----github.com--open-mmlab--mmediting--blob--master--restorers--basicvsr--basicvsr_vimeo90k_bd + Name: basicvsr_vimeo90k Results: - Dataset: VIMEO90K Metrics: diff --git a/configs/restorers/basicvsr_plusplus/metafile.yml b/configs/restorers/basicvsr_plusplus/metafile.yml index ab1a7c1df0..1279392aa6 100644 --- a/configs/restorers/basicvsr_plusplus/metafile.yml +++ b/configs/restorers/basicvsr_plusplus/metafile.yml @@ -11,7 +11,7 @@ Models: In Collection: BasicVSR++ Metadata: Training Data: REDS - Name: restorers--basicvsr--basicvsr_plusplus_c64n7_8x1_600k_reds4 + Name: basicvsr_c64n7 Results: - Dataset: REDS Metrics: @@ -39,7 +39,7 @@ Models: In Collection: BasicVSR++ Metadata: Training Data: VIMEO90K - Name: restorers--basicvsr--basicvsr_plusplus_c64n7_4x2_300k_vimeo90k_bi + Name: basicvsr_300k Results: - Dataset: VIMEO90K Metrics: @@ -67,7 +67,7 @@ Models: In Collection: BasicVSR++ Metadata: Training Data: VIMEO90K - Name: restorers--basicvsr--basicvsr_plusplus_c64n7_4x2_300k_vimeo90k_bd + Name: basicvsr_4x2 Results: - Dataset: VIMEO90K Metrics: diff --git a/configs/restorers/dic/metafile.yml b/configs/restorers/dic/metafile.yml index cb19ae612e..ce39015949 100644 --- a/configs/restorers/dic/metafile.yml +++ b/configs/restorers/dic/metafile.yml @@ -11,7 +11,7 @@ Models: In Collection: DIC Metadata: Training Data: CELEBA - Name: restorers--dic--dic_x8c48b6_g4_150k_CelebAHQ + Name: dic_150k Results: - Dataset: CELEBA Metrics: @@ -24,7 +24,7 @@ Models: In Collection: DIC Metadata: Training Data: CELEBA - Name: restorers--dic--dic_gan_x8c48b6_g4_500k_CelebAHQ + Name: dic_gan Results: - Dataset: CELEBA Metrics: diff --git a/configs/restorers/edsr/metafile.yml b/configs/restorers/edsr/metafile.yml index 0ac627b77a..e2ab2b6925 100644 --- a/configs/restorers/edsr/metafile.yml +++ b/configs/restorers/edsr/metafile.yml @@ -11,7 +11,7 @@ Models: In Collection: EDSR Metadata: Training Data: DIV2K - Name: restorers--edsr--edsr_x2c64b16_g1_300k_div2k + Name: edsr_x2c64b16 Results: - Dataset: DIV2K Metrics: @@ -30,7 +30,7 @@ Models: In Collection: EDSR Metadata: Training Data: DIV2K - Name: restorers--edsr--edsr_x3c64b16_g1_300k_div2k + Name: edsr_x3c64b16 Results: - Dataset: DIV2K Metrics: @@ -49,7 +49,7 @@ Models: In Collection: EDSR Metadata: Training Data: DIV2K - Name: restorers--edsr--edsr_x4c64b16_g1_300k_div2k + Name: edsr_x4c64b16 Results: - Dataset: DIV2K Metrics: diff --git a/configs/restorers/edvr/metafile.yml b/configs/restorers/edvr/metafile.yml index c125a0e5af..e6fdfc7c64 100644 --- a/configs/restorers/edvr/metafile.yml +++ b/configs/restorers/edvr/metafile.yml @@ -11,7 +11,7 @@ Models: In Collection: EDVR Metadata: Training Data: REDS - Name: restorers--edvr--edvrm_wotsa_x4_g8_600k_reds + Name: edvrm_wotsa Results: - Dataset: REDS Metrics: @@ -24,7 +24,7 @@ Models: In Collection: EDVR Metadata: Training Data: REDS - Name: restorers--edvr--edvrm_x4_g8_600k_reds + Name: edvrm Results: - Dataset: REDS Metrics: diff --git a/configs/restorers/esrgan/metafile.yml b/configs/restorers/esrgan/metafile.yml index fc5a5f9259..49ca9b91e5 100644 --- a/configs/restorers/esrgan/metafile.yml +++ b/configs/restorers/esrgan/metafile.yml @@ -11,7 +11,7 @@ Models: In Collection: ESRGAN Metadata: Training Data: DIV2K - Name: restorers--esrgan--esrgan_psnr_x4c64b23g32_g1_1000k_div2k + Name: esrgan_psnr Results: - Dataset: DIV2K Metrics: @@ -30,7 +30,7 @@ Models: In Collection: ESRGAN Metadata: Training Data: DIV2K - Name: restorers--esrgan--esrgan_x4c64b23g32_g1_400k_div2k + Name: esrgan_400k Results: - Dataset: DIV2K Metrics: diff --git a/configs/restorers/glean/README.md b/configs/restorers/glean/README.md index e7394b46df..02b9d9732f 100644 --- a/configs/restorers/glean/README.md +++ b/configs/restorers/glean/README.md @@ -28,7 +28,7 @@ We show that pre-trained Generative Adversarial Networks (GANs), e.g., StyleGAN, For the meta info used in training and test, please refer to [here](https://github.com/ckkelvinchan/GLEAN). The results are evaluated on RGB channels. | Method | PSNR | Download | -|--------------------------------------------------------------------------------------------------------------------|-------|-------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------| +| :----------------------------------------------------------------------------------------------------------------: | :---: | :---------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------: | | [glean_cat_8x](/configs/restorers/glean/glean_cat_8x.py) | 23.98 | [model](https://download.openmmlab.com/mmediting/restorers/glean/glean_cat_8x_20210614-d3ac8683.pth) \| [log](https://download.openmmlab.com/mmediting/restorers/glean/glean_cat_8x_20210614_145540.log.json) | | [glean_ffhq_16x](/configs/restorers/glean/glean_ffhq_16x.py) | 26.91 | [model](https://download.openmmlab.com/mmediting/restorers/glean/glean_ffhq_16x_20210527-61a3afad.pth) \| [log](https://download.openmmlab.com/mmediting/restorers/glean/glean_ffhq_16x_20210527_194536.log.json) | | [glean_cat_16x](/configs/restorers/glean/glean_cat_16x.py) | 20.88 | [model](https://download.openmmlab.com/mmediting/restorers/glean/glean_cat_16x_20210527-68912543.pth) \| [log](https://download.openmmlab.com/mmediting/restorers/glean/glean_cat_16x_20210527_103708.log.json) | diff --git a/configs/restorers/glean/metafile.yml b/configs/restorers/glean/metafile.yml index 4d9c09acf8..8370ebddb5 100644 --- a/configs/restorers/glean/metafile.yml +++ b/configs/restorers/glean/metafile.yml @@ -6,4 +6,48 @@ Collections: Paper: - https://arxiv.org/abs/2012.00739 README: configs/restorers/glean/README.md -Models: [] +Models: +- Config: configs/restorers/glean/glean_cat_8x.py + In Collection: GLEAN + Metadata: + Training Data: CAT + Name: glean_cat + Results: + - Dataset: CAT + Metrics: + PSNR: 23.98 + Task: restorers + Weights: https://download.openmmlab.com/mmediting/restorers/glean/glean_cat_8x_20210614-d3ac8683.pth +- Config: configs/restorers/glean/glean_ffhq_16x.py + In Collection: GLEAN + Metadata: + Training Data: FFHQ + Name: glean_ffhq + Results: + - Dataset: FFHQ + Metrics: + PSNR: 26.91 + Task: restorers + Weights: https://download.openmmlab.com/mmediting/restorers/glean/glean_ffhq_16x_20210527-61a3afad.pth +- Config: configs/restorers/glean/glean_cat_16x.py + In Collection: GLEAN + Metadata: + Training Data: CAT + Name: glean_16x + Results: + - Dataset: CAT + Metrics: + PSNR: 20.88 + Task: restorers + Weights: https://download.openmmlab.com/mmediting/restorers/glean/glean_cat_16x_20210527-68912543.pth +- Config: configs/restorers/glean/glean_in128out1024_4x2_300k_ffhq_celebahq.py + In Collection: GLEAN + Metadata: + Training Data: CELEBA + Name: glean_in128out1024 + Results: + - Dataset: CELEBA + Metrics: + PSNR: 27.94 + Task: restorers + Weights: https://download.openmmlab.com/mmediting/restorers/glean/glean_in128out1024_4x2_300k_ffhq_celebahq_20210812-acbcb04f.pth diff --git a/configs/restorers/iconvsr/README.md b/configs/restorers/iconvsr/README.md index b6720ecef2..83ecf14124 100644 --- a/configs/restorers/iconvsr/README.md +++ b/configs/restorers/iconvsr/README.md @@ -31,6 +31,6 @@ The pretrained weights of the IconVSR components can be found here: [SPyNet](htt | Method | REDS4 (BIx4)
PSNR/SSIM (RGB) | Vimeo-90K-T (BIx4)
PSNR/SSIM (Y) | Vid4 (BIx4)
PSNR/SSIM (Y) | UDM10 (BDx4)
PSNR/SSIM (Y) | Vimeo-90K-T (BDx4)
PSNR/SSIM (Y) | Vid4 (BDx4)
PSNR/SSIM (Y) | Download | |:---------------------------------------------------------------------------------------------------------------------------:|:-------------------------------:|:-----------------------------------:|:----------------------------:|:-----------------------------:|:-----------------------------------:|:----------------------------:|:-------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------:| -| [iconvsr_reds4](https://github.com/open-mmlab/mmediting/blob/master/configs/restorers/iconvsr/iconvsr_reds.py) | **31.6926/0.8951** | 36.4983/0.9416 | **27.4809/0.8354** | 35.3377/0.9471 | 34.4299/0.9287 | 25.2110/0.7732 | [model](https://download.openmmlab.com/mmediting/restorers/iconvsr/iconvsr_reds4_20210413-9e09d621.pth) \| [log](https://download.openmmlab.com/mmediting/restorers/iconvsr/iconvsr_reds4_20210413_222735.log.json) | +| [iconvsr_reds4](https://github.com/open-mmlab/mmediting/blob/master/configs/restorers/iconvsr/iconvsr_reds4.py) | **31.6926/0.8951** | 36.4983/0.9416 | **27.4809/0.8354** | 35.3377/0.9471 | 34.4299/0.9287 | 25.2110/0.7732 | [model](https://download.openmmlab.com/mmediting/restorers/iconvsr/iconvsr_reds4_20210413-9e09d621.pth) \| [log](https://download.openmmlab.com/mmediting/restorers/iconvsr/iconvsr_reds4_20210413_222735.log.json) | | [iconvsr_vimeo90k_bi](https://github.com/open-mmlab/mmediting/blob/master/configs/restorers/iconvsr/iconvsr_vimeo90k_bi.py) | 30.3452/0.8659 | **37.3729/0.9467** | 27.4238/0.8297 | 34.2595/0.9398 | 34.5548/0.9295 | 24.6666/0.7491 | [model](https://download.openmmlab.com/mmediting/restorers/iconvsr/iconvsr_vimeo90k_bi_20210413-7c7418dc.pth) \| [log](https://download.openmmlab.com/mmediting/restorers/iconvsr/iconvsr_vimeo90k_bi_20210413_222757.log.json) | | [iconvsr_vimeo90k_bd](https://github.com/open-mmlab/mmediting/blob/master/configs/restorers/iconvsr/iconvsr_vimeo90k_bd.py) | 29.0150/0.8465 | 34.6780/0.9339 | 26.3109/0.8028 | **40.0640/0.9697** | **37.7573/0.9517** | **28.2464/0.8612** | [model](https://download.openmmlab.com/mmediting/restorers/iconvsr/iconvsr_vimeo90k_bd_20210414-5f38cb34.pth) \| [log](https://download.openmmlab.com/mmediting/restorers/iconvsr/iconvsr_vimeo90k_bd_20210414_084128.log.json) | diff --git a/configs/restorers/iconvsr/metafile.yml b/configs/restorers/iconvsr/metafile.yml index 7de0fc61cd..7bc81c7a02 100644 --- a/configs/restorers/iconvsr/metafile.yml +++ b/configs/restorers/iconvsr/metafile.yml @@ -7,11 +7,11 @@ Collections: - https://arxiv.org/abs/2012.02181 README: configs/restorers/iconvsr/README.md Models: -- Config: https://github.com/open-mmlab/mmediting/blob/master/configs/restorers/iconvsr/iconvsr_reds.py +- Config: https://github.com/open-mmlab/mmediting/blob/master/configs/restorers/iconvsr/iconvsr_reds4.py In Collection: IconVSR Metadata: Training Data: REDS - Name: https:----github.com--open-mmlab--mmediting--blob--master--restorers--iconvsr--iconvsr_reds + Name: iconvsr_reds4 Results: - Dataset: REDS Metrics: @@ -39,7 +39,7 @@ Models: In Collection: IconVSR Metadata: Training Data: VIMEO90K - Name: https:----github.com--open-mmlab--mmediting--blob--master--restorers--iconvsr--iconvsr_vimeo90k_bi + Name: iconvsr_bi Results: - Dataset: VIMEO90K Metrics: @@ -67,7 +67,7 @@ Models: In Collection: IconVSR Metadata: Training Data: VIMEO90K - Name: https:----github.com--open-mmlab--mmediting--blob--master--restorers--iconvsr--iconvsr_vimeo90k_bd + Name: iconvsr_vimeo90k Results: - Dataset: VIMEO90K Metrics: diff --git a/configs/restorers/liif/metafile.yml b/configs/restorers/liif/metafile.yml index 1842ed3fbc..e6ac7db782 100644 --- a/configs/restorers/liif/metafile.yml +++ b/configs/restorers/liif/metafile.yml @@ -11,7 +11,7 @@ Models: In Collection: LIIF Metadata: Training Data: DIV2K - Name: restorers--liif--liif_edsr_norm_c64b16_g1_1000k_div2k + Name: liif_edsr Results: - Dataset: DIV2K Metrics: @@ -30,7 +30,7 @@ Models: In Collection: LIIF Metadata: Training Data: DIV2K - Name: restorers--liif--liif_edsr_norm_c64b16_g1_1000k_div2k + Name: liif_edsr Results: - Dataset: DIV2K Metrics: @@ -49,7 +49,7 @@ Models: In Collection: LIIF Metadata: Training Data: DIV2K - Name: restorers--liif--liif_edsr_norm_c64b16_g1_1000k_div2k + Name: liif_edsr Results: - Dataset: DIV2K Metrics: @@ -68,7 +68,7 @@ Models: In Collection: LIIF Metadata: Training Data: DIV2K - Name: restorers--liif--liif_edsr_norm_c64b16_g1_1000k_div2k + Name: liif_edsr Results: - Dataset: DIV2K Metrics: @@ -87,7 +87,7 @@ Models: In Collection: LIIF Metadata: Training Data: DIV2K - Name: restorers--liif--liif_edsr_norm_c64b16_g1_1000k_div2k + Name: liif_edsr Results: - Dataset: DIV2K Metrics: @@ -106,7 +106,7 @@ Models: In Collection: LIIF Metadata: Training Data: DIV2K - Name: restorers--liif--liif_edsr_norm_c64b16_g1_1000k_div2k + Name: liif_edsr Results: - Dataset: DIV2K Metrics: @@ -125,7 +125,7 @@ Models: In Collection: LIIF Metadata: Training Data: DIV2K - Name: restorers--liif--liif_rdn_norm_x2-4_c64b16_g1_1000k_div2k + Name: liif_rdn Results: - Dataset: DIV2K Metrics: @@ -144,7 +144,7 @@ Models: In Collection: LIIF Metadata: Training Data: DIV2K - Name: restorers--liif--liif_rdn_norm_x2-4_c64b16_g1_1000k_div2k + Name: liif_rdn Results: - Dataset: DIV2K Metrics: @@ -163,7 +163,7 @@ Models: In Collection: LIIF Metadata: Training Data: DIV2K - Name: restorers--liif--liif_rdn_norm_x2-4_c64b16_g1_1000k_div2k + Name: liif_rdn Results: - Dataset: DIV2K Metrics: @@ -182,7 +182,7 @@ Models: In Collection: LIIF Metadata: Training Data: DIV2K - Name: restorers--liif--liif_rdn_norm_x2-4_c64b16_g1_1000k_div2k + Name: liif_rdn Results: - Dataset: DIV2K Metrics: @@ -201,7 +201,7 @@ Models: In Collection: LIIF Metadata: Training Data: DIV2K - Name: restorers--liif--liif_rdn_norm_x2-4_c64b16_g1_1000k_div2k + Name: liif_rdn Results: - Dataset: DIV2K Metrics: @@ -220,7 +220,7 @@ Models: In Collection: LIIF Metadata: Training Data: DIV2K - Name: restorers--liif--liif_rdn_norm_x2-4_c64b16_g1_1000k_div2k + Name: liif_rdn Results: - Dataset: DIV2K Metrics: diff --git a/configs/restorers/rdn/metafile.yml b/configs/restorers/rdn/metafile.yml index f28f3973bc..e3e65189b2 100644 --- a/configs/restorers/rdn/metafile.yml +++ b/configs/restorers/rdn/metafile.yml @@ -11,7 +11,7 @@ Models: In Collection: RDN Metadata: Training Data: DIV2K - Name: restorers--rdn--rdn_x2c64b16_g1_1000k_div2k + Name: rdn_x2c64b16 Results: - Dataset: DIV2K Metrics: @@ -30,7 +30,7 @@ Models: In Collection: RDN Metadata: Training Data: DIV2K - Name: restorers--rdn--rdn_x3c64b16_g1_1000k_div2k + Name: rdn_x3c64b16 Results: - Dataset: DIV2K Metrics: @@ -49,7 +49,7 @@ Models: In Collection: RDN Metadata: Training Data: DIV2K - Name: restorers--rdn--rdn_x4c64b16_g1_1000k_div2k + Name: rdn_x4c64b16 Results: - Dataset: DIV2K Metrics: diff --git a/configs/restorers/real_esrgan/metafile.yml b/configs/restorers/real_esrgan/metafile.yml index 21ac6047ad..9172213cd3 100644 --- a/configs/restorers/real_esrgan/metafile.yml +++ b/configs/restorers/real_esrgan/metafile.yml @@ -11,7 +11,7 @@ Models: In Collection: Real-ESRGAN Metadata: Training Data: Others - Name: restorers--real_esrgan--realesrnet_c64b23g32_12x4_lr2e-4_1000k_df2k_ost + Name: realesrnet Results: - Dataset: Others Metrics: diff --git a/configs/restorers/srcnn/metafile.yml b/configs/restorers/srcnn/metafile.yml index 4d039b2c5f..0e3053afc2 100644 --- a/configs/restorers/srcnn/metafile.yml +++ b/configs/restorers/srcnn/metafile.yml @@ -11,7 +11,7 @@ Models: In Collection: SRCNN Metadata: Training Data: DIV2K - Name: restorers--srcnn--srcnn_x4k915_g1_1000k_div2k + Name: srcnn Results: - Dataset: DIV2K Metrics: diff --git a/configs/restorers/srresnet_srgan/metafile.yml b/configs/restorers/srresnet_srgan/metafile.yml index 5ad18de6f6..daed3a1a7e 100644 --- a/configs/restorers/srresnet_srgan/metafile.yml +++ b/configs/restorers/srresnet_srgan/metafile.yml @@ -11,7 +11,7 @@ Models: In Collection: SRGAN Metadata: Training Data: DIV2K - Name: restorers--srresnet_srgan--msrresnet_x4c64b16_g1_1000k_div2k + Name: msrresnet Results: - Dataset: DIV2K Metrics: @@ -30,7 +30,7 @@ Models: In Collection: SRGAN Metadata: Training Data: DIV2K - Name: restorers--srresnet_srgan--srgan_x4c64b16_g1_1000k_div2k + Name: srgan Results: - Dataset: DIV2K Metrics: diff --git a/configs/restorers/tdan/metafile.yml b/configs/restorers/tdan/metafile.yml index 50e27dcf56..b4a4a0a380 100644 --- a/configs/restorers/tdan/metafile.yml +++ b/configs/restorers/tdan/metafile.yml @@ -11,7 +11,7 @@ Models: In Collection: TDAN Metadata: Training Data: VIMEO90K - Name: restorers--tdan--tdan_vimeo90k_bix4_ft_lr5e-5_400k + Name: tdan_lr5e-5 Results: - Dataset: VIMEO90K Metrics: @@ -33,7 +33,7 @@ Models: In Collection: TDAN Metadata: Training Data: VIMEO90K - Name: restorers--tdan--tdan_vimeo90k_bdx4_ft_lr5e-5_800k + Name: tdan_ft Results: - Dataset: VIMEO90K Metrics: diff --git a/configs/restorers/tof/metafile.yml b/configs/restorers/tof/metafile.yml index f87989c2cd..904bf94f62 100644 --- a/configs/restorers/tof/metafile.yml +++ b/configs/restorers/tof/metafile.yml @@ -11,7 +11,7 @@ Models: In Collection: TOFlow Metadata: Training Data: VIMEO90K - Name: restorers--tof--tof_x4_vimeo90k_official + Name: tof Results: - Dataset: VIMEO90K Metrics: diff --git a/configs/restorers/ttsr/metafile.yml b/configs/restorers/ttsr/metafile.yml index 5669a30d51..4307659ea8 100644 --- a/configs/restorers/ttsr/metafile.yml +++ b/configs/restorers/ttsr/metafile.yml @@ -11,7 +11,7 @@ Models: In Collection: TTSR Metadata: Training Data: CUFED - Name: restorers--ttsr--ttsr-rec_x4_c64b16_g1_200k_CUFED + Name: ttsr-rec_200k Results: - Dataset: CUFED Metrics: @@ -24,7 +24,7 @@ Models: In Collection: TTSR Metadata: Training Data: CUFED - Name: restorers--ttsr--ttsr-gan_x4_c64b16_g1_500k_CUFED + Name: ttsr-gan_500k Results: - Dataset: CUFED Metrics: diff --git a/configs/synthesizers/cyclegan/metafile.yml b/configs/synthesizers/cyclegan/metafile.yml index b37e7fc441..4520652fbe 100644 --- a/configs/synthesizers/cyclegan/metafile.yml +++ b/configs/synthesizers/cyclegan/metafile.yml @@ -11,7 +11,7 @@ Models: In Collection: CycleGAN Metadata: Training Data: FACADES - Name: synthesizers--cyclegan--cyclegan_lsgan_resnet_in_1x1_80k_facades + Name: cyclegan_80k Results: - Dataset: FACADES Metrics: @@ -23,7 +23,7 @@ Models: In Collection: CycleGAN Metadata: Training Data: FACADES - Name: synthesizers--cyclegan--cyclegan_lsgan_id0_resnet_in_1x1_80k_facades + Name: cyclegan_id0 Results: - Dataset: FACADES Metrics: @@ -35,7 +35,7 @@ Models: In Collection: CycleGAN Metadata: Training Data: SUMMER2WINTER - Name: synthesizers--cyclegan--cyclegan_lsgan_resnet_in_1x1_246200_summer2winter + Name: cyclegan_246200 Results: - Dataset: SUMMER2WINTER Metrics: @@ -47,7 +47,7 @@ Models: In Collection: CycleGAN Metadata: Training Data: SUMMER2WINTER - Name: synthesizers--cyclegan--cyclegan_lsgan_resnet_in_1x1_246200_summer2winter + Name: cyclegan_246200 Results: - Dataset: SUMMER2WINTER Metrics: @@ -59,7 +59,7 @@ Models: In Collection: CycleGAN Metadata: Training Data: SUMMER2WINTER - Name: synthesizers--cyclegan--cyclegan_lsgan_id0_resnet_in_1x1_246200_summer2winter + Name: cyclegan_summer2winter Results: - Dataset: SUMMER2WINTER Metrics: @@ -71,7 +71,7 @@ Models: In Collection: CycleGAN Metadata: Training Data: SUMMER2WINTER - Name: synthesizers--cyclegan--cyclegan_lsgan_id0_resnet_in_1x1_246200_summer2winter + Name: cyclegan_summer2winter Results: - Dataset: SUMMER2WINTER Metrics: @@ -83,7 +83,7 @@ Models: In Collection: CycleGAN Metadata: Training Data: HORSE2ZEBRA - Name: synthesizers--cyclegan--cyclegan_lsgan_resnet_in_1x1_266800_horse2zebra + Name: cyclegan_266800 Results: - Dataset: HORSE2ZEBRA Metrics: @@ -95,7 +95,7 @@ Models: In Collection: CycleGAN Metadata: Training Data: HORSE2ZEBRA - Name: synthesizers--cyclegan--cyclegan_lsgan_id0_resnet_in_1x1_266800_horse2zebra + Name: cyclegan_horse2zebra Results: - Dataset: HORSE2ZEBRA Metrics: @@ -107,7 +107,7 @@ Models: In Collection: CycleGAN Metadata: Training Data: HORSE2ZEBRA - Name: synthesizers--cyclegan--cyclegan_lsgan_resnet_in_1x1_266800_horse2zebra + Name: cyclegan_266800 Results: - Dataset: HORSE2ZEBRA Metrics: @@ -119,7 +119,7 @@ Models: In Collection: CycleGAN Metadata: Training Data: HORSE2ZEBRA - Name: synthesizers--cyclegan--cyclegan_lsgan_id0_resnet_in_1x1_266800_horse2zebra + Name: cyclegan_horse2zebra Results: - Dataset: HORSE2ZEBRA Metrics: diff --git a/configs/synthesizers/pix2pix/metafile.yml b/configs/synthesizers/pix2pix/metafile.yml index 856e29a3b8..b08bd5f55f 100644 --- a/configs/synthesizers/pix2pix/metafile.yml +++ b/configs/synthesizers/pix2pix/metafile.yml @@ -11,7 +11,7 @@ Models: In Collection: Pix2Pix Metadata: Training Data: FACADES - Name: synthesizers--pix2pix--pix2pix_vanilla_unet_bn_1x1_80k_facades + Name: pix2pix_1x1 Results: - Dataset: FACADES Metrics: @@ -23,7 +23,7 @@ Models: In Collection: Pix2Pix Metadata: Training Data: MAPS - Name: synthesizers--pix2pix--pix2pix_vanilla_unet_bn_a2b_1x1_219200_maps + Name: pix2pix_a2b Results: - Dataset: MAPS Metrics: @@ -35,7 +35,7 @@ Models: In Collection: Pix2Pix Metadata: Training Data: MAPS - Name: synthesizers--pix2pix--pix2pix_vanilla_unet_bn_b2a_1x1_219200_maps + Name: pix2pix_b2a Results: - Dataset: MAPS Metrics: @@ -47,7 +47,7 @@ Models: In Collection: Pix2Pix Metadata: Training Data: EDGES2SHOES - Name: synthesizers--pix2pix--pix2pix_vanilla_unet_bn_wo_jitter_flip_1x4_186840_edges2shoes + Name: pix2pix_wo Results: - Dataset: EDGES2SHOES Metrics: