From 3645208e752ae8cce1bac934d3453915bb99434b Mon Sep 17 00:00:00 2001 From: shangliang Xu Date: Mon, 25 Jul 2022 11:11:34 +0800 Subject: [PATCH] [dev] alter ppyoloe nms params (#6491) --- configs/ppyoloe/_base_/ppyoloe_crn.yml | 2 +- configs/ppyoloe/ppyoloe_crn_l_36e_coco_xpu.yml | 4 ++-- configs/ppyoloe/ppyoloe_crn_s_400e_coco.yml | 8 ++------ 3 files changed, 5 insertions(+), 9 deletions(-) diff --git a/configs/ppyoloe/_base_/ppyoloe_crn.yml b/configs/ppyoloe/_base_/ppyoloe_crn.yml index 2fde30fb5..7abee87a2 100644 --- a/configs/ppyoloe/_base_/ppyoloe_crn.yml +++ b/configs/ppyoloe/_base_/ppyoloe_crn.yml @@ -39,7 +39,7 @@ PPYOLOEHead: beta: 6.0 nms: name: MultiClassNMS - nms_top_k: 10000 + nms_top_k: 1000 keep_top_k: 300 score_threshold: 0.01 nms_threshold: 0.7 diff --git a/configs/ppyoloe/ppyoloe_crn_l_36e_coco_xpu.yml b/configs/ppyoloe/ppyoloe_crn_l_36e_coco_xpu.yml index 5b4d644fe..379728886 100644 --- a/configs/ppyoloe/ppyoloe_crn_l_36e_coco_xpu.yml +++ b/configs/ppyoloe/ppyoloe_crn_l_36e_coco_xpu.yml @@ -64,6 +64,6 @@ PPYOLOEHead: nms: name: MultiClassNMS nms_top_k: 1000 - keep_top_k: 100 + keep_top_k: 300 score_threshold: 0.01 - nms_threshold: 0.6 + nms_threshold: 0.7 diff --git a/configs/ppyoloe/ppyoloe_crn_s_400e_coco.yml b/configs/ppyoloe/ppyoloe_crn_s_400e_coco.yml index dcc1514c2..c3cddf48b 100644 --- a/configs/ppyoloe/ppyoloe_crn_s_400e_coco.yml +++ b/configs/ppyoloe/ppyoloe_crn_s_400e_coco.yml @@ -17,13 +17,9 @@ width_mult: 0.50 TrainReader: batch_size: 32 -LearningRate: - base_lr: 0.04 - - epoch: 400 LearningRate: - base_lr: 0.025 + base_lr: 0.04 schedulers: - !CosineDecay max_epochs: 480 @@ -44,7 +40,7 @@ PPYOLOEHead: static_assigner_epoch: 133 nms: name: MultiClassNMS - nms_top_k: 10000 + nms_top_k: 1000 keep_top_k: 300 score_threshold: 0.01 nms_threshold: 0.7 -- GitLab