open-mmlab · zengyh1900 · Dec 29, 2022 · Dec 28, 2022 · Dec 29, 2022
diff --git a/.dev_scripts/update_model_index.py b/.dev_scripts/update_model_index.py
@@ -144,20 +144,26 @@ def parse_md(md_file):
         Name=collection_name,
         Metadata={'Architecture': []},
         README=readme,
-        Paper=[])
+        Paper=[],
+        Task=[],
+        Year=0,
+    )
     models = []
     # force utf-8 instead of system defined
     with open(md_file, 'r', encoding='utf-8') as md:
         lines = md.readlines()
         i = 0
         name = lines[0][2:]
+        year = re.sub('[^0-9]', '', name.split('(', 1)[-1])
         name = name.split('(', 1)[0].strip()
         collection['Metadata']['Architecture'].append(name)
         collection['Name'] = name
         collection_name = name
         is_liif = collection_name.upper() == 'LIIF'
         task_line = lines[4]
         task = task_line.strip().split(':')[-1].strip()
+        collection['Task'] = task.lower().split(', ')
+        collection['Year'] = int(year)
         while i < len(lines):
             # parse reference
             if lines[i].startswith('> ['):

diff --git a/configs/aot_gan/metafile.yml b/configs/aot_gan/metafile.yml
@@ -6,6 +6,9 @@ Collections:
   Paper:
   - https://arxiv.org/pdf/2104.01431.pdf
   README: configs/aot_gan/README.md
+  Task:
+  - inpainting
+  Year: 2021
 Models:
 - Config: configs/aot_gan/aot-gan_smpgan_4xb4_places-512x512.py
   In Collection: AOT-GAN

diff --git a/configs/basicvsr/metafile.yml b/configs/basicvsr/metafile.yml
@@ -6,6 +6,9 @@ Collections:
   Paper:
   - https://arxiv.org/abs/2012.02181
   README: configs/basicvsr/README.md
+  Task:
+  - video super-resolution
+  Year: 2021
 Models:
 - Config: configs/basicvsr/basicvsr_2xb4_reds4.py
   In Collection: BasicVSR

diff --git a/configs/basicvsr_pp/metafile.yml b/configs/basicvsr_pp/metafile.yml
@@ -6,6 +6,9 @@ Collections:
   Paper:
   - https://arxiv.org/abs/2104.13371
   README: configs/basicvsr_pp/README.md
+  Task:
+  - video super-resolution
+  Year: 2022
 Models:
 - Config: configs/basicvsr_pp/basicvsr-pp_c64n7_8xb1-600k_reds4.py
   In Collection: BasicVSR++

diff --git a/configs/biggan/metafile.yml b/configs/biggan/metafile.yml
@@ -6,6 +6,9 @@ Collections:
   Paper:
   - https://openreview.net/forum?id=B1xsqj09Fm
   README: configs/biggan/README.md
+  Task:
+  - conditional gans
+  Year: 2019
 Models:
 - Config: configs/biggan/biggan_2xb25-500kiters_cifar10-32x32.py
   In Collection: BigGAN

diff --git a/configs/cain/metafile.yml b/configs/cain/metafile.yml
@@ -6,6 +6,9 @@ Collections:
   Paper:
   - https://aaai.org/ojs/index.php/AAAI/article/view/6693/6547
   README: configs/cain/README.md
+  Task:
+  - video interpolation
+  Year: 2020
 Models:
 - Config: configs/cain/cain_g1b32_1xb5_vimeo90k-triplet.py
   In Collection: CAIN

diff --git a/configs/cyclegan/metafile.yml b/configs/cyclegan/metafile.yml
@@ -8,6 +8,9 @@ Collections:
   Paper:
   - https://openaccess.thecvf.com/content_iccv_2017/html/Zhu_Unpaired_Image-To-Image_Translation_ICCV_2017_paper.html
   README: configs/cyclegan/README.md
