Provide model type in header for ZMQ detector (#20000)

* Provide model type in header

* Formatting
This commit is contained in:
Nicolas Mowen
2025-09-09 16:53:36 -06:00
committed by GitHub
parent fd6e7afea9
commit 15729e0f19

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")