Going somewhere with a coffee

This commit is contained in:
didier committed 2024-11-16 09:16:25 -05:00
1 parent 236dd82871
commit a10fdbee50
7 files changed
+247 -143

No files matched your search

+1 -2
View File
@@ -10,8 +10,7 @@ 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",
"category"])
DetectedObject = tensor_of(Float32, ["xmin", "ymin", "xmax", "ymax", "score", "category"])
TrackedDetectedObject = tensor_of(Float64,
list(DetectedObject.columns) + ["object_id", "index"])