random_seed: 42
enc_key: 'nvidia_tlt'
verbose: True
model_config {
  input_image_config {
  image_type: RGB
  image_channel_order: 'bgr'
  size_height_width {
  height: 1080
  width: 1920
  }
      image_channel_mean {
          key: 'b'
          value: 126
  }
      image_channel_mean {
          key: 'g'
          value: 211
  }
      image_channel_mean {
          key: 'r'
          value: 33
  }
  image_scaling_factor: 1.0
  max_objects_num_per_image: 200
  }
  arch: "resnet:18"
  anchor_box_config {
    scale: 64.0
    scale: 128.0
    scale: 256.0
    ratio: 1.0
    ratio: 0.5
    ratio: 2.0
  }
  freeze_bn: True
  freeze_blocks: 0
  freeze_blocks: 1
  roi_mini_batch: 256
  rpn_stride: 16
  use_bias: False
  roi_pooling_config {
    pool_size: 7
    pool_size_2x: False
  }
  all_projections: True
  use_pooling:False
}
dataset_config {
  data_sources: {
    tfrecords_path: "/workspace/tlt-experiments/FASTER_RCNN/MANZANA_DA_TV_FULL_BBOX_RESIZED/TFRECORDS/*"
    image_directory_path: "/workspace/tlt-experiments/FASTER_RCNN/MANZANA_DA_TV_FULL_BBOX_RESIZED/"
  }
  image_extension: 'jpg'
  target_class_mapping {
    key: 'manzana'
    value: 'manzana'
  }
  validation_fold: 0
}
augmentation_config {
  preprocessing {
    output_image_width: 1920
    output_image_height: 1080
    output_image_channel: 3
    min_bbox_width: 1.0
    min_bbox_height: 1.0
  }
  spatial_augmentation {
    hflip_probability: 0.5
    vflip_probability: 0.0
    zoom_min: 1.0
    zoom_max: 1.0
    translate_max_x: 0
    translate_max_y: 0
  }
  color_augmentation {
    hue_rotation_max: 0.0
    saturation_shift_max: 0.0
    contrast_scale_max: 0.0
    contrast_center: 0.5
  }
}
training_config {
  enable_augmentation: True
  enable_qat: False
  batch_size_per_gpu: 1
  num_epochs: 500
  #pretrained_weights: "/workspace/tlt-experiments/FASTER_RCNN/pretrained_resnet18/resnet_18.hdf5"
  resume_from_model: "/workspace/tlt-experiments/FASTER_RCNN/MANZANA_DA_TV_FULL_BBOX_RESIZED/weights/frcnn_kitti_resnet18.epoch200.tlt"
  output_model: "/workspace/tlt-experiments/FASTER_RCNN/MANZANA_DA_TV_FULL_BBOX_RESIZED/weights/frcnn_kitti_resnet18.tlt"
  rpn_min_overlap: 0.3
  rpn_max_overlap: 0.7
  classifier_min_overlap: 0.0
  classifier_max_overlap: 0.5
  gt_as_roi: False
  std_scaling: 1.0
  classifier_regr_std {
    key: 'x'
    value: 10.0
  }
  classifier_regr_std {
    key: 'y'
    value: 10.0
  }
  classifier_regr_std {
    key: 'w'
    value: 5.0
  }
  classifier_regr_std {
    key: 'h'
    value: 5.0
  }

  rpn_mini_batch: 256
  rpn_pre_nms_top_N: 12000
  rpn_nms_max_boxes: 2000
  rpn_nms_overlap_threshold: 0.7

  regularizer {
    type: L2
    weight: 1e-4
  }

  optimizer {
    sgd {
      lr: 0.02
      momentum: 0.9
      decay: 0.0
      nesterov: False
    }
  }

  learning_rate {
    soft_start {
      base_lr: 0.02
      start_lr: 0.002
      soft_start: 0.1
      annealing_points: 0.8
      annealing_points: 0.9
      annealing_divider: 10.0
    }
  }

  lambda_rpn_regr: 1.0
  lambda_rpn_class: 1.0
  lambda_cls_regr: 1.0
  lambda_cls_class: 1.0
}
inference_config {
  images_dir: '/workspace/tlt-experiments/FASTER_RCNN/MANZANA_DA_TV_FULL_BBOX_RESIZED/IMG_TEST'
  model: '/workspace/tlt-experiments/FASTER_RCNN/MANZANA_DA_TV_FULL_BBOX_RESIZED/weights/frcnn_kitti_resnet18.epoch200.tlt'
  batch_size: 1
  detection_image_output_dir: '/workspace/tlt-experiments/FASTER_RCNN/MANZANA_DA_TV_FULL_BBOX_RESIZED/output'
  labels_dump_dir: '/workspace/tlt-experiments/FASTER_RCNN/MANZANA_DA_TV_FULL_BBOX_RESIZED/output_label'
  rpn_pre_nms_top_N: 6000
  rpn_nms_max_boxes: 300
  rpn_nms_overlap_threshold: 0.7
  object_confidence_thres: 0.0001
  bbox_visualize_threshold: 0.6
  classifier_nms_max_boxes: 100
  classifier_nms_overlap_threshold: 0.3
  #trt_inference {
  #trt_engine: '/workspace/tlt-experiments/data/faster_rcnn/trt.int8.engine'
  #}
}
evaluation_config {
  model: '/workspace/tlt-experiments/FASTER_RCNN/MANZANA_DA_TV_FULL_BBOX_RESIZED/weights/frcnn_kitti_resnet18.tlt'
  batch_size: 1
  validation_period_during_training: 100
  rpn_pre_nms_top_N: 6000
  rpn_nms_max_boxes: 300
  rpn_nms_overlap_threshold: 0.7
  classifier_nms_max_boxes: 100
  classifier_nms_overlap_threshold: 0.3
  object_confidence_thres: 0.0001
  use_voc07_11point_metric:False
  #trt_evaluation {
  #trt_engine: '/workspace/tlt-experiments/data/faster_rcnn/trt.int8.engine'
  #}
  gt_matching_iou_threshold: 0.5
}