Provide model type in header

This commit is contained in:
Nicolas Mowen 2025-09-09 16:30:12 -06:00
parent fd6e7afea9
commit 240f998121

View File

@ -82,6 +82,7 @@ class ZmqIpcDetector(DetectionApi):
header: dict[str, Any] = { header: dict[str, Any] = {
"shape": list(tensor_input.shape), "shape": list(tensor_input.shape),
"dtype": str(tensor_input.dtype.name), "dtype": str(tensor_input.dtype.name),
"model_type": str(self.detector_config.model.model_type.name)
} }
return json.dumps(header).encode("utf-8") return json.dumps(header).encode("utf-8")