diff --git a/model_zoo/research/cv/FaceDetection/eval.py b/model_zoo/research/cv/FaceDetection/eval.py index fd42a35153..c2052911c5 100644 --- a/model_zoo/research/cv/FaceDetection/eval.py +++ b/model_zoo/research/cv/FaceDetection/eval.py @@ -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)