pix2pix_cityscapes.yaml 1.2 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 25
epochs: 200
isTrain: True
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:
L
LielinJiang 已提交
26
    name: PairedDataset
L
LielinJiang 已提交
27
    dataroot: data/cityscapes
28
    num_workers: 4
L
LielinJiang 已提交
29 30 31 32 33 34 35 36 37 38 39 40 41
    phase: train
    max_dataset_size: inf
    direction: BtoA
    input_nc: 3
    output_nc: 3
    serial_batches: False
    pool_size: 0
    transform:
      load_size: 286
      crop_size: 256
      preprocess: resize_and_crop
      no_flip: False
  test:
L
LielinJiang 已提交
42
    name: PairedDataset
L
LielinJiang 已提交
43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72
    dataroot: data/cityscapes/
    phase: test
    max_dataset_size: inf
    direction: BtoA
    input_nc: 3
    output_nc: 3
    serial_batches: True
    pool_size: 50
    transform:
      load_size: 256
      crop_size: 256
      preprocess: resize_and_crop
      no_flip: True

optimizer:
  name: Adam
  beta1: 0.5
  lr_scheduler:
    name: linear
    learning_rate: 0.0002
    start_epoch: 100
    decay_epochs: 100

log_config:
  interval: 100
  visiual_interval: 500

snapshot_config:
  interval: 5