Update cocoevaluate.py

cocoevaluate.py

@@ -72,12 +72,9 @@ def summarize_if_long_list(obj):
 @evaluate.utils.file_utils.add_start_docstrings(_DESCRIPTION, _KWARGS_DESCRIPTION)
 class COCOEvaluate(evaluate.Metric):
     """TODO: Short description of my evaluation module."""
-    def __init__(self, coco_dataset, iou_types=['bbox'], **kwargs):
+    def __init__(self, coco, iou_types=['bbox'], **kwargs):
         super().__init__(**kwargs)
-        # self.coco_dataset = CocoDetection(coco_path, feature_extractor, annotation_path)
-        self.coco_dataset = coco_dataset
-        base_ds = get_coco_api_from_dataset(self.coco_dataset)
-        self.coco_evaluator = CocoEvaluator(base_ds, iou_types)
+        self.coco_evaluator = CocoEvaluator(coco, iou_types)
 
 
     def _info(self):
@@ -102,13 +99,13 @@ class COCOEvaluate(evaluate.Metric):
                 'references': [
                         datasets.Features(
                         {
-                            'size': datasets.Sequence(datasets.Value("int64")),
+                            'size': datasets.Sequence(datasets.Value("float")),
                             'image_id': datasets.Sequence(datasets.Value("int64")),
                             'boxes': datasets.Sequence(datasets.Sequence(datasets.Value("float"))),
                             'class_labels': datasets.Sequence(datasets.Value("int64")),
                             'iscrowd': datasets.Sequence(datasets.Value("int64")),
-                            'orig_size': datasets.Sequence(datasets.Value("int64")),
-                            'area': datasets.Sequence(datasets.Value("int64")),
+                            'orig_size': datasets.Sequence(datasets.Value("float")),
+                            'area': datasets.Sequence(datasets.Value("float")),
                             
                         }
                     )