+  Task:
+  - image2image
+  Year: 2017
 Models:
 - Config: configs/cyclegan/cyclegan_lsgan-resnet-in_1xb1-80kiters_facades.py
   In Collection: 'CycleGAN: Unpaired Image-to-Image Translation Using Cycle-Consistent

diff --git a/configs/dcgan/metafile.yml b/configs/dcgan/metafile.yml
@@ -8,6 +8,9 @@ Collections:
   Paper:
   - https://arxiv.org/abs/1511.06434
   README: configs/dcgan/README.md
+  Task:
+  - unconditional gans
+  Year: 2016
 Models:
 - Config: configs/dcgan/dcgan_Glr4e-4_Dlr1e-4_1xb128-5kiters_mnist-64x64.py
   In Collection: Unsupervised Representation Learning with Deep Convolutional Generative

diff --git a/configs/deepfillv1/metafile.yml b/configs/deepfillv1/metafile.yml
@@ -6,6 +6,9 @@ Collections:
   Paper:
   - https://arxiv.org/abs/1801.07892
   README: configs/deepfillv1/README.md
+  Task:
+  - inpainting
+  Year: 2018
 Models:
 - Config: configs/deepfillv1/deepfillv1_8xb2_places-256x256.py
   In Collection: DeepFillv1

diff --git a/configs/deepfillv2/metafile.yml b/configs/deepfillv2/metafile.yml
@@ -6,6 +6,9 @@ Collections:
   Paper:
   - https://arxiv.org/abs/1806.03589
   README: configs/deepfillv2/README.md
+  Task:
+  - inpainting
+  Year: 2019
 Models:
 - Config: configs/deepfillv2/deepfillv2_8xb2_places-256x256.py
   In Collection: DeepFillv2

diff --git a/configs/dic/metafile.yml b/configs/dic/metafile.yml
@@ -6,6 +6,9 @@ Collections:
   Paper:
   - https://arxiv.org/abs/2003.13063
   README: configs/dic/README.md
+  Task:
+  - image super-resolution
+  Year: 2020
 Models:
 - Config: configs/dic/dic_x8c48b6_4xb2-150k_celeba-hq.py
   In Collection: DIC

diff --git a/configs/dim/metafile.yml b/configs/dim/metafile.yml
@@ -6,6 +6,9 @@ Collections:
   Paper:
   - https://arxiv.org/abs/1703.03872
   README: configs/dim/README.md
+  Task:
+  - matting
+  Year: 2017
 Models:
 - Config: configs/dim/dim_stage1-v16_1xb1-1000k_comp1k.py
   In Collection: DIM

diff --git a/configs/disco_diffusion/metafile.yml b/configs/disco_diffusion/metafile.yml
@@ -6,6 +6,11 @@ Collections:
   Paper:
   - https://github.com/alembics/disco-diffusion
   README: configs/disco_diffusion/README.md
+  Task:
+  - text2image
+  - image2image
+  - diffusion
+  Year: 2022
 Models:
 - Config: configs/disco_diffusion/disco-diffusion_adm-u-finetuned_imagenet-512x512.py
   In Collection: Disco Diffusion

diff --git a/configs/edsr/metafile.yml b/configs/edsr/metafile.yml
@@ -6,6 +6,9 @@ Collections:
   Paper:
   - https://arxiv.org/abs/1707.02921
   README: configs/edsr/README.md
+  Task:
+  - image super-resolution
+  Year: 2017
 Models:
 - Config: configs/edsr/edsr_x2c64b16_1xb16-300k_div2k.py
   In Collection: EDSR

diff --git a/configs/edvr/metafile.yml b/configs/edvr/metafile.yml
@@ -6,6 +6,9 @@ Collections:
   Paper:
   - https://arxiv.org/abs/1905.02716?utm_source=feedburner&utm_medium=feed&utm_campaign=Feed%253A+arxiv%252FQSXk+%2528ExcitingAds%2521+cs+updates+on+arXiv.org%2529
   README: configs/edvr/README.md
