[fix] bug in no-cactus handling

sarhakor · fracape · commit b7d439e52483 · 2025-07-30T16:53:16.000+02:00
diff --git a/scripts/metrics/compute_overall_kmac_per_px.py b/scripts/metrics/compute_overall_kmac_per_px.py
@@ -248,7 +248,8 @@ def generate_csv_classwise_video_gmac(dataset_name, result_path, list_of_classwi
         }
 
         if args.dataset_name == "SFU" and args.no_cactus:
-            class_ab["CLASS-AB"].remove("Cactus_1920x1080_50")
+            if "Cactus_1920x1080_50" in class_ab["CLASS-AB"]:
+                class_ab["CLASS-AB"].remove("Cactus_1920x1080_50")
 
         output_df = generate_csv_classwise_video_gmac(
             args.dataset_name,

Original file line number	Diff line number	Diff line change
`@@ -248,7 +248,8 @@ def generate_csv_classwise_video_gmac(dataset_name, result_path, list_of_classwi`
`248`	`248`	`}`
`249`	`249`
`250`	`250`	`if args.dataset_name == "SFU" and args.no_cactus:`
`251`		`- class_ab["CLASS-AB"].remove("Cactus_1920x1080_50")`
	`251`	`+ if "Cactus_1920x1080_50" in class_ab["CLASS-AB"]:`
	`252`	`+ class_ab["CLASS-AB"].remove("Cactus_1920x1080_50")`
`252`	`253`
`253`	`254`	`output_df = generate_csv_classwise_video_gmac(`
`254`	`255`	`args.dataset_name,`