Skip to content

Commit 992ff13

Browse files
committed
[fix] bug - miou
1 parent 9f4733a commit 992ff13

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

scripts/metrics/gen_mpeg_cttc_csv.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -276,11 +276,12 @@ def generate_csv_classwise_video_miou(
276276
dataset_path,
277277
dict_of_class_seq,
278278
nb_operation_points: int = 4,
279+
dataset_prefix: str = None,
279280
):
280281
seq_list = []
281282
[seq_list.extend(sequences) for sequences in dict_of_class_seq.values()]
282283

283-
results_df = read_df_rec(result_path, "", seq_list, nb_operation_points)
284+
results_df = read_df_rec(result_path, dataset_prefix, seq_list, nb_operation_points)
284285

285286
# sort
286287
sorterIndex = dict(zip(seq_list, range(len(seq_list))))
@@ -309,7 +310,7 @@ def generate_csv_classwise_video_miou(
309310
len(items) > 0
310311
), "Nothing relevant information found from given directories..."
311312

312-
miou = compute_per_class_mIoU(class_name, items)
313+
miou = compute_per_class_mIoU(items)
313314
class_wise_mious.append(miou)
314315

315316
matched_seq_names = []

0 commit comments

Comments
 (0)