random_seed: 42 model_config { model_input_width: 640 model_input_height: 480 model_input_channels: 3 num_layers: 18 all_projections: false arch: "resnet" use_batch_norm: False training_precision { backend_floatx: FLOAT32 } } training_config { batch_size: 3 epochs: 200 log_summary_steps: 10 checkpoint_interval: 1 loss: "cross_entropy" learning_rate:0.0001 regularizer { type: L2 weight: 2e-5 } optimizer { adam { epsilon: 9.99999993923e-09 beta1: 0.899999976158 beta2: 0.999000012875 } } } dataset_config { dataset: "custom" augment: False augmentation_config { spatial_augmentation { hflip_probability : 0.5 vflip_probability : 0 crop_and_resize_prob : 0 } brightness_augmentation { delta: 0.2 } } input_image_type: "color" train_images_path:"/home/ubuntu/Desktop/dpstream/tlt-experiments/data/train" train_masks_path:"/home/ubuntu/Desktop/dpstream/tlt-experiments/data/train_labels" val_images_path:"/home/ubuntu/Desktop/dpstream/tlt-experiments/data/val" val_masks_path:"/home/ubuntu/Desktop/dpstream/tlt-experiments/data/val_labels" test_images_path:"/home/ubuntu/Desktop/dpstream/tlt-experiments/data/test" data_class_config { target_classes { name: "Truck" mapping_class: "Truck" label_id: 0 } target_classes { name: "People" mapping_class: "People" label_id: 1 } target_classes { name: "Car" mapping_class: "Car" label_id: 2 } target_classes { name: "Fence" mapping_class: "Fence" label_id: 3 } target_classes { name: "TrainTrack" mapping_class: "TrainTrack" label_id: 4 } target_classes { name: "TrainRoad" mapping_class: "TrainRoad" label_id: 5 } target_classes { name: "Train" mapping_class: "Train" label_id: 6 } target_classes { name: "Vegetation" mapping_class: "Vegetation" label_id: 7 } target_classes { name: "Building" mapping_class: "Building" label_id: 8 } target_classes { name: "CarRoad" mapping_class: "CarRoad" label_id: 9 } target_classes { name: "RoadSign" mapping_class: "RoadSign" label_id: 10 } target_classes { name: "Irrelevant" mapping_class: "Irrelevant" label_id: 11 } target_classes { name: "TrainSign" mapping_class: "TrainSign" label_id: 12 } target_classes { name: "Station" mapping_class: "Station" label_id: 13 } target_classes { name: "Unwalkable" mapping_class: "Unwalkable" label_id: 14 } target_classes { name: "Pole" mapping_class: "Pole" label_id: 15 } target_classes { name: "Sky" mapping_class: "Sky" label_id: 16 } target_classes { name: "Dynamic" mapping_class: "Dynamic" label_id: 17 } } }