a

my_metricv2.py

@@ -73,10 +73,27 @@ class MyMetricv2(evaluate.Metric):
             # This defines the format of each prediction and reference
             features=datasets.Features({
                 "predictions": [
-                    [datasets.Value("float32"), datasets.Value("float32"), datasets.Value("float32"), datasets.Value("float32")]
+                    datasets.Features(
+                        {
+                            "scores": datasets.Sequence(datasets.Value("float")),
+                            "ids": datasets.Sequence(datasets.Value("int64")),
+                            "frames": datasets.Sequence(datasets.Value("int64")),
+                            "boxes": datasets.Sequence(
+                                datasets.Sequence(datasets.Value("float"))
+                            ),
+                        }
+                    )
                 ],
                 "references": [
-                    [datasets.Value("float32"), datasets.Value("float32"), datasets.Value("float32"), datasets.Value("float32")]
+                    datasets.Features(
+                        {
+                            "ids": datasets.Sequence(datasets.Value("int64")),
+                            "frames": datasets.Sequence(datasets.Value("int64")),
+                            "boxes": datasets.Sequence(
+                                datasets.Sequence(datasets.Value("float"))
+                            ),
+                        }
+                    )
                 ],
             }),
             # Homepage of the module for documentation
@@ -98,11 +115,11 @@ class MyMetricv2(evaluate.Metric):
         # <frame number>, <object id>, <bb_left>, <bb_top>, <bb_width>, <bb_height>, <confidence>, <x>, <y>, <z>
         # predictions array de formato []
 
-       #  accuracy = sum(i == j for i, j in zip(predictions, references)) / len(predictions)
+        # accuracy = sum(i == j for i, j in zip(predictions, references)) / len(predictions)
 
 
         return {
-           
+        
             "predictions": predictions,
             "references": references