未验证 提交 3645208e 编写于 作者: S shangliang Xu 提交者: GitHub

[dev] alter ppyoloe nms params (#6491)

上级 00e961c8
...@@ -39,7 +39,7 @@ PPYOLOEHead: ...@@ -39,7 +39,7 @@ PPYOLOEHead:
beta: 6.0 beta: 6.0
nms: nms:
name: MultiClassNMS name: MultiClassNMS
nms_top_k: 10000 nms_top_k: 1000
keep_top_k: 300 keep_top_k: 300
score_threshold: 0.01 score_threshold: 0.01
nms_threshold: 0.7 nms_threshold: 0.7
...@@ -64,6 +64,6 @@ PPYOLOEHead: ...@@ -64,6 +64,6 @@ PPYOLOEHead:
nms: nms:
name: MultiClassNMS name: MultiClassNMS
nms_top_k: 1000 nms_top_k: 1000
keep_top_k: 100 keep_top_k: 300
score_threshold: 0.01 score_threshold: 0.01
nms_threshold: 0.6 nms_threshold: 0.7
...@@ -17,13 +17,9 @@ width_mult: 0.50 ...@@ -17,13 +17,9 @@ width_mult: 0.50
TrainReader: TrainReader:
batch_size: 32 batch_size: 32
LearningRate:
base_lr: 0.04
epoch: 400 epoch: 400
LearningRate: LearningRate:
base_lr: 0.025 base_lr: 0.04
schedulers: schedulers:
- !CosineDecay - !CosineDecay
max_epochs: 480 max_epochs: 480
...@@ -44,7 +40,7 @@ PPYOLOEHead: ...@@ -44,7 +40,7 @@ PPYOLOEHead:
static_assigner_epoch: 133 static_assigner_epoch: 133
nms: nms:
name: MultiClassNMS name: MultiClassNMS
nms_top_k: 10000 nms_top_k: 1000
keep_top_k: 300 keep_top_k: 300
score_threshold: 0.01 score_threshold: 0.01
nms_threshold: 0.7 nms_threshold: 0.7
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册