Spaces:
Runtime error
Runtime error
Update cocoevaluate.py
Browse files- cocoevaluate.py +5 -8
cocoevaluate.py
CHANGED
@@ -72,12 +72,9 @@ def summarize_if_long_list(obj):
|
|
72 |
@evaluate.utils.file_utils.add_start_docstrings(_DESCRIPTION, _KWARGS_DESCRIPTION)
|
73 |
class COCOEvaluate(evaluate.Metric):
|
74 |
"""TODO: Short description of my evaluation module."""
|
75 |
-
def __init__(self,
|
76 |
super().__init__(**kwargs)
|
77 |
-
|
78 |
-
self.coco_dataset = coco_dataset
|
79 |
-
base_ds = get_coco_api_from_dataset(self.coco_dataset)
|
80 |
-
self.coco_evaluator = CocoEvaluator(base_ds, iou_types)
|
81 |
|
82 |
|
83 |
def _info(self):
|
@@ -102,13 +99,13 @@ class COCOEvaluate(evaluate.Metric):
|
|
102 |
'references': [
|
103 |
datasets.Features(
|
104 |
{
|
105 |
-
'size': datasets.Sequence(datasets.Value("
|
106 |
'image_id': datasets.Sequence(datasets.Value("int64")),
|
107 |
'boxes': datasets.Sequence(datasets.Sequence(datasets.Value("float"))),
|
108 |
'class_labels': datasets.Sequence(datasets.Value("int64")),
|
109 |
'iscrowd': datasets.Sequence(datasets.Value("int64")),
|
110 |
-
'orig_size': datasets.Sequence(datasets.Value("
|
111 |
-
'area': datasets.Sequence(datasets.Value("
|
112 |
|
113 |
}
|
114 |
)
|
|
|
72 |
@evaluate.utils.file_utils.add_start_docstrings(_DESCRIPTION, _KWARGS_DESCRIPTION)
|
73 |
class COCOEvaluate(evaluate.Metric):
|
74 |
"""TODO: Short description of my evaluation module."""
|
75 |
+
def __init__(self, coco, iou_types=['bbox'], **kwargs):
|
76 |
super().__init__(**kwargs)
|
77 |
+
self.coco_evaluator = CocoEvaluator(coco, iou_types)
|
|
|
|
|
|
|
78 |
|
79 |
|
80 |
def _info(self):
|
|
|
99 |
'references': [
|
100 |
datasets.Features(
|
101 |
{
|
102 |
+
'size': datasets.Sequence(datasets.Value("float")),
|
103 |
'image_id': datasets.Sequence(datasets.Value("int64")),
|
104 |
'boxes': datasets.Sequence(datasets.Sequence(datasets.Value("float"))),
|
105 |
'class_labels': datasets.Sequence(datasets.Value("int64")),
|
106 |
'iscrowd': datasets.Sequence(datasets.Value("int64")),
|
107 |
+
'orig_size': datasets.Sequence(datasets.Value("float")),
|
108 |
+
'area': datasets.Sequence(datasets.Value("float")),
|
109 |
|
110 |
}
|
111 |
)
|