Skip to content

Commit 37ccd80

Browse files
sarhakorfracape
authored andcommitted
[fix] add no-scale option for class-wise mac calculation
1 parent 962aae3 commit 37ccd80

1 file changed

Lines changed: 19 additions & 1 deletion

File tree

scripts/metrics/compute_overall_kmac_per_px.py

Lines changed: 19 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -222,6 +222,13 @@ def generate_csv_classwise_video_gmac(dataset_name, result_path, list_of_classwi
222222
help="exclude Cactus sequence for FCM eval",
223223
)
224224

225+
parser.add_argument(
226+
"--add-non-scale",
227+
action="store_true",
228+
default=False,
229+
help="Add non-scale option using ns_Traffic/ns_BQTerrace with original GT",
230+
)
231+
225232
args = parser.parse_args()
226233

227234
assert args.dataset_name.lower() in Path(args.result_path).name.lower()
@@ -253,10 +260,21 @@ def generate_csv_classwise_video_gmac(dataset_name, result_path, list_of_classwi
253260
if "Cactus_1920x1080_50" in class_ab["CLASS-AB"]:
254261
class_ab["CLASS-AB"].remove("Cactus_1920x1080_50")
255262

263+
sfu_classes = [class_ab, class_c, class_d]
264+
265+
if args.dataset_name == "SFU" and args.add_non_scale:
266+
class_ab_star = {
267+
"CLASS-AB*": [
268+
"ns_Traffic",
269+
"ns_BQTerrace",
270+
]
271+
}
272+
sfu_classes.append(class_ab_star)
273+
256274
output_df = generate_csv_classwise_video_gmac(
257275
args.dataset_name,
258276
args.result_path,
259-
[class_ab, class_c, class_d],
277+
sfu_classes,
260278
)
261279
elif args.dataset_name == "OIV6":
262280
oiv6 = ["detection", "segmentation"]

0 commit comments

Comments
 (0)