Add StackGAN++-Mode-Seeking

Author: HelenMao

StackGAN++-Mode-Seeking/cfg/birds_3stages.yml

@@ -0,0 +1,37 @@
+CONFIG_NAME: '3stages'
+
+DATASET_NAME: 'birds'
+EMBEDDING_TYPE: 'cnn-rnn'
+DATA_DIR: 'E:\\GAN-Datasets\\birds'
+GPU_ID: '0'
+WORKERS: 0
+
+
+TREE:
+    BRANCH_NUM: 3
+
+
+TRAIN:
+    FLAG: True
+    NET_G: ''  # './output/birds_3stages/Model/netG_epoch_700.pth'
+    NET_D: ''  # './output/birds_3stages/Model/netD'
+    BATCH_SIZE: 4
+    MAX_EPOCH: 1000
+    SNAPSHOT_INTERVAL: 2000
+    DISCRIMINATOR_LR: 0.0002
+    GENERATOR_LR: 0.0002
+    COEFF:
+      UNCOND_LOSS: 1.0
+      MS_LOSS: 1.0
+
+
+GAN:
+    EMBEDDING_DIM: 128
+    DF_DIM: 64
+    GF_DIM: 64
+    Z_DIM: 100
+    R_NUM: 2
+    B_CONDITION: True
+
+TEXT:
+    DIMENSION: 1024

StackGAN++-Mode-Seeking/cfg/eval_birds.yml

@@ -0,0 +1,29 @@
+CONFIG_NAME: '3stages'
+
+DATASET_NAME: 'birds'
+EMBEDDING_TYPE: 'cnn-rnn'
+DATA_DIR: 'E:\\GAN-Datasets\\birds'
+GPU_ID: '0'
+WORKERS: 1  # 4
+
+TREE:
+    BRANCH_NUM: 3
+
+
+TEST:
+    B_EXAMPLE: False
+
+
+TRAIN:
+    FLAG: False
+    NET_G: 'J:\\qimao\\Text-to-image\\StackGAN-v2\\official\\netG_210000.pth'
+    BATCH_SIZE: 1
+
+
+GAN:
+    EMBEDDING_DIM: 128
+    DF_DIM: 64
+    GF_DIM: 64
+    Z_DIM: 100
+    R_NUM: 2
+    B_CONDITION: True