config.yaml

# paths
paths:
  train: './data'
  valid: './data'
  checkpoint: 'data/models'
  logs: 'data/logs'

# lookups
lookups:
  train: 'train.txt'
  valid: 'valid.txt'
  debug: 'debug_list.txt'

# shapes
shapes:
  image:
    - 192
    - 192
    - 3
  mask:
    - 192
    - 192
    - 1
  image_generator:
    - 10
    - 192
    - 192
    - 3
  mask_generator:
    - 10
    - 192
    - 192
    - 1
  debug_image_generator:
    - 2
    - 192
    - 192
    - 3
  debug_mask_generator:
    - 2
    - 192
    - 192
    - 1

# batch params
batch:
  size: 10
  debug_size: 2

# loss params
loss:
  alpha: 0.7
  beta: 0.3
  gamma: 0.75
  smooth: 1e-6

# train params
training:
  lr: 0.001
  epochs: 50
  debug_epochs: 2
  augmentations:
    - 'flip'
    - 'rotate'
  augment_proba: 0.4

# model params
model:
  name: 'MobileUNet'
  train_encoder: True
  weights: 'cityscapes'
  weight_decay: True
  mode: 'binary'
  n_classes: 1