+  Task:
+  - video super-resolution
+  Year: 2019
 Models:
 - Config: configs/edvr/edvrm_wotsa_8xb4-600k_reds.py
   In Collection: EDVR

diff --git a/configs/eg3d/metafile.yml b/configs/eg3d/metafile.yml
@@ -6,6 +6,9 @@ Collections:
   Paper:
   - https://openaccess.thecvf.com/content/CVPR2022/html/Chan_Efficient_Geometry-Aware_3D_Generative_Adversarial_Networks_CVPR_2022_paper.html
   README: configs/eg3d/README.md
+  Task:
+  - 3d-aware generation
+  Year: 2022
 Models:
 - Config: configs/eg3d/eg3d_cvt-official-rgb_shapenet-128x128.py
   In Collection: EG3D

diff --git a/configs/esrgan/metafile.yml b/configs/esrgan/metafile.yml
@@ -6,6 +6,9 @@ Collections:
   Paper:
   - https://arxiv.org/abs/1809.00219
   README: configs/esrgan/README.md
+  Task:
+  - image super-resolution
+  Year: 2018
 Models:
 - Config: configs/esrgan/esrgan_psnr-x4c64b23g32_1xb16-1000k_div2k.py
   In Collection: ESRGAN

diff --git a/configs/flavr/metafile.yml b/configs/flavr/metafile.yml
@@ -6,6 +6,9 @@ Collections:
   Paper:
   - https://arxiv.org/pdf/2012.08512.pdf
   README: configs/flavr/README.md
+  Task:
+  - video interpolation
+  Year: 2020
 Models:
 - Config: configs/flavr/flavr_in4out1_8xb4_vimeo90k-septuplet.py
   In Collection: FLAVR

diff --git a/configs/gca/metafile.yml b/configs/gca/metafile.yml
@@ -6,6 +6,9 @@ Collections:
   Paper:
   - https://arxiv.org/abs/2001.04069
   README: configs/gca/README.md
+  Task:
+  - matting
+  Year: 2020
 Models:
 - Config: configs/gca/baseline_r34_4xb10-200k_comp1k.py
   In Collection: GCA

diff --git a/configs/ggan/metafile.yml b/configs/ggan/metafile.yml
@@ -6,6 +6,9 @@ Collections:
   Paper:
   - https://arxiv.org/abs/1705.02894
   README: configs/ggan/README.md
+  Task:
+  - unconditional gans
+  Year: 2017
 Models:
 - Config: configs/ggan/ggan_dcgan-archi_lr1e-3-1xb128-12Mimgs_celeba-cropped-64x64.py
   In Collection: GGAN

diff --git a/configs/glean/metafile.yml b/configs/glean/metafile.yml
@@ -6,6 +6,9 @@ Collections:
   Paper:
   - https://arxiv.org/abs/2012.00739
   README: configs/glean/README.md
+  Task:
+  - image super-resolution
+  Year: 2021
 Models:
 - Config: configs/glean/glean_x8_2xb8_cat.py
   In Collection: GLEAN

diff --git a/configs/global_local/metafile.yml b/configs/global_local/metafile.yml
@@ -6,6 +6,9 @@ Collections:
   Paper:
   - http://iizuka.cs.tsukuba.ac.jp/projects/completion/data/completion_sig2017.pdf
   README: configs/global_local/README.md
+  Task:
+  - inpainting
+  Year: 2017
 Models:
 - Config: configs/global_local/gl_8xb12_places-256x256.py
   In Collection: Global&Local

diff --git a/configs/guided_diffusion/metafile.yml b/configs/guided_diffusion/metafile.yml
@@ -6,6 +6,9 @@ Collections:
   Paper:
   - https://papers.nips.cc/paper/2021/file/49ad23d1ec9fa4bd8d77d02681df5cfa-Paper.pdf
   README: configs/guided_diffusion/README.md
