bascobasculino commited on
Commit
98ec43a
1 Parent(s): b7a472f

Update PanopticQuality.py

Browse files
Files changed (1) hide show
  1. 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
- #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()}
 
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()}