diff --git a/configs/yolov4/yolov4_cspdarknet_coco.yml b/configs/yolov4/yolov4_cspdarknet_coco.yml index b4f7908fa0969c936cc7944a41400c931a2eaf29..41821ec2e8a6fdfac30ab31ac750310c5be5534e 100644 --- a/configs/yolov4/yolov4_cspdarknet_coco.yml +++ b/configs/yolov4/yolov4_cspdarknet_coco.yml @@ -134,8 +134,9 @@ TrainReader: downsample_ratios: [8, 16, 32] iou_thresh: 0.213 batch_size: 8 - mosaic_prob: 0.3 - mosaic_epoch: 200 + mixup_epoch: 250 + mosaic_prob: 0.5 + mosaic_epoch: 260 shuffle: true drop_last: true worker_num: 8 diff --git a/configs/yolov4/yolov4_cspdarknet_voc.yml b/configs/yolov4/yolov4_cspdarknet_voc.yml index cc9629a169ef29545d6e8b5f2836c71349e8d6e7..24751f86346a1322130e4b78e6e3a959da56d1ac 100644 --- a/configs/yolov4/yolov4_cspdarknet_voc.yml +++ b/configs/yolov4/yolov4_cspdarknet_voc.yml @@ -1,6 +1,6 @@ architecture: YOLOv4 use_gpu: true -max_iters: 70000 +max_iters: 72000 log_smooth_window: 20 save_dir: output snapshot_iter: 2000 @@ -135,8 +135,8 @@ TrainReader: iou_thresh: 0.213 batch_size: 8 mixup_epoch: 250 - mosaic_prob: 0.3 - mosaic_epoch: 300 + mosaic_prob: 0.5 + mosaic_epoch: 260 shuffle: true drop_last: true worker_num: 8