Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Questions about the size of discriminator network #132

Open
June01 opened this issue Mar 4, 2022 · 1 comment
Open

Questions about the size of discriminator network #132

June01 opened this issue Mar 4, 2022 · 1 comment

Comments

@June01
Copy link

June01 commented Mar 4, 2022

Hi thanks for the excellent work! I hope everything is running well with you guys.

While I am training this network on GOPRO, I found out the size of fake image is [1, 3, 256, 256], which was supposed to be a valid image, right? So I guess the output of a full discriminator should be a value (i.e. [1,1,1]) to indicate if it is real or not. However, I found out the output of full discriminator is [1, 1, 11, 11]. So could you please tell me why it happens? Any quick response is highly appreciated. All the best.

@June01
Copy link
Author

June01 commented Mar 4, 2022

Btw, here is my config.yaml for reference.

---
project: deblur_gan
experiment_desc: fpn

train:
  files_a: &FILES_A /path/to/GOPRO/train/*/blur/*.png
  files_b: &FILES_B /path/to/GOPRO/train/*/sharp/*.png
  size: &SIZE 256
  crop: none #random
  preload: &PRELOAD false
  preload_size: &PRELOAD_SIZE 0
  bounds: [0, .9]
  scope: geometric
  corrupt: &CORRUPT
    - name: cutout
      prob: 0.5
      num_holes: 3
      max_h_size: 25
      max_w_size: 25
    - name: jpeg
      quality_lower: 70
      quality_upper: 90
    - name: motion_blur
    - name: median_blur
    - name: gamma
    - name: rgb_shift
    - name: hsv_shift
    - name: sharpen

val:
  files_a: &FILES_C /path/to/GOPRO/test/*/blur/*.png
  files_b: &FILES_D /path/to/GOPRO/test/*/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
  # g_name: InceptionResNet-v2
  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
  disc_loss: gan
  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: [256, 256]

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

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

1 participant