Spaces:
Sleeping
Sleeping
franzi2505
commited on
Commit
•
2b24bc8
1
Parent(s):
9192ecd
minor
Browse files- PanopticQuality.py +2 -2
PanopticQuality.py
CHANGED
@@ -189,8 +189,8 @@ class PQMetric(evaluate.Metric):
|
|
189 |
for i, numeric_label in enumerate(things_stuffs)}
|
190 |
result_dict["numbers"].update({"overall": [tp.sum().item(), fp.sum().item(), fn.sum().item(), iou.sum().item()]})
|
191 |
else:
|
192 |
-
result_dict["scores"] = result.tolist() if self.split_sq_rq else [result.tolist()]
|
193 |
-
result_dict["numbers"] = [tp.sum().item(), fp.sum().item(), fn.sum().item(), iou.sum().item()]
|
194 |
|
195 |
return result_dict
|
196 |
|
|
|
189 |
for i, numeric_label in enumerate(things_stuffs)}
|
190 |
result_dict["numbers"].update({"overall": [tp.sum().item(), fp.sum().item(), fn.sum().item(), iou.sum().item()]})
|
191 |
else:
|
192 |
+
result_dict["scores"] = {"overall": result.tolist() if self.split_sq_rq else [result.tolist()]}
|
193 |
+
result_dict["numbers"] = {"overall": [tp.sum().item(), fp.sum().item(), fn.sum().item(), iou.sum().item()]}
|
194 |
|
195 |
return result_dict
|
196 |
|