bascobasculino commited on
Commit
b7a472f
·
verified ·
1 Parent(s): d37ed60

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()}