# Copyright (c) 2017-2020, NVIDIA CORPORATION.  All rights reserved.
random_seed: 42
enc_key: 'tlt'
verbose: True
model_config {
input_image_config {
image_type: RGB
image_channel_order: 'bgr'
size_height_width {
height: 512
width: 512
}
    image_channel_mean {
        key: 'b'
        value: 103.939
}
    image_channel_mean {
        key: 'g'
        value: 116.779
}
    image_channel_mean {
        key: 'r'
        value: 123.68
}
image_scaling_factor: 1.0
max_objects_num_per_image: 100
}
arch: "resnet:18"
# https://medium.com/@smallfishbigsea/faster-r-cnn-explained-864d4fb7e3f8
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/insect-thesis/data/kitti/final/tfrecords/kitti_trainval/kitti_trainval*"
    image_directory_path: "/workspace/tlt-experiments/insect-thesis/data/kitti/final/train"
  }
image_extension: 'jpg'

target_class_mapping {
key: 'noctuidae'
value: 'noctuidae'
}
target_class_mapping {
key: 'geometridae'
value: 'geometridae'
}
target_class_mapping {
key: 'coleoptera'
value: 'coleoptera'
}
target_class_mapping {
key: 'diptera'
value: 'diptera'
}
target_class_mapping {
key: 'odonata'
value: 'odonata'
}
target_class_mapping {
key: 'orthoptera'
value: 'orthoptera'
}
target_class_mapping {
key: 'hemiptera'
value: 'hemiptera'
}
target_class_mapping {
key: 'hymenoptera'
value: 'hymenoptera'
}
target_class_mapping {
key: 'trichoptera'
value: 'trichoptera'
}

validation_fold: 0
}
augmentation_config {
preprocessing {
output_image_width: 512
output_image_height: 512
output_image_channel: 3
min_bbox_width: 5.0
min_bbox_height: 5.0
}
spatial_augmentation {
hflip_probability: 0.5
vflip_probability: 0.5
zoom_min: 0.4
zoom_max: 1.0
translate_max_x: 200
translate_max_y: 200
}
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: 4 # best 4
num_epochs: 30
pretrained_weights: "/workspace/tlt-experiments/insect-thesis/data/faster_rcnn/resnet_18.hdf5"
resume_from_model: "/workspace/tlt-experiments/insect-thesis/data/kitti/final/faster_rcnn/frcnn_kitti_resnet18.epoch26.tlt"
output_model: "/workspace/tlt-experiments/insect-thesis/data/kitti/final/faster_rcnn/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/insect-thesis/data/kitti/final/inference/images'
model: '/workspace/tlt-experiments/insect-thesis/data/kitti/final/faster_rcnn/frcnn_kitti_resnet18.epoch1.tlt'
batch_size: 1
detection_image_output_dir: '/workspace/tlt-experiments/insect-thesis/data/kitti/final/faster_rcnn/inference_images'
labels_dump_dir: '/workspace/tlt-experiments/insect-thesis/data/kitti/final/faster_rcnn/inference_labels'
rpn_pre_nms_top_N: 6000
rpn_nms_max_boxes: 300
rpn_nms_overlap_threshold: 0.7
object_confidence_thres: 0.4
bbox_visualize_threshold: 0.4
classifier_nms_max_boxes: 40
classifier_nms_overlap_threshold: 0.5
}
evaluation_config {
model: '/workspace/tlt-experiments/insect-thesis/data/kitti/final/faster_rcnn/frcnn_kitti_resnet18.epoch1.tlt'
batch_size: 1
validation_period_during_training: 1
labels_dump_dir: '/workspace/tlt-experiments/insect-thesis/data/kitti/final/faster_rcnn/inference_labels'
rpn_pre_nms_top_N: 6000
rpn_nms_max_boxes: 300
rpn_nms_overlap_threshold: 0.7
classifier_nms_max_boxes: 40
classifier_nms_overlap_threshold: 0.4
object_confidence_thres: 0.4
use_voc07_11point_metric:False
gt_matching_iou_threshold: 0.5
}