random_seed: 42
yolov4_config {
  big_anchor_shape: "[(114.94, 60.67), (159.06, 114.59), (297.59, 176.38)]"
  mid_anchor_shape: "[(42.99, 31.91), (79.57, 31.75), (56.80, 56.93)]"
  small_anchor_shape: "[(15.60, 13.88), (30.25, 20.25), (20.67, 49.63)]"
  box_matching_iou: 0.25
  matching_neutral_box_iou: 0.5
  arch: "resnet"
  nlayers: 18
  arch_conv_blocks: 2
  loss_loc_weight: 1.0
  loss_neg_obj_weights: 1.0
  loss_class_weights: 1.0
  label_smoothing: 0.0
  big_grid_xy_extend: 0.05
  mid_grid_xy_extend: 0.1
  small_grid_xy_extend: 0.2
  freeze_bn: false
  force_relu: false
}
training_config {
  batch_size_per_gpu: 32
  num_epochs: 100
  enable_qat: false
  checkpoint_interval: 10
  learning_rate {
    soft_start_cosine_annealing_schedule {
      min_learning_rate: 1e-7
      max_learning_rate: 1e-4
      soft_start: 0.3
    }
  }
  regularizer {
    type: L1
    weight: 3e-5
  }
  optimizer {
    adam {
      epsilon: 1e-7
      beta1: 0.9
      beta2: 0.999
      amsgrad: false
    }
  }
  pretrain_model_path: "/workspace/tao-experiments/pretrained/resnet_18.hdf5"
}
eval_config {
  average_precision_mode: SAMPLE
  batch_size: 32
  matching_iou_threshold: 0.5
}
nms_config {
  confidence_threshold: 0.001
  clustering_iou_threshold: 0.5
  top_k: 200
}
augmentation_config {
  hue: 0.1
  saturation: 1.5
  exposure:1.5
  vertical_flip:0
  horizontal_flip: 0.5
  jitter: 0.3
  output_width: 416
  output_height: 416
  output_channel: 3
  randomize_input_shape_period: 100
  mosaic_prob: 0.5
  mosaic_min_ratio:0.2
  image_mean {
    key: 'b'
    value: 103.9
  }
  image_mean {
    key: 'g'
    value: 116.8
  }
  image_mean {
    key: 'r'
    value: 123.7
  }
}
dataset_config {
  data_sources: {
       label_directory_path: "/workspace/tao-experiments/labels"
       image_directory_path: "/workspace/tao-experiments/images"
  }
  include_difficult_in_training: true
  image_extension: "jpg"
  target_class_mapping { key: 'iphone5s' value: 'iphone5s'}
target_class_mapping { key: 'oppoa1' value: 'oppoa1'}
target_class_mapping { key: 'iphone13progold' value: 'iphone13progold'}
target_class_mapping { key: 'oneplus7t' value: 'oneplus7t'}
target_class_mapping { key: 'oppox3pro' value: 'oppox3pro'}
target_class_mapping { key: 'iphone11promax' value: 'iphone11promax'}
target_class_mapping { key: 'oneplus8t' value: 'oneplus8t'}
target_class_mapping { key: 'iphone12problue' value: 'iphone12problue'}
target_class_mapping { key: 'iphone12miniblack' value: 'iphone12miniblack'}
target_class_mapping { key: 'iphone8' value: 'iphone8'}
target_class_mapping { key: 'oppoa73' value: 'oppoa73'}
target_class_mapping { key: 'samsungj7prime' value: 'samsungj7prime'}
target_class_mapping { key: 'iphone11pro' value: 'iphone11pro'}
target_class_mapping { key: 'iphone12' value: 'iphone12'}
target_class_mapping { key: 'iphonexr' value: 'iphonexr'}
target_class_mapping { key: 'vivov15pro' value: 'vivov15pro'}
target_class_mapping { key: 'vivos1' value: 'vivos1'}
target_class_mapping { key: 'iphone12purple' value: 'iphone12purple'}
target_class_mapping { key: 'iphonexwhite' value: 'iphonexwhite'}
target_class_mapping { key: 'nokia105' value: 'nokia105'}
target_class_mapping { key: 'iphone12green' value: 'iphone12green'}
target_class_mapping { key: 'iphone12black' value: 'iphone12black'}
target_class_mapping { key: 'iphone13prosilver' value: 'iphone13prosilver'}
target_class_mapping { key: 'iphone12mini' value: 'iphone12mini'}
target_class_mapping { key: 'iphone12white' value: 'iphone12white'}
target_class_mapping { key: 'iphone6se' value: 'iphone6se'}
target_class_mapping { key: 'iphone6swhite' value: 'iphone6swhite'}
target_class_mapping { key: 'vivov20' value: 'vivov20'}
target_class_mapping { key: 'iphone12progold' value: 'iphone12progold'}
target_class_mapping { key: 'oppoa12' value: 'oppoa12'}
target_class_mapping { key: 'redminote10promax' value: 'redminote10promax'}
target_class_mapping { key: 'iphone11white' value: 'iphone11white'}
target_class_mapping { key: 'vivox60pro' value: 'vivox60pro'}
target_class_mapping { key: 'samsungs6edge' value: 'samsungs6edge'}
target_class_mapping { key: 'iphonexblack' value: 'iphonexblack'}
target_class_mapping { key: 'xiaomimi9' value: 'xiaomimi9'}
target_class_mapping { key: 'vivoy50' value: 'vivoy50'}
target_class_mapping { key: 'oneplus9' value: 'oneplus9'}
target_class_mapping { key: 'iphone12prosilver' value: 'iphone12prosilver'}
target_class_mapping { key: 'oneplus9r' value: 'oneplus9r'}
target_class_mapping { key: 'oneplusn200' value: 'oneplusn200'}
target_class_mapping { key: 'xiaomimia3' value: 'xiaomimia3'}
target_class_mapping { key: 'iphone7' value: 'iphone7'}
target_class_mapping { key: 'oppof15' value: 'oppof15'}
target_class_mapping { key: 'oppoa7n' value: 'oppoa7n'}
target_class_mapping { key: 'iphonexswhite' value: 'iphonexswhite'}
target_class_mapping { key: 'iphone13promaxgraphite' value: 'iphone13promaxgraphite'}
target_class_mapping { key: 'iphone13prographite' value: 'iphone13prographite'}
target_class_mapping { key: 'iphone11black' value: 'iphone11black'}
target_class_mapping { key: 'vivoy12' value: 'vivoy12'}

  validation_data_sources: {
      label_directory_path: "/workspace/tao-experiments/val/labels"
      image_directory_path: "/workspace/tao-experiments/val/images"
  }
}