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

fix background_label in nms (#2125)

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