mirror of
https://github.com/blakeblackshear/frigate.git
synced 2025-12-06 13:34:13 +03:00
Provide model type in header for ZMQ detector (#20000)
* Provide model type in header * Formatting
This commit is contained in:
parent
fd6e7afea9
commit
15729e0f19
@ -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")
|
||||||
|
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user