+  Task:
+  - image generation
+  Year: 2021
 Models:
 - Config: configs/guided_diffusion/adm-u_8xb32_imagenet-64x64.py
   In Collection: Guided Diffusion

diff --git a/configs/iconvsr/metafile.yml b/configs/iconvsr/metafile.yml
@@ -6,6 +6,9 @@ Collections:
   Paper:
   - https://arxiv.org/abs/2012.02181
   README: configs/iconvsr/README.md
+  Task:
+  - video super-resolution
+  Year: 2021
 Models:
 - Config: configs/iconvsr/iconvsr_2xb4_reds4.py
   In Collection: IconVSR

diff --git a/configs/indexnet/metafile.yml b/configs/indexnet/metafile.yml
@@ -6,6 +6,9 @@ Collections:
   Paper:
   - https://arxiv.org/abs/1908.00672
   README: configs/indexnet/README.md
+  Task:
+  - matting
+  Year: 2019
 Models:
 - Config: configs/indexnet/indexnet_mobv2_1xb16-78k_comp1k.py
   In Collection: IndexNet

diff --git a/configs/inst_colorization/metafile.yml b/configs/inst_colorization/metafile.yml
@@ -6,6 +6,9 @@ Collections:
   Paper:
   - https://openaccess.thecvf.com/content_CVPR_2020/html/Su_Instance-Aware_Image_Colorization_CVPR_2020_paper.html
   README: configs/inst_colorization/README.md
+  Task:
+  - colorization
+  Year: 2020
 Models:
 - Config: configs/inst_colorization/inst-colorizatioon_full_official_cocostuff-256x256.py
   In Collection: Instance-aware Image Colorization

diff --git a/configs/liif/metafile.yml b/configs/liif/metafile.yml
@@ -6,6 +6,9 @@ Collections:
   Paper:
   - https://arxiv.org/abs/2012.09161
   README: configs/liif/README.md
+  Task:
+  - image super-resolution
+  Year: 2021
 Models:
 - Config: configs/liif/liif-edsr-norm_c64b16_1xb16-1000k_div2k.py
   In Collection: LIIF

diff --git a/configs/lsgan/metafile.yml b/configs/lsgan/metafile.yml
@@ -6,6 +6,9 @@ Collections:
   Paper:
   - https://openaccess.thecvf.com/content_iccv_2017/html/Mao_Least_Squares_Generative_ICCV_2017_paper.html
   README: configs/lsgan/README.md
+  Task:
+  - unconditional gans
+  Year: 2017
 Models:
 - Config: configs/lsgan/lsgan_dcgan-archi_lr1e-3-1xb128-12Mimgs_celeba-cropped-64x64.py
   In Collection: LSGAN

diff --git a/configs/nafnet/metafile.yml b/configs/nafnet/metafile.yml
@@ -6,6 +6,9 @@ Collections:
   Paper:
   - https://arxiv.org/abs/2204.04676
   README: configs/nafnet/README.md
+  Task:
+  - image restoration
+  Year: 2022
 Models:
 - Config: configs/nafnet/nafnet_c64eb2248mb12db2222_8xb8-lr1e-3-400k_sidd.py
   In Collection: NAFNet

diff --git a/configs/partial_conv/metafile.yml b/configs/partial_conv/metafile.yml
@@ -6,6 +6,9 @@ Collections:
   Paper:
   - https://arxiv.org/abs/1804.07723
   README: configs/partial_conv/README.md
+  Task:
+  - inpainting
+  Year: 2018
 Models:
 - Config: configs/partial_conv/pconv_stage1_8xb12_places-256x256.py
   In Collection: PConv

diff --git a/configs/pggan/metafile.yml b/configs/pggan/metafile.yml
@@ -6,6 +6,9 @@ Collections:
   Paper:
   - https://arxiv.org/abs/1710.10196
   README: configs/pggan/README.md
