pix2pix_facades.yaml 1.8 KB
Newer Older
L
LielinJiang 已提交
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24
epochs: 200
output_dir: output_dir
lambda_L1: 100

model:
  name: Pix2PixModel
  generator:
    name: UnetGenerator
    norm_type: batch
    input_nc: 3
    output_nc: 3
    num_downs: 8 #unet256
    ngf: 64
    use_dropout: False
  discriminator:
    name: NLayerDiscriminator
    ndf: 64
    n_layers: 3
    input_nc: 6
    norm_type: batch
  gan_mode: vanilla

dataset:
  train:
25 26
    name: PairedDataset
    dataroot: data/facades/
L
LielinJiang 已提交
27 28
    num_workers: 0
    batch_size: 1
L
LielinJiang 已提交
29 30 31 32 33 34 35
    phase: train
    max_dataset_size: inf
    direction: BtoA
    input_nc: 3
    output_nc: 3
    serial_batches: False
    pool_size: 0
L
LielinJiang 已提交
36 37 38
    transforms:
      - name: Resize
        size: [286, 286]
L
LielinJiang 已提交
39
        interpolation: 'bicubic' #cv2.INTER_CUBIC
L
LielinJiang 已提交
40 41
        keys: [image, image]
      - name: PairedRandomCrop
L
LielinJiang 已提交
42
        size: [256, 256]
L
LielinJiang 已提交
43 44 45 46
        keys: [image, image]
      - name: PairedRandomHorizontalFlip
        prob: 0.5
        keys: [image, image]
L
LielinJiang 已提交
47
      - name: Transpose
L
LielinJiang 已提交
48 49 50 51 52
        keys: [image, image]
      - name: Normalize
        mean: [127.5, 127.5, 127.5]
        std: [127.5, 127.5, 127.5]
        keys: [image, image]
L
LielinJiang 已提交
53
  test:
54 55
    name: PairedDataset
    dataroot: data/facades/
L
LielinJiang 已提交
56 57 58 59 60 61 62
    phase: test
    max_dataset_size: inf
    direction: BtoA
    input_nc: 3
    output_nc: 3
    serial_batches: True
    pool_size: 50
L
LielinJiang 已提交
63 64 65
    transforms:
      - name: Resize
        size: [256, 256]
L
LielinJiang 已提交
66
        interpolation: 'bicubic' #cv2.INTER_CUBIC
L
LielinJiang 已提交
67
        keys: [image, image]
L
LielinJiang 已提交
68
      - name: Transpose
L
LielinJiang 已提交
69 70 71 72 73
        keys: [image, image]
      - name: Normalize
        mean: [127.5, 127.5, 127.5]
        std: [127.5, 127.5, 127.5]
        keys: [image, image]
L
LielinJiang 已提交
74 75 76 77

optimizer:
  name: Adam
  beta1: 0.5
L
LielinJiang 已提交
78 79 80 81 82 83

lr_scheduler:
  name: linear
  learning_rate: 0.0002
  start_epoch: 100
  decay_epochs: 100
L
LielinJiang 已提交
84 85 86 87 88 89 90

log_config:
  interval: 100
  visiual_interval: 500

snapshot_config:
  interval: 5