未验证 提交 13f0b561 编写于 作者: G Guanghua Yu 提交者: GitHub

fix background_label in nms (#2125)

上级 6f9fd567
...@@ -57,4 +57,3 @@ FCOSPostProcess: ...@@ -57,4 +57,3 @@ FCOSPostProcess:
keep_top_k: 100 keep_top_k: 100
score_threshold: 0.025 score_threshold: 0.025
nms_threshold: 0.6 nms_threshold: 0.6
background_label: -1
...@@ -67,4 +67,3 @@ BBoxPostProcess: ...@@ -67,4 +67,3 @@ BBoxPostProcess:
post_threshold: 0.01 post_threshold: 0.01
nms_top_k: -1 nms_top_k: -1
normalized: false normalized: false
background_label: -1
...@@ -42,4 +42,3 @@ BBoxPostProcess: ...@@ -42,4 +42,3 @@ BBoxPostProcess:
nms_threshold: 0.45 nms_threshold: 0.45
nms_top_k: 1000 nms_top_k: 1000
normalized: false normalized: false
background_label: -1
...@@ -43,4 +43,3 @@ BBoxPostProcess: ...@@ -43,4 +43,3 @@ BBoxPostProcess:
nms_threshold: 0.45 nms_threshold: 0.45
nms_top_k: 1000 nms_top_k: 1000
normalized: false normalized: false
background_label: -1
...@@ -44,4 +44,3 @@ BBoxPostProcess: ...@@ -44,4 +44,3 @@ BBoxPostProcess:
nms_threshold: 0.45 nms_threshold: 0.45
nms_top_k: 1000 nms_top_k: 1000
normalized: false normalized: false
background_label: -1
...@@ -44,4 +44,3 @@ BBoxPostProcess: ...@@ -44,4 +44,3 @@ BBoxPostProcess:
nms_threshold: 0.45 nms_threshold: 0.45
nms_top_k: 1000 nms_top_k: 1000
normalized: false normalized: false
background_label: -1
...@@ -54,7 +54,7 @@ class Trainer(object): ...@@ -54,7 +54,7 @@ class Trainer(object):
self.model = create(cfg.architecture) self.model = create(cfg.architecture)
# model slim build # model slim build
if cfg.slim: if 'slim' in cfg and cfg.slim:
if self.mode == 'train': if self.mode == 'train':
self.load_weights(cfg.pretrain_weights, cfg.weight_type) self.load_weights(cfg.pretrain_weights, cfg.weight_type)
slim = create(cfg.slim) slim = create(cfg.slim)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册