random_seed: 42
yolo_config {
  big_anchor_shape: "[(116,90), (156,198), (373,326)]"
  mid_anchor_shape: "[(30,61), (62,45), (59,119)]"
  small_anchor_shape: "[(10,13), (16,30), (33,23)]"
  matching_neutral_box_iou: 0.5
  arch: "mobilenet_v1"
  loss_loc_weight: 0.75
  loss_neg_obj_weights: 200.0
  loss_class_weights: 1.0  
  freeze_bn: false
  freeze_blocks: 0
}
training_config {
  batch_size_per_gpu: 64
  num_epochs: 100
  enable_qat: true
  learning_rate {
  soft_start_annealing_schedule {
    min_learning_rate: 1e-7
    max_learning_rate: 1e-4
    soft_start: 0.15
    annealing: 0.5
    }
  }
  regularizer {
    type: L2
    weight: 3e-5
  }
}
eval_config {
  validation_period_during_training: 10
  average_precision_mode: SAMPLE
  matching_iou_threshold: 0.5
  batch_size: 64
}
nms_config {
  confidence_threshold: 0.01
  clustering_iou_threshold: 0.6
  top_k: 200
}
augmentation_config {
  preprocessing {
    output_image_width: 1280
    output_image_height: 960
    output_image_channel: 3
    crop_right:1280
    crop_bottom:960
    min_bbox_width: 1.0
    min_bbox_height: 1.0
  }
  spatial_augmentation {
    hflip_probability: 0.5
    vflip_probability: 0.5
    zoom_min: 0.9
    zoom_max: 1.1
    
  }
  color_augmentation {
    hue_rotation_max: 25.0
    saturation_shift_max: 0.20000000298
    contrast_scale_max: 0.10000000149
    contrast_center: 0.5
  }
}
dataset_config {
  data_sources: {
    tfrecords_path: "/workspace/tlt-experiments/train/tfrecord/*"
    image_directory_path: "/workspace/tlt-experiments"
  }
  image_extension: "png"
  target_class_mapping {
      key: "person"
      value: "person"
  }
  validation_fold: 0
}