diff --git a/configs/ppyoloe/_base_/ppyoloe_crn.yml b/configs/ppyoloe/_base_/ppyoloe_crn.yml index 2fde30fb51139a4cca9a617e4847e3498fe29066..7abee87a2433bd6b83a5257f254809f5acda3908 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 5b4d644fe923d5ea84e26e1edbddd150a14bae1c..3797288864c978f7872c9b994af2477bb6c31acd 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 dcc1514c2719a0c916ffcf52260e52ccfdd54ea2..c3cddf48bb75d733aaf15bfbd148b8e480d48493 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