franzi2505 commited on
Commit
6790ab3
1 Parent(s): 87a956d

add option to change method

Browse files
Files changed (1) hide show
  1. PanopticQuality.py +3 -3
PanopticQuality.py CHANGED
@@ -86,7 +86,7 @@ class PQMetric(evaluate.Metric):
86
  (0**2, 6**2),
87
  (6**2, 12**2),
88
  (12**2, 1e5**2)],
89
- method: Literal["iou", "hungarian"] = "iou",
90
  device: str = None,
91
  class_agnostic: bool = False,
92
  **kwargs
@@ -147,8 +147,8 @@ class PQMetric(evaluate.Metric):
147
  return_per_class=per_class,
148
  return_sq_and_rq=split_sq_rq,
149
  areas=area_rng,
150
- device=device
151
- #method=method
152
  )
153
  self.cont_to_cat = {label:key for key, label in self.pq_metric.metric.cat_id_to_continuous_id.items()}
154
 
 
86
  (0**2, 6**2),
87
  (6**2, 12**2),
88
  (12**2, 1e5**2)],
89
+ method: Literal["iou", "hungarian"] = "hungarian",
90
  device: str = None,
91
  class_agnostic: bool = False,
92
  **kwargs
 
147
  return_per_class=per_class,
148
  return_sq_and_rq=split_sq_rq,
149
  areas=area_rng,
150
+ device=device,
151
+ method=method
152
  )
153
  self.cont_to_cat = {label:key for key, label in self.pq_metric.metric.cat_id_to_continuous_id.items()}
154