Skip to content

Commit 20bdeb2

Browse files
committed
fix: scripts, import for mota computation
1 parent 8e89795 commit 20bdeb2

File tree

1 file changed

+4
-3
lines changed

1 file changed

+4
-3
lines changed

scripts/metrics/compute_overall_miou.py

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -40,12 +40,11 @@
4040
import csv
4141
import json
4242

43+
import compute_overall_mot
4344
import utils
4445

4546
from compressai_vision.evaluators.evaluators import BaseEvaluator
4647

47-
from .compute_overall_mota import compute_overall_mota
48-
4948
CLASSES = ["PANDAM1", "PANDAM2", "PANDAM2"]
5049

5150
SEQS_BY_CLASS = {
@@ -166,7 +165,9 @@ def compute_overall_mIoU(class_name, items):
166165
len(items) > 0
167166
), "Nothing relevant information found from given directories..."
168167

169-
summary, names = compute_overall_mota(args.class_to_compute, items)
168+
summary, names = compute_overall_mot.compute_overall_mota(
169+
args.class_to_compute, items
170+
)
170171

171172
motas = [100.0 * sv[13] for sv in summary.values]
172173

0 commit comments

Comments
 (0)