From 4e87fbe7bcb37f6342d7a4c7d2851bbed26e9e2a Mon Sep 17 00:00:00 2001 From: dengkaipeng Date: Thu, 11 Jun 2020 06:09:57 +0000 Subject: [PATCH] update config --- configs/yolov4/yolov4_cspdarknet_coco.yml | 5 +++-- configs/yolov4/yolov4_cspdarknet_voc.yml | 6 +++--- 2 files changed, 6 insertions(+), 5 deletions(-) diff --git a/configs/yolov4/yolov4_cspdarknet_coco.yml b/configs/yolov4/yolov4_cspdarknet_coco.yml index b4f7908fa..41821ec2e 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 cc9629a16..24751f863 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 -- GitLab