hyp.yaml

augmentations:
  mean:
    - 104
    - 117
    - 123
    
cfg:
  max_iter: 120000
  feature_maps:
    - 38
    - 19
    - 10
    - 5
    - 3
    - 1
  steps:
    - 8
    - 16
    - 32
    - 64
    - 100
    - 300
  min_sizes:
    - 30
    - 60
    - 111
    - 162
    - 213
    - 264
  max_sizes:
    - 60
    - 111
    - 162
    - 213
    - 264
    - 315
  aspect_ratios:
    - [2]
    - [2, 3]
    - [2, 3]
    - [2, 3]
    - [2]
    - [2]
  variance:
    - 0.1
    - 0.2
  clip: true

detect:
  bkg_label: 0
  top_k: 200 
  conf_thresh: 0.01
  nms_thresh: 0.45

opt: # SGD
  lr: 1e-3 # Initial learning rate
  momentum: 0.9 # Momentum value for optim
  weight_decay: 5e-4 # Weight decay for SGD
  gamma: 0.1 # Gamma update for SGD
  lr_steps:
    - 80000
    - 100000
    - 120000