Browse Source

!12696 Fix FaceDetection network eval fail bug

From: @zhanghuiyao
Reviewed-by: 
Signed-off-by:
tags/v1.2.0-rc1
mindspore-ci-bot Gitee 4 years ago
parent
commit
2197f01d0c
1 changed files with 6 additions and 10 deletions
  1. +6
    -10
      model_zoo/research/cv/FaceDetection/eval.py

+ 6
- 10
model_zoo/research/cv/FaceDetection/eval.py View File

@@ -48,13 +48,14 @@ def parse_args():
parser.add_argument('--local_rank', type=int, default=0, help='current rank to support distributed')
parser.add_argument('--world_size', type=int, default=1, help='current process number to support distributed')

args, _ = parser.parse_known_args()
arg, _ = parser.parse_known_args()

return args
return arg


def val(args):
'''eval'''
if __name__ == "__main__":
args = parse_args()

print('=============yolov3 start evaluating==================')

# logger
@@ -190,7 +191,7 @@ def val(args):

aps_str = ''
for cls in evaluate:
per_line, = plt.plot(evaluate[cls]['recall'], evaluate[cls]['presicion'], 'b-')
per_line, = plt.plot(evaluate[cls]['recall'], evaluate[cls]['precision'], 'b-')
per_line.set_label('%s:AP=%.3f' % (cls, evaluate[cls]['ap']))
aps_str += '_%s_AP_%.3f' % (cls, evaluate[cls]['ap'])
plt.plot([i / 1000.0 for i in range(1, 1001)], [i / 1000.0 for i in range(1, 1001)], 'y--')
@@ -208,8 +209,3 @@ def val(args):
plt.savefig(ap_save_path)

print('=============yolov3 evaluating finished==================')


if __name__ == "__main__":
arg = parse_args()
val(arg)

Loading…
Cancel
Save