未验证 提交 524f56f9 编写于 作者: W wangxinxin08 提交者: GitHub

fix bias problem while training (#2161)

上级 b83aab25
...@@ -115,9 +115,10 @@ class Trainer(object): ...@@ -115,9 +115,10 @@ class Trainer(object):
return return
if self.cfg.metric == 'COCO': if self.cfg.metric == 'COCO':
# TODO: bias should be unified # TODO: bias should be unified
bias = 1 if 'bias' in self.cfg else 0
self._metrics = [ self._metrics = [
COCOMetric( COCOMetric(
anno_file=self.dataset.get_anno(), bias=self.cfg.bias) anno_file=self.dataset.get_anno(), bias=bias)
] ]
elif self.cfg.metric == 'VOC': elif self.cfg.metric == 'VOC':
self._metrics = [ self._metrics = [
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册