mindspore-Ecosystem
/
mindspore
21
0
Fork 27
Code Issues Pull Requests 6 Projects Releases Wiki Activity

!12696 Fix FaceDetection network eval fail bug 

From: @zhanghuiyao
Reviewed-by: 
Signed-off-by:
This commit is contained in:
mindspore-ci-bot 2021-03-01 11:52:01 +08:00 committed by Gitee
parent 0f3c5b1d0f fa1dc51068
commit 2197f01d0c
1 changed files with 6 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
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)