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

[dev] alter ppyoloe nms params (#6491)

上级 00e961c8
......@@ -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
......@@ -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
......@@ -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
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册