random_seed: 42
dataset_config {
	data_sources: {
		tfrecords_path: "/workspace/tlt-experiments/Dataset/tfrecords/peopletrain/*"
		image_directory_path: "/workspace/tlt-experiments/data/images"
        
	}
	image_extension: "jpg"
	target_class_mapping {
		key: "person"
		value: "person"
	}
	validation_fold: 0

}

augmentation_config {
	
	preprocessing {
		output_image_width: 960
		output_image_height: 544
		crop_right: 960
		crop_bottom: 544
		min_bbox_width: 1.0
		min_bbox_height: 1.0
        output_image_channel: 3
	}
	
	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/dataset/resnet34_peoplenet.tlt"
	num_layers: 34
	#freeze_blocks: 0
	arch: "resnet"
	use_batch_norm: true
	#load_graph:true
	objective_set {
		
		bbox {
		scale: 35.0
		offset: 0.5
		}
		
		cov {
		}
	}
	training_precision {
		backend_floatx: FLOAT32
	}
}