random_seed: 42 dataset_config { data_sources { tfrecords_path: "/workspace/tao-experiments/data/tfrecords/trainval/*" image_directory_path: "/workspace/tao-experiments/data" } image_extension: "jpg" target_class_mapping { key: "car" value: "car" } target_class_mapping { key: "bus" value: "bus" } target_class_mapping { key: "truck" value: "truck" } validation_data_source: { tfrecords_path: "/workspace/tao-experiments/data/tfrecords/test/*" image_directory_path: "/workspace/tao-experiments/data" } } augmentation_config { preprocessing { output_image_width: 960 output_image_height: 544 min_bbox_width: 1.0 min_bbox_height: 1.0 output_image_channel: 3 enable_auto_resize: True } spatial_augmentation { hflip_probability: 0.5 zoom_min: 1.0 zoom_max: 1.0 translate_max_x: 8.0 translate_max_y: 8.0 } } postprocessing_config { target_class_config { key: "car" value { clustering_config { #clustering_algorithm: NMS #nms_iou_threshold: 0.22 #nms_confidence_threshold: 0.6 clustering_algorithm: DBSCAN dbscan_confidence_threshold: 0.6 dbscan_eps: 0.3 dbscan_min_samples: 0.05 coverage_threshold: 0.01 minimum_bounding_box_height: 1 } } } target_class_config { key: "bus" value { clustering_config { #clustering_algorithm: NMS #nms_iou_threshold: 0.2 #nms_confidence_threshold: 0.6 clustering_algorithm: DBSCAN dbscan_confidence_threshold: 0.6 dbscan_eps: 0.3 dbscan_min_samples: 0.05 coverage_threshold: 0.01 minimum_bounding_box_height: 1 } } } target_class_config { key: "truck" value { clustering_config { #clustering_algorithm: NMS #nms_iou_threshold: 0.18 #nms_confidence_threshold: 0.6 clustering_algorithm: DBSCAN dbscan_confidence_threshold: 0.6 dbscan_eps: 0.3 dbscan_min_samples: 0.05 coverage_threshold: 0.01 minimum_bounding_box_height: 1 } } } }