add U-GAT-IT

2020-08-21 16:14:30 +08:00
parent 323bf2f6ab
commit 1a1cb9b00f
18 changed files with 815 additions and 55 deletions
--- a/configs/synthesizers/UGATIT.yml
+++ b/configs/synthesizers/UGATIT.yml
@@ -0,0 +1,110 @@
+name: selfie2anime
+engine: UGATIT
+result_dir: ./result
+max_iteration: 100000
+
+distributed:
+  model:
+    # broadcast_buffers: False
+
+misc:
+  random_seed: 324
+
+checkpoints:
+  interval: 1000
+
+model:
+  generator:
+    _type: UGATIT-Generator
+    in_channels: 3
+    out_channels: 3
+    base_channels: 64
+    num_blocks: 4
+    img_size: 256
+    light: True
+  local_discriminator:
+    _type: UGATIT-Discriminator
+    in_channels: 3
+    base_channels: 64
+    num_blocks: 3
+  global_discriminator:
+    _type: UGATIT-Discriminator
+    in_channels: 3
+    base_channels: 64
+    num_blocks: 5
+
+loss:
+  gan:
+    loss_type: lsgan
+    weight: 1.0
+    real_label_val: 1.0
+    fake_label_val: 0.0
+  cycle:
+    level: 1
+    weight: 10.0
+  id:
+    level: 1
+    weight: 10.0
+  cam:
+    weight: 1000
+
+optimizers:
+  generator:
+    _type: Adam
+    lr: 0.0001
+    betas: [0.5, 0.999]
+    weight_decay: 0.0001
+  discriminator:
+    _type: Adam
+    lr: 1e-4
+    betas: [0.5, 0.999]
+    weight_decay: 0.0001
+
+data:
+  train:
+    buffer_size: 50
+    dataloader:
+      batch_size: 8
+      shuffle: True
+      num_workers: 2
+      pin_memory: True
+      drop_last: True
+    dataset:
+      _type: GenerationUnpairedDataset
+      root_a: "/data/i2i/selfie2anime/trainA"
+      root_b: "/data/i2i/selfie2anime/trainB"
+      random_pair: True
+      pipeline:
+        - Load
+        - Resize:
+            size: [286, 286]
+        - RandomCrop:
+            size: [256, 256]
+        - RandomHorizontalFlip
+        - ToTensor
+        - Normalize:
+            mean: [0.5, 0.5, 0.5]
+            std: [0.5, 0.5, 0.5]
+    scheduler:
+      start: 50000
+      target_lr: 0
+  test:
+    dataloader:
+      batch_size: 4
+      shuffle: False
+      num_workers: 1
+      pin_memory: False
+      drop_last: False
+    dataset:
+      _type: GenerationUnpairedDataset
+      root_a: "/data/i2i/selfie2anime/testA"
+      root_b: "/data/i2i/selfie2anime/testB"
+      random_pair: False
+      pipeline:
+        - Load
+        - Resize:
+            size: [256, 256]
+        - ToTensor
+        - Normalize:
+            mean: [0.5, 0.5, 0.5]
+            std: [0.5, 0.5, 0.5]