all-dressed-programming/dodgeball-tracker
1
0 Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Going somewhere with a coffee.

Browse Source
This commit is contained in:
didier committed 2024-11-15 10:00:22 -05:00
1 parent 84b9c2674c
commit f5b8e24e01
6 files changed
+191 -5
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

dodgeballtracker/typing.py
+2 -2
View File
@@ -10,11 +10,11 @@ def tensor_of(data_type: pa.DataType, cols: list[str]) -> pa.DataFrameSchema:
return pa.DataFrameSchema({col: pa.Column(data_type) for col in cols})
DetectedObject_ = tensor_of(Float32, ["xmin", "ymin", "xmax", "ymax", "score",
DetectedObject = tensor_of(Float32, ["xmin", "ymin", "xmax", "ymax", "score",
"category"])
TrackedDetectedObject = tensor_of(Float64,
list(DetectedObject_.columns) + ["object_id", "index"])
list(DetectedObject.columns) + ["object_id", "index"])
Color = tuple[int, int, int]