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
SOLOv2-pytorch
Commits
a6ec45ba
Commit
a6ec45ba
authored
Apr 10, 2019
by
Kai Chen
Browse files
add more map terms in the log
parent
12d91ee9
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
8 additions
and
2 deletions
+8
-2
mmdet/core/evaluation/eval_hooks.py
mmdet/core/evaluation/eval_hooks.py
+8
-2
No files found.
mmdet/core/evaluation/eval_hooks.py
View file @
a6ec45ba
...
...
@@ -150,7 +150,13 @@ class CocoDistEvalmAPHook(DistEvalHook):
cocoEval
.
evaluate
()
cocoEval
.
accumulate
()
cocoEval
.
summarize
()
field
=
'{}_mAP'
.
format
(
res_type
)
runner
.
log_buffer
.
output
[
field
]
=
cocoEval
.
stats
[
0
]
metrics
=
[
'mAP'
,
'mAP_50'
,
'mAP_75'
,
'mAP_s'
,
'mAP_m'
,
'mAP_l'
]
for
i
in
range
(
len
(
metrics
)):
key
=
'{}_{}'
.
format
(
res_type
,
metrics
[
i
])
val
=
float
(
'{:.3f}'
.
format
(
cocoEval
.
stats
[
i
]))
runner
.
log_buffer
.
output
[
key
]
=
val
runner
.
log_buffer
.
output
[
'{}_mAP_copypaste'
.
format
(
res_type
)]
=
(
'{ap[0]:.3f} {ap[1]:.3f} {ap[2]:.3f} {ap[3]:.3f} '
'{ap[4]:.3f} {ap[5]:.3f}'
).
format
(
ap
=
cocoEval
.
stats
[:
6
])
runner
.
log_buffer
.
ready
=
True
os
.
remove
(
tmp_file
)
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