added pandera

dodgeballtracker/typing.py

@@ -1,6 +1,8 @@
 from typing import NewType
 from dataclasses import dataclass
 import numpy as np
+import pandera as pa
+from pandera.dtypes import Float32
 
 XCoord = NewType('XCoord', float)
 YCoord = NewType('YCoord', float)
@@ -33,5 +35,25 @@ class DetectedObject:
         return np.array([self.xmin, self.ymax, self.xmax, self.ymax, self.score, self.category])
 
 
+DetectedObject_ = pa.DataFrameSchema({
+    "xmin": pa.Column(Float32),
+    "ymin": pa.Column(Float32),
+    "xmax": pa.Column(Float32),
+    "ymin": pa.Column(Float32),
+    "score": pa.Column(Float32),
+    "category": pa.Column(Float32),
+})
+
+TrackedDetectedObject_ = pa.DataFrameSchema({
+    "xmin": pa.Column(Float32),
+    "ymin": pa.Column(Float32),
+    "xmax": pa.Column(Float32),
+    "ymin": pa.Column(Float32),
+    "object_id": pa.Column(Float32),
+    "score": pa.Column(Float32),
+    "category": pa.Column(Float32),
+    "index": pa.Column(Float32),
+})
+
 
 Color = tuple[int, int, int]