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

Update PanopticQuality.py

Browse files
Files changed (1) hide show
  1. PanopticQuality.py +0 -1
PanopticQuality.py CHANGED
@@ -141,7 +141,6 @@ class PQMetric(evaluate.Metric):
141
  self.stuff = stuff if stuff is not None else DEFAULT_STUFF
142
  self.per_class = per_class
143
  self.split_sq_rq = split_sq_rq
144
- print("HF: DEVICE -> ", device)
145
  self.pq_metric = PanopticQuality(
146
  things=set([self.label2id[label] for label in self.label2id.keys() if label not in self.stuff]),
147
  stuffs=set([self.label2id[label] for label in self.label2id.keys() if label in self.stuff]),
 
141
  self.stuff = stuff if stuff is not None else DEFAULT_STUFF
142
  self.per_class = per_class
143
  self.split_sq_rq = split_sq_rq
 
144
  self.pq_metric = PanopticQuality(
145
  things=set([self.label2id[label] for label in self.label2id.keys() if label not in self.stuff]),
146
  stuffs=set([self.label2id[label] for label in self.label2id.keys() if label in self.stuff]),