ssd_config {
  aspect_ratios_global: "[1.0, 2.0, 0.5, 3.0, 1.0/3.0]"
  scales: "[0.05, 0.1, 0.25, 0.4, 0.55, 0.7, 0.85]"
  two_boxes_for_ar1: true
  clip_boxes: false
  loss_loc_weight: 0.8
  focal_loss_alpha: 0.25
  focal_loss_gamma: 2.0
  variances: "[0.1, 0.1, 0.2, 0.2]"
  arch: "resnet18"
  freeze_bn: false
  freeze_blocks: 0
}
training_config {
  batch_size_per_gpu: 8
  num_epochs: 50
  learning_rate {
  soft_start_annealing_schedule {
    min_learning_rate: 5e-5
    max_learning_rate: 2e-2
    soft_start: 0.1
    annealing: 0.3
    }
  }
  regularizer {
    type: L1
    weight: 3.00000002618e-09
  }
}
eval_config {
  validation_period_during_training: 10
  average_precision_mode: SAMPLE
  batch_size: 16
  matching_iou_threshold: 0.5
}
nms_config {
  confidence_threshold: 0.01
  clustering_iou_threshold: 0.6
  top_k: 200
} 
augmentation_config {
  preprocessing {
    output_image_width: 800
    output_image_height: 576
    output_image_channel: 3
    crop_right: 800
    crop_bottom: 576
    min_bbox_width: 8.0
    min_bbox_height: 8.0
  }
  spatial_augmentation {
    hflip_probability: 0.5
    vflip_probability: 0.0
    zoom_min: 0.7
    zoom_max: 1.8
    translate_max_x: 8.0
    translate_max_y: 8.0
  }
  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/tfrecord/*"
    image_directory_path: "/workspace/tlt-experiments/kitti"
  }
  image_extension: "jpg"
  target_class_mapping {
    key: "vehicle0"
    value: "vehicle0"
  }
  target_class_mapping {
    key: "vehicle1"
    value: "vehicle1"
  }
  target_class_mapping {
    key: "vehicle2"
    value: "vehicle2"
  }
  target_class_mapping {
    key: "vehicle3"
    value: "vehicle3"
  }
  target_class_mapping {
    key: "vehicle4"
    value: "vehicle4"
  }
  target_class_mapping {
    key: "vehicle5"
    value: "vehicle5"
  }
  target_class_mapping {
    key: "vehicle6"
    value: "vehicle6"
  }
  target_class_mapping {
    key: "vehicle7"
    value: "vehicle7"
  }
  target_class_mapping {
    key: "vehicle8"
    value: "vehicle8"
  }
  target_class_mapping {
    key: "vehicle9"
    value: "vehicle9"
  }
  target_class_mapping {
    key: "vehicle10"
    value: "vehicle10"
  }
  target_class_mapping {
    key: "vehicle11"
    value: "vehicle11"
  }
  target_class_mapping {
    key: "vehicle12"
    value: "vehicle12"
  }
  target_class_mapping {
    key: "vehicle13"
    value: "vehicle13"
  }
  target_class_mapping {
    key: "vehicle14"
    value: "vehicle4"
  }
  target_class_mapping {
    key: "vehicle15"
    value: "vehicle15"
  }
  target_class_mapping {
    key: "vehicle16"
    value: "vehicle16"
  }
  target_class_mapping {
    key: "vehicle17"
    value: "vehicle17"
  }
validation_fold: 0
}