Skip to content

Commit 7e6a15f

Browse files
authored
Update utils.py
Test
1 parent 038a8ce commit 7e6a15f

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

src/patchcore/utils.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -134,16 +134,18 @@ def compute_and_store_final_results(
134134
"Full PRO",
135135
"Anomaly Pixel AUROC",
136136
"Anomaly PRO",
137+
"Image AP",
138+
"Pixel AP"
137139
],
138140
):
139141
"""Store computed results as CSV file.
140142
141143
Args:
142144
results_path: [str] Where to store result csv.
143145
results: [List[List]] List of lists containing results per dataset,
144-
with results[i][0] == 'dataset_name' and results[i][1:6] =
146+
with results[i][0] == 'dataset_name' and results[i][1:8] =
145147
[instance_auroc, full_pixelwisew_auroc, full_pro,
146-
anomaly-only_pw_auroc, anomaly-only_pro]
148+
anomaly-only_pw_auroc, anomaly-only_pro, image_ap, pixel_ap]
147149
"""
148150
if row_names is not None:
149151
assert len(row_names) == len(results), "#Rownames != #Result-rows."

0 commit comments

Comments
 (0)