[Feature] Support BigGAN and BigGAN-deep (#45)

* Add generator for BigGAN * Add discriminator's architecture of bigGAN * pylint fix * refine the implementation * refine unittest * complete docs and unittest
open-mmlab · Jun 16, 2021 · 548112e · 548112e
1 parent e8b7913
commit 548112e
Show file tree

Hide file tree

Showing 5 changed files with 1,299 additions and 1 deletion.
diff --git a/mmgen/models/architectures/__init__.py b/mmgen/models/architectures/__init__.py
@@ -1,3 +1,4 @@
+from .biggan import BigGANGenerator
 from .cyclegan import ResnetGenerator
 from .dcgan import DCGANDiscriminator, DCGANGenerator
 from .fid_inception import InceptionV3
@@ -28,5 +29,5 @@
     'MSStyleGAN2Discriminator', 'MSStyleGANv2Generator',
     'generation_init_weights', 'PatchDiscriminator', 'ResnetGenerator',
     'PerceptualLoss', 'WGANGPDiscriminator', 'WGANGPGenerator',
-    'LSGANDiscriminator', 'LSGANGenerator'
+    'LSGANDiscriminator', 'LSGANGenerator', 'BigGANGenerator'
 ]
diff --git a/mmgen/models/architectures/biggan/__init__.py b/mmgen/models/architectures/biggan/__init__.py
@@ -0,0 +1,7 @@
+from .generator_discriminator import BigGANDiscriminator, BigGANGenerator
+from .modules import BigGANConditionBN, BigGANGenResBlock
+
+__all__ = [
+    'BigGANGenerator', 'BigGANGenResBlock', 'BigGANConditionBN',
+    'BigGANDiscriminator'
+]