From 4e0d65d45bc032845d7648f15b0a12a1de32e28b Mon Sep 17 00:00:00 2001 From: Sepehr Sameni Date: Fri, 9 Aug 2024 16:11:13 +0200 Subject: [PATCH] fix a small typo in evaluation.py --- evaluation.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/evaluation.py b/evaluation.py index f556f08..03a4eab 100644 --- a/evaluation.py +++ b/evaluation.py @@ -248,7 +248,7 @@ def evaluate_propagation(PredsEval: PredsmIoU, gts:torch.Tensor, preds: torch.Te def evaluate_localizations(PredsEval, gts, preds, evaluation_protocol, logging_directory, many_to_one=False, precision_based=False): ## gets tensors with the [bs, fs, input_resolution, input_resolution] and does the evaluation - ## visualization of the heatmaps is very ugly. It should be changed some how. + ## visualization of the heatmaps is very ugly. It should be changed somehow. bs, fs, h, w = preds.shape scores = []