LPD Training Spec for YoloV4?

[HighMans]

Hello,

Would you be able to post the training specifications used to generate the LPDNet YoloV4 model?

Thank you!

[Morganh]

Just share with you for reference. But please note that the yolov4 LPDnet model was trained about one year ago. So, some parameters are not compatible with latest TAO docker.

yolov4_config {
  big_anchor_shape: "[(161.00, 80.00), (130.08, 136.36),  (190.48, 184.62)]"
  mid_anchor_shape: "[(121.77, 54.04), (100.00, 100.00),(87.91, 117.65)]"
  box_matching_iou: 0.25
  matching_neutral_box_iou: 0.5
  arch: "cspdarknet_tiny"
  arch_conv_blocks: 0
  loss_loc_weight: 0.8
  loss_neg_obj_weights: 100.0
  loss_class_weights: 0.5
  label_smoothing: 0.0
  big_grid_xy_extend: 0.05
  mid_grid_xy_extend: 0.05
  freeze_bn: false
  force_relu: false
}
training_config {
  batch_size_per_gpu: 8
  num_epochs: 120
  enable_qat: false
  checkpoint_interval: 1
  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
    }
  }
  # resume_model_path: "/workspace/lpd/yolo_v4_tiny_result/weights/yolov4_cspdarknet_tiny_epoch_048.tlt"
  # pretrain_model_path: "/workspace/tlt-experiments/tltv2_pretrain_model/resnet_18.hdf5"
}
eval_config {
  average_precision_mode: SAMPLE
  batch_size: 8
  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: 640
  output_height: 480
  output_channel: 3
  randomize_input_shape_period: 10
  mosaic_prob: 0.5
  mosaic_min_ratio:0.2
}
dataset_config {
  data_sources: {
      image_directory_path: "/workspace/lpd/image_640x480"
      label_directory_path: "/workspace/lpd/label_640_480"
  }
  include_difficult_in_training: true
  target_class_mapping {
    key: "lpd"
    value: "lpd"
  }
  validation_data_sources: {
      image_directory_path: "/workspace/lpd/3k_testImg_640_480"
      label_directory_path: "/workspace/lpd/3k_testImg_640_480_label"
  }
}