Spaces:
Build error
Build error
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()}
|