Spaces:
Sleeping
Sleeping
bascobasculino
commited on
Commit
•
fc34022
1
Parent(s):
98ec43a
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]),
|