Spaces:
Sleeping
Sleeping
bascobasculino
commited on
Commit
•
98ec43a
1
Parent(s):
b7a472f
Update PanopticQuality.py
Browse files- PanopticQuality.py +1 -1
PanopticQuality.py
CHANGED
@@ -148,7 +148,7 @@ class PQMetric(evaluate.Metric):
|
|
148 |
return_per_class=per_class,
|
149 |
return_sq_and_rq=split_sq_rq,
|
150 |
areas=area_rng,
|
151 |
-
|
152 |
#method=method
|
153 |
)
|
154 |
self.cont_to_cat = {label:key for key, label in self.pq_metric.metric.cat_id_to_continuous_id.items()}
|
|
|
148 |
return_per_class=per_class,
|
149 |
return_sq_and_rq=split_sq_rq,
|
150 |
areas=area_rng,
|
151 |
+
device=device
|
152 |
#method=method
|
153 |
)
|
154 |
self.cont_to_cat = {label:key for key, label in self.pq_metric.metric.cat_id_to_continuous_id.items()}
|