From 4944957f30371517d56bd4a616c0e22860669e45 Mon Sep 17 00:00:00 2001 From: Vladimir Date: Wed, 24 Apr 2024 11:50:49 +0200 Subject: [PATCH] Fix Game State "Recognition" -> "Reconstruction" --- tracklab/wrappers/datasets/soccernet/soccernet_game_state.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tracklab/wrappers/datasets/soccernet/soccernet_game_state.py b/tracklab/wrappers/datasets/soccernet/soccernet_game_state.py index 43c16c7a..3b454643 100644 --- a/tracklab/wrappers/datasets/soccernet/soccernet_game_state.py +++ b/tracklab/wrappers/datasets/soccernet/soccernet_game_state.py @@ -45,7 +45,7 @@ def process_trackeval_results(self, results, dataset_config, eval_config): # In all keys, replace the substring "HOTA" with "GS-HOTA" combined_results['GS-HOTA'] = {k.replace('HOTA', 'GS-HOTA'): v for k, v in combined_results['GS-HOTA'].items()} - log.info(f"SoccerNet Game State Recognition performance GS-HOTA = {combined_results['GS-HOTA']['GS-HOTA']}% (config: EVAL_SPACE={dataset_config['EVAL_SPACE']}, USE_JERSEY_NUMBERS={dataset_config['USE_JERSEY_NUMBERS']}, USE_TEAMS={dataset_config['USE_TEAMS']}, USE_ROLES={dataset_config['USE_ROLES']}, EVAL_DIST_TOL={dataset_config['EVAL_DIST_TOL']})") + log.info(f"SoccerNet Game State Reconstruction performance GS-HOTA = {combined_results['GS-HOTA']['GS-HOTA']}% (config: EVAL_SPACE={dataset_config['EVAL_SPACE']}, USE_JERSEY_NUMBERS={dataset_config['USE_JERSEY_NUMBERS']}, USE_TEAMS={dataset_config['USE_TEAMS']}, USE_ROLES={dataset_config['USE_ROLES']}, EVAL_DIST_TOL={dataset_config['EVAL_DIST_TOL']})") log.info(f"Have a look at 'tracklab/tracklab/configs/dataset/soccernet_gs.yaml' for more details about the GS-HOTA metric and the evaluation configuration.") return combined_results