From 45cf03b65e8915f29d73d705f19156796968bb9b Mon Sep 17 00:00:00 2001 From: kaiyangzhou Date: Wed, 28 Apr 2021 16:54:06 +0800 Subject: [PATCH] update print in parse_test_res.py --- tools/parse_test_res.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/tools/parse_test_res.py b/tools/parse_test_res.py index bc57579..fd5b018 100644 --- a/tools/parse_test_res.py +++ b/tools/parse_test_res.py @@ -81,13 +81,15 @@ def main(args): fullpath = os.path.join(args.directory, directory) filepath = glob.glob(os.path.join(fullpath, 'test.log*'))[0] check_isfile(filepath) + print(f'Parsing {filepath}') res = parse_file( filepath, regex_mAP, regex_r1, regex_r5, regex_r10, regex_r20 ) for key, value in res.items(): final_res[key].append(value) - print(f'* Average results over {num_dirs} splits') + print('Finished parsing') + print(f'The average results over {num_dirs} splits are shown below') for key, values in final_res.items(): mean_val = np.mean(values)