Merge branch 'main' of https://huggingface.co/spaces/SEA-AI/PanopticQuality

PanopticQuality.py

@@ -87,6 +87,7 @@ class PQMetric(evaluate.Metric):
                                         (6**2, 12**2),
                                         (12**2, 1e5**2)],
         method: Literal["iou", "hungarian"] = "iou",
+        device: str = None,
         class_agnostic: bool = False,
         **kwargs
     ):
@@ -146,6 +147,7 @@ class PQMetric(evaluate.Metric):
             return_per_class=per_class,
             return_sq_and_rq=split_sq_rq,
             areas=area_rng,
+            device=device
             #method=method
         )
         self.cont_to_cat = {label:key for key, label in self.pq_metric.metric.cat_id_to_continuous_id.items()}

requirements.txt

@@ -1,4 +1,4 @@
 git+https://github.com/huggingface/evaluate@main
-git+https://github.com/SEA-AI/seametrics@develop
+git+https://github.com/SEA-AI/seametrics@feature/ps_unit_tests
 fiftyone
 torchmetrics