From eee625bef2eb2a9a0a13f9393488c3c51a169458 Mon Sep 17 00:00:00 2001
From: alec-groff <148134052+alec-groff@users.noreply.github.com>
Date: Fri, 26 Jan 2024 14:25:00 -0600
Subject: [PATCH] Update detector_config.py

---
 frigate/detectors/detector_config.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/frigate/detectors/detector_config.py b/frigate/detectors/detector_config.py
index ca1915449..3103e4aa8 100644
--- a/frigate/detectors/detector_config.py
+++ b/frigate/detectors/detector_config.py
@@ -51,6 +51,9 @@ class ModelConfig(BaseModel):
     model_type: ModelTypeEnum = Field(
         default=ModelTypeEnum.ssd, title="Object Detection Model Type"
     )
+    order: list[int] = Field(
+        default=[0,1,2,3], title="Order Output Tensors [0=boxes,1=scores,2=class_ids,3=count]"
+    )
     _merged_labelmap: Optional[Dict[int, str]] = PrivateAttr()
     _colormap: Dict[int, Tuple[int, int, int]] = PrivateAttr()
     _model_hash: str = PrivateAttr()