random_seed: 42 yolov4_config { big_anchor_shape: "[(392.44, 330.96), (395.08, 277.88), (413.56, 414.38)]" mid_anchor_shape: "[(344.09, 168.46), (346.94, 399.21), (361.97, 232.38)]" small_anchor_shape: "[(210.03, 361.02), (281.53, 382.96), (328.25, 296.02)]" 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 freeze_blocks: 0 force_relu: false } training_config { batch_size_per_gpu: 8 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" #resume_model_path: "/workspace/tao-experiments/results/weights/yolov4_resnet18_epoch_50.tlt" } 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: 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: "Biscuits" value: "Biscuits" } target_class_mapping { key: "Chips" value: "Chips" } target_class_mapping { key: "Chocolate" value: "Chocolate" } validation_data_sources: { label_directory_path: "/workspace/tao-experiments/val/labels" image_directory_path: "/workspace/tao-experiments/val/images" } }