提交 5dc93ac0 编写于 作者: W weishengyu

dbg

上级 b23e72b1
......@@ -33,20 +33,11 @@ class CombinedMetrics(nn.Layer):
metric_params = config[metric_name]
self.metric_func_list.append(eval(metric_name)(**metric_params))
def __call__(self,
similarities_matrix,
query_img_id,
gallery_img_id,
x=None,
label=None):
def __call__(self, **kwargs):
metric_dict = OrderedDict()
for idx, metric_func in enumerate(self.metric_func_list):
if x is None:
metric_dict.update(metric_func(x, label))
else:
metric_dict.update(
metric_func(similarities_matrix, query_img_id,
gallery_img_id))
metric_dict.update(metric_func(**kwargs))
return metric_dict
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册