mirror of
https://github.com/blakeblackshear/frigate.git
synced 2025-12-06 05:24:11 +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] = {
|
||||
"shape": list(tensor_input.shape),
|
||||
"dtype": str(tensor_input.dtype.name),
|
||||
"model_type": str(self.detector_config.model.model_type.name),
|
||||
}
|
||||
return json.dumps(header).encode("utf-8")
|
||||
|
||||
|
||||
Loading…
Reference in New Issue
Block a user