random_seed: 42 yolo_config { big_anchor_shape: "[(474.00, 464.00), (576.00, 447.00), (638.00, 576.00)]" mid_anchor_shape: "[(407.00, 364.00), (407.00, 411.00), (500.00, 390.00)]" small_anchor_shape: "[(232.88, 138.00), (314.81, 360.00), (376.00, 304.03)]" matching_neutral_box_iou: 0.5 arch: "mobilenet_v2" nlayers: 53 arch_conv_blocks: 2 loss_loc_weight: 0.75 loss_neg_obj_weights: 200.0 loss_class_weights: 1.0 freeze_blocks: 0 freeze_bn: false } training_config { batch_size_per_gpu: 2 num_epochs: 80 enable_qat: false learning_rate { soft_start_annealing_schedule { min_learning_rate: 1e-6 max_learning_rate: 1e-4 soft_start: 0.1 annealing: 0.8 } } regularizer { type: L1 weight: 5e-5 } } eval_config { validation_period_during_training: 10 average_precision_mode: SAMPLE batch_size: 2 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: 480 output_image_height: 320 output_image_channel: 3 crop_right: 480 crop_bottom: 320 min_bbox_width: 1.0 min_bbox_height: 1.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/data/tfrecords/kitti_trainval/kitti_trainval*" image_directory_path: "/workspace/tlt-experiments/data/training" } image_extension: "jpg" target_class_mapping { key: "box" value: "box" } target_class_mapping { key: "bag" value: "bag" } validation_fold: 0 }