+  Task:
+  - unconditional gans
+  Year: 2018
 Models:
 - Config: configs/pggan/pggan_8xb4-12Mimgs_celeba-cropped-128x128.py
   In Collection: PGGAN

diff --git a/configs/pix2pix/metafile.yml b/configs/pix2pix/metafile.yml
@@ -6,6 +6,9 @@ Collections:
   Paper:
   - https://openaccess.thecvf.com/content_cvpr_2017/html/Isola_Image-To-Image_Translation_With_CVPR_2017_paper.html
   README: configs/pix2pix/README.md
+  Task:
+  - image2image
+  Year: 2017
 Models:
 - Config: configs/pix2pix/pix2pix_vanilla-unet-bn_1xb1-80kiters_facades.py
   In Collection: Pix2Pix

diff --git a/configs/positional_encoding_in_gans/README.md b/configs/positional_encoding_in_gans/README.md
@@ -1,4 +1,4 @@
-# Positional Encoding in GANs
+# Positional Encoding in GANs (CVPR'2021)
 
 > [Positional Encoding as Spatial Inductive Bias in GANs](https://openaccess.thecvf.com/content/CVPR2021/html/Xu_Positional_Encoding_As_Spatial_Inductive_Bias_in_GANs_CVPR_2021_paper.html)
 

diff --git a/configs/positional_encoding_in_gans/metafile.yml b/configs/positional_encoding_in_gans/metafile.yml
@@ -6,6 +6,9 @@ Collections:
   Paper:
   - https://openaccess.thecvf.com/content/CVPR2021/html/Xu_Positional_Encoding_As_Spatial_Inductive_Bias_in_GANs_CVPR_2021_paper.html
   README: configs/positional_encoding_in_gans/README.md
+  Task:
+  - unconditional gans
+  Year: 2021
 Models:
 - Config: configs/positional_encoding_in_gans/stylegan2_c2_8xb3-1100kiters_ffhq-256x256.py
   In Collection: Positional Encoding in GANs

diff --git a/configs/rdn/metafile.yml b/configs/rdn/metafile.yml
@@ -6,6 +6,9 @@ Collections:
   Paper:
   - https://arxiv.org/abs/1802.08797
   README: configs/rdn/README.md
+  Task:
+  - image super-resolution
+  Year: 2018
 Models:
 - Config: configs/rdn/rdn_x4c64b16_1xb16-1000k_div2k.py
   In Collection: RDN

diff --git a/configs/real_basicvsr/metafile.yml b/configs/real_basicvsr/metafile.yml
@@ -6,6 +6,9 @@ Collections:
   Paper:
   - https://arxiv.org/abs/2111.12704
   README: configs/real_basicvsr/README.md
+  Task:
+  - video super-resolution
+  Year: 2022
 Models:
 - Config: configs/real_basicvsr/realbasicvsr_c64b20-1x30x8_8xb1-lr5e-5-150k_reds.py
   In Collection: RealBasicVSR

diff --git a/configs/real_esrgan/metafile.yml b/configs/real_esrgan/metafile.yml
@@ -6,6 +6,9 @@ Collections:
   Paper:
   - https://arxiv.org/abs/2107.10833
   README: configs/real_esrgan/README.md
+  Task:
+  - image super-resolution
+  Year: 2021
 Models:
 - Config: configs/real_esrgan/realesrnet_c64b23g32_4xb12-lr2e-4-1000k_df2k-ost.py
   In Collection: Real-ESRGAN

diff --git a/configs/restormer/metafile.yml b/configs/restormer/metafile.yml
@@ -6,6 +6,11 @@ Collections:
   Paper:
   - https://arxiv.org/abs/2111.09881
   README: configs/restormer/README.md
+  Task:
+  - denoising
+  - deblurring
+  - deraining
+  Year: 2022
 Models:
 - Config: configs/restormer/restormer_official_rain13k.py
   In Collection: Restormer