VITA-Group / DeblurGANv2

[ICCV 2019] "DeblurGAN-v2: Deblurring (Orders-of-Magnitude) Faster and Better" by Orest Kupyn, Tetiana Martyniuk, Junru Wu, Zhangyang Wang
Other
1.03k stars 265 forks source link

dataset config.yaml #103

Closed WhXl closed 3 years ago

WhXl commented 3 years ago

I get this question!!! yaml.parser.ParserError: while parsing a block mapping in "config/config.yaml", line 6, column 3 expected , but found '' in "config/config.yaml", line 7, column 21

config.yaml

project: deblur_gan experiment_desc: fpn

train: files_a: &FILES_A ./deblurganv3/train/blur/.png files_b: FILES_A ./deblurganv3/train/sharp/*.png size: &SIZE 320 crop: random preload: &PRELOAD false preload_size: &PRELOAD_SIZE 0 bounds: [0, .9] scope: geometric corrupt: &CORRUPT

val: files_a: &FILES_A ./deblurganv3/val/blur/.png files_b: FILES_A .deblurganv3/val/sharp/.png size: SIZE scope: geometric crop: center preload: PRELOAD preload_size: PRELOAD_SIZE bounds: [.9, 1] corrupt: *CORRUPT

phase: train warmup_num: 3 model: g_name: fpn_inception blocks: 9 d_name: double_gan # may be no_gan, patch_gan, double_gan, multi_scale d_layers: 3 content_loss: perceptual adv_lambda: 0.001 disc_loss: wgan-gp learn_residual: True norm_layer: instance dropout: True

num_epochs: 200 train_batches_per_epoch: 1000 val_batches_per_epoch: 100 batch_size: 1 image_size: [320, 320]

optimizer: name: adam lr: 0.0001 scheduler: name: linear start_epoch: 50 min_lr: 0.0000001

dataset tree deblurganv3 ├── train │   ├── blur │   │   ├── 0_0.png │   │   ├── 0_1.png │   │   ├── 0_2.png │   │   ├── 0_3.png │   │   ├── 0_4.png │   │   ├── 0_5.png │   │   ├── 0_6.png │   │   ├── 0_7.png │   │   ├── 0_8.png │   │   └── 0_9.png │   └── sharp │   ├── 0_0.png │   ├── 0_1.png │   ├── 0_2.png │   ├── 0_3.png │   ├── 0_4.png │   ├── 0_5.png │   ├── 0_6.png │   ├── 0_7.png │   ├── 0_8.png │   └── 0_9.png └── val ├── blur │   ├── 0_0.png │   ├── 0_1.png │   ├── 0_2.png │   ├── 0_3.png │   ├── 0_4.png │   ├── 0_5.png │   ├── 0_6.png │   ├── 0_7.png │   ├── 0_8.png │   └── 0_9.png └── sharp ├── 0_0.png ├── 0_1.png ├── 0_2.png ├── 0_3.png ├── 0_4.png ├── 0_5.png ├── 0_6.png ├── 0_7.png ├── 0_8.png └── 0_9.png

WhXl commented 3 years ago

I fix this question! config.yaml

project: deblur_gan experiment_desc: fpn

train: files_a: &FILES_A /home/wx/dataset/DeblurGan激烈湍流/deblurganv2/train/blur/.png files_b: &FILES_B /home/wx/dataset/DeblurGan激烈湍流/deblurganv2/train/sharp/.png size: &SIZE 320 crop: random preload: &PRELOAD false preload_size: &PRELOAD_SIZE 0 bounds: [0, .9] scope: geometric corrupt: &CORRUPT

val: files_a: &FILES_C /home/wx/dataset/DeblurGan激烈湍流/deblurganv2/val/blur/.png files_b: &FILES_D /home/wx/dataset/DeblurGan激烈湍流/deblurganv2/val/sharp/.png size: SIZE scope: geometric crop: center preload: PRELOAD preload_size: PRELOAD_SIZE bounds: [.9, 1] corrupt: CORRUPT

phase: train warmup_num: 3 model: g_name: fpn_inception blocks: 9 d_name: double_gan # may be no_gan, patch_gan, double_gan, multi_scale d_layers: 3 content_loss: perceptual adv_lambda: 0.001 disc_loss: wgan-gp learn_residual: True norm_layer: instance dropout: True

num_epochs: 200 train_batches_per_epoch: 1000 val_batches_per_epoch: 100 batch_size: 1 image_size: [320, 320]

optimizer: name: adam lr: 0.0001 scheduler: name: linear start_epoch: 50 min_lr: 0.0000001