yolov4_config { big_anchor_shape: "[(271.00, 199.47), (119.00, 186.37), (183.00, 96.71)]" mid_anchor_shape: "[(63.00, 145.07), (87.00, 83.61), (118.00, 57.42)]" small_anchor_shape: "[(45.00, 72.53), (63.00, 46.34), (35.00, 35.26)]" box_matching_iou: 0.25 arch: "cspdarknet" nlayers: 19 arch_conv_blocks: 2 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.1 small_grid_xy_extend: 0.2 freeze_bn: False freeze_blocks: 0 force_relu: False } training_config { batch_size_per_gpu: 2 num_epochs: 20 enable_qat: False checkpoint_interval: 1 learning_rate { soft_start_annealing_schedule { min_learning_rate: 1e-06 max_learning_rate: 0.0001 soft_start: 1e-05 annealing: 0.05 } } regularizer { type: L1 weight: 3e-05 } optimizer { adam { epsilon: 1e-07 beta1: 0.9 beta2: 0.999 amsgrad: False } } pretrain_model_path: "/workspace/TLT/pretrained_models/cspdarknet_19.hdf5" resume_model_path: "/workspace/TLT/T_12/models/weights/yolov4_cspdarknet19_epoch_009.tlt" } eval_config { average_precision_mode: SAMPLE batch_size: 2 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: 1920 output_height: 1088 output_channel: 3 randomize_input_shape_period: 0 mosaic_prob: 0.5 mosaic_min_ratio:0.2 image_mean { key: 'b' value: 114.43321267514027 } image_mean { key: 'g' value: 118.0073516752478 } image_mean { key: 'r' value: 117.51632311746967 } } dataset_config { data_sources: { label_directory_path: "/workspace/DAB/D_6/train/labels" image_directory_path: "/workspace/DAB/D_6/train/images" } include_difficult_in_training: false target_class_mapping { key: "p_1" value: "p" } target_class_mapping { key: "p_2" value: "p" } target_class_mapping { key: "p_3" value: "p" } target_class_mapping { key: "p_4" value: "p" } target_class_mapping { key: "p_5" value: "p" } target_class_mapping { key: "p_6" value: "p" } target_class_mapping { key: "p_7" value: "p" } target_class_mapping { key: "p_8" value: "p" } target_class_mapping { key: "r_1" value: "r" } target_class_mapping { key: "r_2" value: "r" } target_class_mapping { key: "r_3" value: "r" } target_class_mapping { key: "r_4" value: "r" } target_class_mapping { key: "r_5" value: "r" } target_class_mapping { key: "r_6" value: "r" } target_class_mapping { key: "r_7" value: "r" } target_class_mapping { key: "r_8" value: "r" } validation_data_sources: { label_directory_path: "/workspace/DAB/D_6/val/labels" image_directory_path: "/workspace/DAB/D_6/val/images" } }