solve conflict with bigGAN

.gitignore

@@ -66,7 +66,7 @@ instance/
 
 # Sphinx documentation
 docs/_build/
-
+docs_zh-CN/_build/
 # PyBuilder
 target/
 
@@ -116,3 +116,6 @@ work_dirs/
 *.DS_Store
 # PyTorch
 *.pth
+mmgen/configs/
+mmgen/tools/
+runs/

MANIFEST.in

@@ -0,0 +1,3 @@
+include mmgen/model_zoo.yml
+recursive-include mmgen/configs *.py *.yml
+recursive-include mmgen/tools *.sh *.py

configs/dcgan/metafile.yml

@@ -0,0 +1,23 @@
+Collections:
+  - Name: dcgan
+    Metadata:
+      Training Data: CelebA-64x64
+      Training Techniques:
+        - Adam
+      Training Resources: 1x TITAN X
+      Architecture:
+        - DCGAN
+    Paper: https://arxiv.org/abs/1511.06434
+    README: configs/dcgan/README.md
+
+Models:
+  - Name: dcgan_celeba_64x64
+    In Collection: dcgan
+    Config: configs/dcgan/dcgan_celeba-cropped_64_b128x1_300k.py
+    Results:
+      - Task: Unconditional Image Generation
+        Dataset: CelebA-64x64
+        Metrics:
+          SWD: 8.93,10.53,50.32/23.26
+          MS-SSIM: 0.2899
+    Weights: https://download.openmmlab.com/mmgen/dcgan/dcgan_mnist-64_b128x1_Glr4e-4_Dlr1e-4_5k_20210512_163926-207a1eaf.pth?versionId=CAEQKxiBgMDTnYH_yhciIGY2MjAwMzBjZjZiZjQzODJhZWRhOTFiNjU4MzdhMWY2

docs/index.rst

@@ -27,6 +27,11 @@ Welcome to MMGeneration's documentation!
    changelog.md
    faq.md
 
+.. toctree::
+   :caption: Switch Language
+
+   switch_language.md
+
 .. toctree::
    :caption: API Reference
 

docs/modelzoo_statistics.md

@@ -1,15 +1,21 @@
 
 # Model Zoo Statistics
 
