mirror of
https://github.com/blakeblackshear/frigate.git
synced 2025-09-26 19:41:29 +08:00
Provide model type in header for ZMQ detector (#20000)
* Provide model type in header * Formatting
This commit is contained in:
@@ -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")
|
||||||
|
|
||||||
|
Reference in New Issue
Block a user