Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
ModelZoo
ResNet50_tensorflow
Commits
2b436a12
"tests/vscode:/vscode.git/clone" did not exist on "0612f48cd05f47e238256392d6b45a38875f55b8"
Commit
2b436a12
authored
Oct 21, 2020
by
A. Unique TensorFlower
Browse files
Internal change
PiperOrigin-RevId: 338364871
parent
95f2b046
Changes
1
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
6 additions
and
3 deletions
+6
-3
official/vision/beta/evaluation/coco_evaluator.py
official/vision/beta/evaluation/coco_evaluator.py
+6
-3
No files found.
official/vision/beta/evaluation/coco_evaluator.py
View file @
2b436a12
...
...
@@ -190,9 +190,12 @@ class COCOEvaluator(object):
if
hasattr
(
coco_eval
,
'category_stats'
):
for
category_index
,
category_id
in
enumerate
(
coco_eval
.
params
.
catIds
):
if
self
.
_annotation_file
:
coco_category
=
self
.
_coco_gt
.
cats
[
category_id
]
# if 'name' is available use it, otherwise use `id`
category_display_name
=
coco_category
.
get
(
'name'
,
category_id
)
else
:
category_display_name
=
category_id
metrics_dict
[
prefix
+
'Precision mAP ByCategory/{}'
.
format
(
category_display_name
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment