random_seed: 42 yolo_config { big_anchor_shape: "[(39.00, 106.00), (60.00, 141.00), (134.00, 234.00)]" mid_anchor_shape: "[(28.00, 32.00), (26.00, 65.00), (52.00, 55.00)]" small_anchor_shape: "[(14.00, 16.00), (13.00, 31.00), (17.00, 44.00)]" matching_neutral_box_iou: 0.5 arch: "resnet" nlayers: 18 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: 8 num_epochs: 150 learning_rate { soft_start_annealing_schedule { min_learning_rate: 1e-6 max_learning_rate: 5e-5 soft_start: 0.1 annealing: 0.7 } } regularizer { type: L1 weight: 3.0e-09 } } eval_config { validation_period_during_training: 10 average_precision_mode: SAMPLE batch_size: 8 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: 608 output_image_height: 608 output_image_channel: 3 crop_right: 608 crop_bottom: 608 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/yolo/tfrecords/kitti_trainval/kitti_trainval*" image_directory_path: "/workspace/tlt-experiments/data/training" } image_extension: "png" target_class_mapping { key: "bike" value: "bike" } target_class_mapping { key: "bus" value: "bus" } target_class_mapping { key: "car" value: "car" } target_class_mapping { key: "minibus" value: "minibus" } target_class_mapping { key: "truck" value: "truck" } validation_fold: 0 }