Merge pull request #186 from open-mmlab/zz/stat

New algorithm stat
open-mmlab · Jan 26, 2021 · 506c0e9 · 506c0e9
2 parents 557b7e8 + 6db194d
commit 506c0e9
Show file tree

Hide file tree

Showing 22 changed files with 129 additions and 210 deletions.
diff --git a/configs/inpainting/deepfillv1/README.md b/configs/inpainting/deepfillv1/README.md
@@ -4,7 +4,7 @@
 
 [ALGORITHM]
 
-```
+```latex
 @inproceedings{yu2018generative,
   title={Generative image inpainting with contextual attention},
   author={Yu, Jiahui and Lin, Zhe and Yang, Jimei and Shen, Xiaohui and Lu, Xin and Huang, Thomas S},
@@ -17,12 +17,14 @@
 
 ## Results and models
 ### Places365-Challenge
+
 |   Method   |  Mask Type  | Resolution | Train Iters |   Test Set    | l1 error |  PSNR  | SSIM  |                                                                                                                           Download                                                                                                                            |
 | :--------: | :---------: | :--------: | :---------: | :-----------: | :------: | :----: | :---: | :-----------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------: |
 | DeepFillv1 | square bbox |  256x256   |    3500k    | Places365-val |  11.019  | 23.429 | 0.862 | [model](https://download.openmmlab.com/mmediting/inpainting/deepfillv1/deepfillv1_256x256_8x2_places_20200619-c00a0e21.pth) \| [log](https://download.openmmlab.com/mmediting/inpainting/deepfillv1/deepfillv1_256x256_8x2_places_20200619-c00a0e21.log.json) |
 
 
 ### CelebA-HQ
+
 |   Method   |  Mask Type  | Resolution | Train Iters |  Test Set  | l1 error |  PSNR  | SSIM  |                                                                                                                           Download                                                                                                                            |
 | :--------: | :---------: | :--------: | :---------: | :--------: | :------: | :----: | :---: | :-----------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------: |
 | DeepFillv1 | square bbox |  256x256   |    1500k    | CelebA-val |  6.677   | 26.878 | 0.911 | [model](https://download.openmmlab.com/mmediting/inpainting/deepfillv1/deepfillv1_256x256_4x4_celeba_20200619-dd51a855.pth) \| [log](https://download.openmmlab.com/mmediting/inpainting/deepfillv1/deepfillv1_256x256_4x4_celeba_20200619-dd51a855.log.json) |
diff --git a/configs/inpainting/deepfillv2/README.md b/configs/inpainting/deepfillv2/README.md
@@ -4,7 +4,7 @@
 
 [ALGORITHM]
 
-```
+```latex
 @inproceedings{yu2019free,
   title={Free-form image inpainting with gated convolution},
   author={Yu, Jiahui and Lin, Zhe and Yang, Jimei and Shen, Xiaohui and Lu, Xin and Huang, Thomas S},
@@ -15,13 +15,15 @@
 ```
 
 ## Results and models
+
 ### Places365-Challenge
+
 |   Method   | Mask Type | Resolution | Train Iters |   Test Set    | l1 error |  PSNR  | SSIM  |                                                                                                                           Download                                                                                                                            |
 | :--------: | :-------: | :--------: | :---------: | :-----------: | :------: | :----: | :---: | :-----------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------: |
 | DeepFillv2 | free-form |  256x256   |    100k     | Places365-val |  8.635   | 22.398 | 0.815 | [model](https://download.openmmlab.com/mmediting/inpainting/deepfillv2/deepfillv2_256x256_8x2_places_20200619-10d15793.pth) \| [log](https://download.openmmlab.com/mmediting/inpainting/deepfillv2/deepfillv2_256x256_8x2_places_20200619-10d15793.log.json) |
 
-
 ### CelebA-HQ
+
 |   Method   | Mask Type | Resolution | Train Iters |  Test Set  | l1 error |  PSNR  | SSIM  |                                                                                                                           Download                                                                                                                            |
 | :--------: | :-------: | :--------: | :---------: | :--------: | :------: | :----: | :---: | :-----------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------: |
 | DeepFillv2 | free-form |  256x256   |     20k     | CelebA-val |  5.411   | 25.721 | 0.871 | [model](https://download.openmmlab.com/mmediting/inpainting/deepfillv2/deepfillv2_256x256_8x2_celeba_20200619-c96e5f12.pth) \| [log](https://download.openmmlab.com/mmediting/inpainting/deepfillv2/deepfillv2_256x256_8x2_celeba_20200619-c96e5f12.log.json) |
diff --git a/configs/inpainting/global_local/README.md b/configs/inpainting/global_local/README.md
@@ -4,7 +4,7 @@
 
 [ALGORITHM]
 
-```
+```latex
 @article{iizuka2017globally,
   title={Globally and locally consistent image completion},
   author={Iizuka, Satoshi and Simo-Serra, Edgar and Ishikawa, Hiroshi},
@@ -20,13 +20,15 @@
 *Note that we do not apply the post-processing module in Global&Local for a fair comparison with current deep inpainting methods.*
 
 ## Results and models
+
 ### Places365-Challenge
+
 |    Method    |  Mask Type  | Resolution | Train Iters |   Test Set    | l1 error |  PSNR  | SSIM  |                                                                                                                      Download                                                                                                                       |
 | :----------: | :---------: | :--------: | :---------: | :-----------: | :------: | :----: | :---: | :-------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------: |
 | Global&Local | square bbox |  256x256   |    500k     | Places365-val |  11.164  | 23.152 | 0.862 | [model](https://download.openmmlab.com/mmediting/inpainting/global_local/gl_256x256_8x12_places_20200619-52a040a8.pth) \| [log](https://download.openmmlab.com/mmediting/inpainting/global_local/gl_256x256_8x12_places_20200619-52a040a8.log.json) |
 
-
 ### CelebA-HQ
+
 |    Method    |  Mask Type  | Resolution | Train Iters |  Test Set  | l1 error |  PSNR  | SSIM  |                                                                                                                      Download                                                                                                                       |
 | :----------: | :---------: | :--------: | :---------: | :--------: | :------: | :----: | :---: | :-------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------: |
 | Global&Local | square bbox |  256x256   |    500k     | CelebA-val |  6.678   | 26.780 | 0.904 | [model](https://download.openmmlab.com/mmediting/inpainting/global_local/gl_256x256_8x12_celeba_20200619-5af0493f.pth) \| [log](https://download.openmmlab.com/mmediting/inpainting/global_local/gl_256x256_8x12_celeba_20200619-5af0493f.log.json) |
diff --git a/configs/inpainting/partial_conv/README.md b/configs/inpainting/partial_conv/README.md
@@ -4,7 +4,7 @@
 
 [ALGORITHM]
 
-```
+```latex
 @inproceedings{liu2018image,
   title={Image inpainting for irregular holes using partial convolutions},
   author={Liu, Guilin and Reda, Fitsum A and Shih, Kevin J and Wang, Ting-Chun and Tao, Andrew and Catanzaro, Bryan},
@@ -15,13 +15,15 @@
 ```
 
 ## Results and models
+
 ### Places365-Challenge
+
 | Method | Mask Type | Resolution | Train Iters |   Test Set    | l1 error |  PSNR  | SSIM  |                                                                                                                        Download                                                                                                                         |
 | :----: | :-------: | :--------: | :---------: | :-----------: | :------: | :----: | :---: | :-----------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------: |
 | PConv  | free-form |  256x256   |    500k     | Places365-val |  8.776   | 22.762 | 0.801 | [model](https://download.openmmlab.com/mmediting/inpainting/pconv/pconv_256x256_stage2_4x2_places_20200619-1ffed0e8.pth) \| [log](https://download.openmmlab.com/mmediting/inpainting/pconv/pconv_256x256_stage2_4x2_places_20200619-1ffed0e8.log.json) |
 
-
 ### CelebA-HQ
+
 | Method | Mask Type | Resolution | Train Iters |  Test Set  | l1 error |  PSNR  | SSIM  |                                                                                                                        Download                                                                                                                         |
 | :----: | :-------: | :--------: | :---------: | :--------: | :------: | :----: | :---: | :-----------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------: |
 | PConv  | free-form |  256x256   |    500k     | CelebA-val |  5.990   | 25.404 | 0.853 | [model](https://download.openmmlab.com/mmediting/inpainting/pconv/pconv_256x256_stage2_4x2_celeba_20200619-860f8b95.pth) \| [log](https://download.openmmlab.com/mmediting/inpainting/pconv/pconv_256x256_stage2_4x2_celeba_20200619-860f8b95.log.json) |
diff --git a/configs/mattors/dim/README.md b/configs/mattors/dim/README.md
@@ -4,7 +4,7 @@
 
 [ALGORITHM]
 
-```
+```latex
 @inproceedings{xu2017deep,
   title={Deep image matting},
   author={Xu, Ning and Price, Brian and Cohen, Scott and Huang, Thomas},

diff --git a/configs/mattors/gca/README.md b/configs/mattors/gca/README.md
@@ -4,7 +4,7 @@
 
 [ALGORITHM]
 
-```
+```latex
 @inproceedings{li2020natural,
   title={Natural Image Matting via Guided Contextual Attention},
   author={Li, Yaoyi and Lu, Hongtao},

diff --git a/configs/mattors/indexnet/README.md b/configs/mattors/indexnet/README.md
@@ -4,7 +4,7 @@
 
 [ALGORITHM]
 
-```
+```latex
 @inproceedings{hao2019indexnet,
   title={Indices Matter: Learning to Index for Deep Image Matting},
   author={Lu, Hao and Dai, Yutong and Shen, Chunhua and Xu, Songcen},

diff --git a/configs/restorers/edsr/README.md b/configs/restorers/edsr/README.md
@@ -4,7 +4,7 @@
 
 [ALGORITHM]
 
-```
+```latex
 @inproceedings{lim2017enhanced,
   title={Enhanced deep residual networks for single image super-resolution},
   author={Lim, Bee and Son, Sanghyun and Kim, Heewon and Nah, Seungjun and Mu Lee, Kyoung},

diff --git a/configs/restorers/edvr/README.md b/configs/restorers/edvr/README.md
@@ -4,7 +4,7 @@
 
 [ALGORITHM]
 
-```
+```latex
 @InProceedings{wang2019edvr,
   author    = {Wang, Xintao and Chan, Kelvin C.K. and Yu, Ke and Dong, Chao and Loy, Chen Change},
   title     = {EDVR: Video restoration with enhanced deformable convolutional networks},

diff --git a/configs/restorers/esrgan/README.md b/configs/restorers/esrgan/README.md
@@ -4,7 +4,7 @@
 
 [ALGORITHM]
 
-```
+```latex
 @inproceedings{wang2018esrgan,
   title={Esrgan: Enhanced super-resolution generative adversarial networks},
   author={Wang, Xintao and Yu, Ke and Wu, Shixiang and Gu, Jinjin and Liu, Yihao and Dong, Chao and Qiao, Yu and Change Loy, Chen},

diff --git a/configs/restorers/srcnn/README.md b/configs/restorers/srcnn/README.md
@@ -4,7 +4,7 @@
 
 [ALGORITHM]
 
-```
+```latex
 @article{dong2015image,
   title={Image super-resolution using deep convolutional networks},
   author={Dong, Chao and Loy, Chen Change and He, Kaiming and Tang, Xiaoou},

diff --git a/configs/restorers/srresnet_srgan/README.md b/configs/restorers/srresnet_srgan/README.md
@@ -4,7 +4,7 @@
 
 [ALGORITHM]
 
-```
+```latex
 @inproceedings{ledig2016photo,
   title={Photo-realistic single image super-resolution using a generative adversarial network},
   author={Ledig, Christian and Theis, Lucas and Husz{\'a}r, Ferenc and Caballero, Jose and Cunningham, Andrew and Acosta, Alejandro and Aitken, Andrew and Tejani, Alykhan and Totz, Johannes and Wang, Zehan},

diff --git a/configs/restorers/tof/README.md b/configs/restorers/tof/README.md
@@ -4,7 +4,7 @@
 
 [ALGORITHM]
 
-```
+```latex
 @article{xue2019video,
   title={Video enhancement with task-oriented flow},
   author={Xue, Tianfan and Chen, Baian and Wu, Jiajun and Wei, Donglai and Freeman, William T},

diff --git a/configs/synthesizers/cyclegan/README.md b/configs/synthesizers/cyclegan/README.md
@@ -4,7 +4,7 @@
 
 [ALGORITHM]
 
-```
+```latex
 @inproceedings{zhu2017unpaired,
   title={Unpaired image-to-image translation using cycle-consistent adversarial networks},
   author={Zhu, Jun-Yan and Park, Taesung and Isola, Phillip and Efros, Alexei A},

diff --git a/configs/synthesizers/pix2pix/README.md b/configs/synthesizers/pix2pix/README.md
@@ -4,7 +4,7 @@
 
 [ALGORITHM]
 
-```
+```latex
 @inproceedings{isola2017image,
   title={Image-to-image translation with conditional adversarial networks},
   author={Isola, Phillip and Zhu, Jun-Yan and Zhou, Tinghui and Efros, Alexei A},

diff --git a/docs/conf.py b/docs/conf.py
@@ -67,6 +67,7 @@
 
 
 def builder_inited_handler(app):
+    subprocess.run(['./merge_docs.sh'])
     subprocess.run(['./stat.py'])
 
 

diff --git a/docs/index.rst b/docs/index.rst
@@ -12,8 +12,11 @@ Welcome to MMEditing's documentation!
    :maxdepth: 2
    :caption: Model Zoo
 
-   model_zoo.md
-   modelzoo_statistics.md
+   modelzoo.md
+   inpainting_models.md
+   mattors_models.md
+   restorers_models.md
+   synthesizers_models.md
 
 .. toctree::
    :maxdepth: 2

diff --git a/docs/merge_docs.sh b/docs/merge_docs.sh
@@ -0,0 +1,36 @@
+#!/usr/bin/env bash
+
+sed -i '$a\\n' ../configs/inpainting/*/*.md
+sed -i '$a\\n' ../configs/mattors/*/*.md
+sed -i '$a\\n' ../configs/restorers/*/*.md
+sed -i '$a\\n' ../configs/synthesizers/*/*.md
+
+cat ../configs/inpainting/*/*.md >inpainting_models.md
+cat ../configs/mattors/*/*.md >mattors_models.md
+cat ../configs/restorers/*/*.md >restorers_models.md
+cat ../configs/synthesizers/*/*.md >synthesizers_models.md
+
+sed -i "s/#/#&/" inpainting_models.md
+sed -i "s/#/#&/" mattors_models.md
+sed -i "s/#/#&/" restorers_models.md
+sed -i "s/#/#&/" synthesizers_models.md
+
+sed -i "s/md###t/html#t/g" inpainting_models.md
+sed -i "s/md###t/html#t/g" mattors_models.md
+sed -i "s/md###t/html#t/g" restorers_models.md
+sed -i "s/md###t/html#t/g" synthesizers_models.md
+
+sed -i '1i\# Inpainting Models' inpainting_models.md
+sed -i '1i\# Matting Models' mattors_models.md
+sed -i '1i\# Restoration Models' restorers_models.md
+sed -i '1i\# Generation Models' synthesizers_models.md
+
+sed -i 's/](\/docs\//](/g' inpainting_models.md # remove /docs/ for link used in doc site
+sed -i 's/](\/docs\//](/g' mattors_models.md
+sed -i 's/](\/docs\//](/g' restorers_models.md
+sed -i 's/](\/docs\//](/g' synthesizers_models.md
+
+sed -i 's=](/=](https://github.com/open-mmlab/mmpose/tree/master/=g' inpainting_models.md
+sed -i 's=](/=](https://github.com/open-mmlab/mmpose/tree/master/=g' mattors_models.md
+sed -i 's=](/=](https://github.com/open-mmlab/mmpose/tree/master/=g' restorers_models.md
+sed -i 's=](/=](https://github.com/open-mmlab/mmpose/tree/master/=g' synthesizers_models.md
diff --git a/docs/model_zoo.md b/docs/model_zoo.md
Original file line number	Diff line number	Diff line change
Expand Up		@@ -67,6 +67,7 @@


		def builder_inited_handler(app):
		subprocess.run(['./merge_docs.sh'])
		subprocess.run(['./stat.py'])


Expand Down