Spaces:
Sleeping
Sleeping
franzi2505
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 |
|