diff --git a/tracklab/wrappers/eval/trackeval_evaluator.py b/tracklab/wrappers/eval/trackeval_evaluator.py index 445ea461..e8aeb3b4 100644 --- a/tracklab/wrappers/eval/trackeval_evaluator.py +++ b/tracklab/wrappers/eval/trackeval_evaluator.py @@ -19,7 +19,7 @@ def __init__(self, cfg, eval_set, show_progressbar, dataset_path, tracking_datas self.cfg = cfg self.tracking_dataset = tracking_dataset self.eval_set = eval_set - self.trackeval_dataset_name = type(self.tracking_dataset).__name__ + self.trackeval_dataset_name = cfg.dataset.dataset_class self.trackeval_dataset_class = getattr(trackeval.datasets, cfg.dataset.dataset_class) self.show_progressbar = show_progressbar self.dataset_path = dataset_path