Spaces:
Build error
Build error
Update PanopticQuality.py
Browse files- 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]),
|