Spaces:
Build error
Build error
franziska-denk
commited on
Commit
·
6790ab3
1
Parent(s):
87a956d
add option to change method
Browse files- 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"] = "
|
| 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 |
-
|
| 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 |
|