mirrors/PaddleClas
1
0
Fork 0
mirror of https://github.com/PaddlePaddle/PaddleClas.git synced 2025-06-03 21:55:06 +08:00
Code Issues Projects Releases Wiki Activity

Update metrics.py

Browse Source
This commit is contained in:
Bin Lu 2021-06-10 10:41:58 +08:00 committed by GitHub
parent 9e83fb4c2f
commit d6fe8c0c1b
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 0 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
ppcls/metric/metrics.py
View File

@ -100,7 +100,6 @@ class RetriMetric(nn.Layer):
assert isinstance(topk, (int, list, tuple)) assert isinstance(topk, (int, list, tuple))
if isinstance(topk, int): if isinstance(topk, int):
topk = [topk] topk = [topk]
self.topk = topk
for k in topk: for k in topk:
metric_dict["recall{}".format(k)] = all_cmc[k - 1] metric_dict["recall{}".format(k)] = all_cmc[k - 1]
if "mAP" in self.config.keys(): if "mAP" in self.config.keys():