random_seed: 42 dataset_config { data_sources: { tfrecords_path: "/workspace/tlt-experiments/nvidia-tlt/data/tfrecords/voc_trainval/*" image_directory_path: "/workspace/tlt-experiments/dataset/voc_960_2012" } data_sources: { tfrecords_path: "/workspace/tlt-experiments/nvidia-tlt/data/tfrecords/coco_trainval/*" image_directory_path: "/workspace/tlt-experiments/dataset/coco_960_2014" } image_extension: 'jpg' target_class_mapping { key: 'person' value: 'person' } validation_data_source: { tfrecords_path: "/workspace/tlt-experiments/nvidia-tlt/data/tfrecords/gar_hc2_trainval/*" image_directory_path: "/workspace/tlt-experiments/dataset/evaluation_dataset/garden" } } augmentation_config { preprocessing { output_image_width: 960 output_image_height: 544 output_image_channel: 3 crop_right: 960 crop_bottom: 544 min_bbox_width: 1.0 min_bbox_height: 1.0 } spatial_augmentation { hflip_probability: 0.5 zoom_min: 1.0 zoom_max: 1.0 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 } } postprocessing_config{ target_class_config{ key: 'person' value: { clustering_config { coverage_threshold: 0.005 dbscan_eps: 0.265 dbscan_min_samples: 0.05 minimum_bounding_box_height: 4 } } } } model_config { pretrained_model_file: "/workspace/tlt-experiments/nvidia-tlt/peoplenet/tlt_peoplenet_vunpruned_v1.0/resnet34_peoplenet.tlt" num_layers: 34 freeze_blocks: 0 arch: 'resnet' use_batch_norm: true objective_set { bbox { scale: 35.0 offset: 0.5 } cov { } } training_precision { backend_floatx: FLOAT32 } } evaluation_config { validation_period_during_training: 10 first_validation_epoch: 30 minimum_detection_ground_truth_overlap { key: 'person' value: 0.5 } evaluation_box_config { key: 'person' value { minimum_height: 40 maximum_height: 9999 minimum_width: 4 maximum_width: 9999 } } } training_config { batch_size_per_gpu: 16 num_epochs: 30 learning_rate { soft_start_annealing_schedule { min_learning_rate: 5e-06 max_learning_rate: 0.0005 soft_start: 0.1 annealing: 0.7 } } regularizer { type: L1 weight: 3e-09 } optimizer { adam { epsilon: 9.9e-09 beta1: 0.9 beta2: 0.999 } } cost_scaling { initial_exponent: 20.0 increment: 0.005 decrement: 1.0 } checkpoint_interval: 10 } cost_function_config { target_classes { name: 'person' class_weight: 1.0 coverage_foreground_weight: 0.0500000007451 objectives { name: 'cov' initial_weight: 1.0 weight_target: 1.0 } objectives { name: 'bbox' initial_weight: 10.0 weight_target: 10.0 } } enable_autoweighting: true max_objective_weight: 0.999899983406 min_objective_weight: 9.99999974738e-05 } bbox_rasterizer_config { target_class_config { key: 'person' value { cov_center_x: 0.5 cov_center_y: 0.5 cov_radius_x: 0.40000000596 cov_radius_y: 0.40000000596 bbox_min_radius: 1.0 } } deadzone_radius: 0.400000154972 }