franzi2505 commited on
Commit
2b24bc8
1 Parent(s): 9192ecd
Files changed (1) hide show
  1. 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