-* Number of papers: 9
-* Number of checkpoints: 51
+* Number of papers: 11
+* Number of checkpoints: 62
 
 	* [CycleGAN: Unpaired Image-to-Image Translation Using Cycle-Consistent Adversarial Networks](https://github.com/open-mmlab/mmgeneration/blob/master/configs/cyclegan) (6 ckpts)
 
 
 	* [Unsupervised Representation Learning with Deep Convolutional Generative Adversarial Networks](https://github.com/open-mmlab/mmgeneration/blob/master/configs/dcgan) (3 ckpts)
 
 
+	* [Geometric GAN](https://github.com/open-mmlab/mmgeneration/blob/master/configs/ggan) (3 ckpts)
+
+
+	* [Least Squares Generative Adversarial Networks](https://github.com/open-mmlab/mmgeneration/blob/master/configs/lsgan) (4 ckpts)
+
+
 	* [Progressive Growing of GANs for Improved Quality, Stability, and Variation](https://github.com/open-mmlab/mmgeneration/blob/master/configs/pggan) (3 ckpts)
 
 
@@ -25,7 +31,7 @@
 	* [A Style-Based Generator Architecture for Generative Adversarial Networks (CVPR'2019)](https://github.com/open-mmlab/mmgeneration/blob/master/configs/styleganv1) (2 ckpts)
 
 
-	* [Analyzing and Improving the Image Quality of Stylegan (CVPR'2020)](https://github.com/open-mmlab/mmgeneration/blob/master/configs/styleganv2) (7 ckpts)
+	* [Analyzing and Improving the Image Quality of Stylegan (CVPR'2020)](https://github.com/open-mmlab/mmgeneration/blob/master/configs/styleganv2) (11 ckpts)
 
 
 	* [Improved Training of Wasserstein GANs](https://github.com/open-mmlab/mmgeneration/blob/master/configs/wgan-gp) (2 ckpts)

docs/switch_language.md

@@ -0,0 +1,3 @@
+## <a href='https://mmgeneration.readthedocs.io/en/latest/'>English</a>
+
+## <a href='https://mmgeneration.readthedocs.io/zh_CN/latest/'>简体中文</a>

docs_zh-CN/Makefile

@@ -0,0 +1,20 @@
+# Minimal makefile for Sphinx documentation
+#
+
+# You can set these variables from the command line, and also
+# from the environment for the first two.
+SPHINXOPTS    ?=
+SPHINXBUILD   ?= sphinx-build
+SOURCEDIR     = .
+BUILDDIR      = _build
+
+# Put it first so that "make" without argument is like "make help".
+help:
+	@$(SPHINXBUILD) -M help "$(SOURCEDIR)" "$(BUILDDIR)" $(SPHINXOPTS) $(O)
+
+.PHONY: help Makefile
+
+# Catch-all target: route all unknown targets to Sphinx using the new
+# "make mode" option.  $(O) is meant as a shortcut for $(SPHINXOPTS).
+%: Makefile
+	@$(SPHINXBUILD) -M $@ "$(SOURCEDIR)" "$(BUILDDIR)" $(SPHINXOPTS) $(O)

docs_zh-CN/api.rst

@@ -0,0 +1,86 @@
+接口参考手册
+=================
+
+mmgen.apis
+--------------
+.. automodule:: mmgen.apis
+    :members:
+
+mmgen.core
+--------------
+
+evaluation
+^^^^^^^^^^
+.. automodule:: mmgen.core.evaluation
+    :members:
+
+hooks
+^^^^^^^^^^
+.. automodule:: mmgen.core.hooks
+    :members:
+
+optimizer
+^^^^^^^^^^
+.. automodule:: mmgen.core.optimizer
+    :members:
+
+runners
+^^^^^^^^^^
+.. automodule:: mmgen.core.runners
+    :members:
+
+scheduler
+^^^^^^^^^^
+.. automodule:: mmgen.core.scheduler
+    :members:
+
+utils
+^^^^^^^^^^
+.. automodule:: mmgen.core.utils
+    :members:
+
+mmgen.datasets
+--------------
+
+datasets
+^^^^^^^^^^
+.. automodule:: mmgen.datasets
+    :members:
+
+pipelines
+^^^^^^^^^^
+.. automodule:: mmgen.datasets.pipelines
+    :members:
+
+mmgen.models
+--------------
+
+architectures
+^^^^^^^^^^
+.. automodule:: mmgen.models.architectures
+    :members:
+
+common
+^^^^^^^^^^
+.. automodule:: mmgen.models.common
+    :members:
+
+gans
+^^^^^^^^^^^^
+.. automodule:: mmgen.models.gans
+    :members:
+
+losses
+^^^^^^^^^^^^
+.. automodule:: mmgen.models.losses
+    :members:
+
+ops
+^^^^^^^^^^
+.. automodule:: mmgen.models.ops
+    :members:
+
+utils
+^^^^^^^^^^
+.. automodule:: mmgen.models.utils
+    :members:

docs_zh-CN/changelog.md

@@ -0,0 +1 @@
+# 版本更新日志

docs_zh-CN/conf copy.py

@@ -0,0 +1,90 @@
+# Configuration file for the Sphinx documentation builder.
+#
+# This file only contains a selection of the most common options. For a full
+# list see the documentation:
+# https://www.sphinx-doc.org/en/master/usage/configuration.html
+
+# -- Path setup --------------------------------------------------------------
+
+# If extensions (or modules to document with autodoc) are in another directory,
+# add these directories to sys.path here. If the directory is relative to the
+# documentation root, use os.path.abspath to make it absolute, like shown here.
+#
+import os
+import subprocess
+import sys
+
+sys.path.insert(0, os.path.abspath('..'))
+
+# -- Project information -----------------------------------------------------
+
+project = 'MMGeneration'
+copyright = '2018-2020, OpenMMLab'
+author = 'MMGeneration Authors'
+version_file = '../mmgen/version.py'
+
+
+def get_version():
+    with open(version_file, 'r') as f:
+        exec(compile(f.read(), version_file, 'exec'))
+    return locals()['__version__']
+
+
+# The full version, including alpha/beta/rc tags
+release = get_version()
+
+# -- General configuration ---------------------------------------------------
+
+# Add any Sphinx extension module names here, as strings. They can be
+# extensions coming with Sphinx (named 'sphinx.ext.*') or your custom
+# ones.
+extensions = [
+    'sphinx.ext.autodoc',
+    'sphinx.ext.napoleon',
+    'sphinx.ext.viewcode',
+    'recommonmark',
+    'sphinx_markdown_tables',
+]
+
+autodoc_mock_imports = [
+    'matplotlib', 'pycocotools', 'terminaltables', 'mmgen.version', 'mmcv.ops'
+]
+
+# Add any paths that contain templates here, relative to this directory.
+templates_path = ['_templates']
+
+# The suffix(es) of source filenames.
+# You can specify multiple suffix as a list of string:
+#
+source_suffix = {
+    '.rst': 'restructuredtext',
+    '.md': 'markdown',
+}
+
+# The master toctree document.
+master_doc = 'index'
+
+# List of patterns, relative to source directory, that match files and
+# directories to ignore when looking for source files.
+# This pattern also affects html_static_path and html_extra_path.
+exclude_patterns = ['_build', 'Thumbs.db', '.DS_Store']
+
+# -- Options for HTML output -------------------------------------------------
+
+# The theme to use for HTML and HTML Help pages.  See the documentation for
+# a list of builtin themes.
+#
+html_theme = 'sphinx_rtd_theme'
+
+# Add any paths that contain custom static files (such as style sheets) here,
+# relative to this directory. They are copied after the builtin static files,
+# so a file named "default.css" will overwrite the builtin "default.css".
+html_static_path = ['_static']
+
+
+def builder_inited_handler(app):
+    subprocess.run(['./stat.py'])
+
+
+def setup(app):
+    app.connect('builder-inited', builder_inited_handler)

docs_zh-CN/conf.py

@@ -0,0 +1,90 @@
+# Configuration file for the Sphinx documentation builder.
+#
+# This file only contains a selection of the most common options. For a full
+# list see the documentation:
+# https://www.sphinx-doc.org/en/master/usage/configuration.html
+
+# -- Path setup --------------------------------------------------------------
+
+# If extensions (or modules to document with autodoc) are in another directory,
+# add these directories to sys.path here. If the directory is relative to the
+# documentation root, use os.path.abspath to make it absolute, like shown here.
+#
+import os
+import subprocess
+import sys
+
+sys.path.insert(0, os.path.abspath('..'))
+
+# -- Project information -----------------------------------------------------
+
+project = 'MMGeneration'
+copyright = '2018-2020, OpenMMLab'
+author = 'MMGeneration Authors'
+version_file = '../mmgen/version.py'
+
+
+def get_version():
+    with open(version_file, 'r') as f:
+        exec(compile(f.read(), version_file, 'exec'))
+    return locals()['__version__']
+
+
+# The full version, including alpha/beta/rc tags
+release = get_version()
+
+# -- General configuration ---------------------------------------------------
+
+# Add any Sphinx extension module names here, as strings. They can be
+# extensions coming with Sphinx (named 'sphinx.ext.*') or your custom
+# ones.
+extensions = [
+    'sphinx.ext.autodoc',
+    'sphinx.ext.napoleon',
+    'sphinx.ext.viewcode',
+    'recommonmark',
+    'sphinx_markdown_tables',
+]
+
+autodoc_mock_imports = [
+    'matplotlib', 'pycocotools', 'terminaltables', 'mmgen.version', 'mmcv.ops'
+]
+
+# Add any paths that contain templates here, relative to this directory.
+templates_path = ['_templates']
+
+# The suffix(es) of source filenames.
+# You can specify multiple suffix as a list of string:
+#
+source_suffix = {
+    '.rst': 'restructuredtext',
+    '.md': 'markdown',
+}
+
+# The master toctree document.
+master_doc = 'index'
+
+# List of patterns, relative to source directory, that match files and
+# directories to ignore when looking for source files.
+# This pattern also affects html_static_path and html_extra_path.
+exclude_patterns = ['_build', 'Thumbs.db', '.DS_Store']
+
+# -- Options for HTML output -------------------------------------------------
+
+# The theme to use for HTML and HTML Help pages.  See the documentation for
+# a list of builtin themes.
+#
+html_theme = 'sphinx_rtd_theme'
+
+# Add any paths that contain custom static files (such as style sheets) here,
+# relative to this directory. They are copied after the builtin static files,
+# so a file named "default.css" will overwrite the builtin "default.css".
+html_static_path = ['_static']
+
+
+def builder_inited_handler(app):
+    subprocess.run(['./stat.py'])
+
+
+def setup(app):
+    app.connect('builder-inited', builder_inited_handler)

docs_zh-CN/faq.md

@@ -0,0 +1 @@
+# 常见问题解答

docs_zh-CN/get_started.md

@@ -0,0 +1,8 @@
+## 依赖项
+
+- Linux
+- Python 3.6+
+- PyTorch 1.5+
+- CUDA 9.2+ (If you build PyTorch from source, CUDA 9.0 is also compatible)
+- GCC 5+
+- [MMCV (MMCV-FULL)](https://mmcv.readthedocs.io/en/latest/#installation)