administrator/frigate
1
0
Fork 0
mirror of https://github.com/blakeblackshear/frigate.git synced 2026-05-04 04:27:42 +03:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

Don't import hailo until it is used

Browse Source
This commit is contained in:
Nicolas Mowen 2025-05-12 08:18:24 -06:00
parent a5c72db855
commit 3276a679f9
1 changed files with 18 additions and 24 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

42
frigate/detectors/plugins/hailo8l.py
View File

@ -8,17 +8,6 @@ from typing import Dict, List, Optional, Tuple
import cv2 import cv2
import numpy as np import numpy as np
try:
from hailo_platform import (
HEF,
FormatType,
HailoSchedulingAlgorithm,
VDevice,
)
except ModuleNotFoundError:
pass
from pydantic import Field from pydantic import Field
from typing_extensions import Literal from typing_extensions import Literal
@ -102,6 +91,16 @@ class HailoAsyncInference:
output_type: Optional[Dict[str, str]] = None, output_type: Optional[Dict[str, str]] = None,
send_original_frame: bool = False, send_original_frame: bool = False,
) -> None: ) -> None:
try:
from hailo_platform import (
HEF,
FormatType,
HailoSchedulingAlgorithm,
VDevice,
)
except ModuleNotFoundError:
pass
self.input_store = input_store self.input_store = input_store
self.output_store = output_store self.output_store = output_store
@ -112,24 +111,19 @@ class HailoAsyncInference:
self.target = VDevice(params) self.target = VDevice(params)
self.infer_model = self.target.create_infer_model(hef_path) self.infer_model = self.target.create_infer_model(hef_path)
self.infer_model.set_batch_size(batch_size) self.infer_model.set_batch_size(batch_size)
if input_type is not None: if input_type is not None:
self._set_input_type(input_type) self.infer_model.input().set_format_type(getattr(FormatType, input_type))
if output_type is not None: if output_type is not None:
self._set_output_type(output_type) for output_name, output_type in output_type.items():
self.infer_model.output(output_name).set_format_type(
getattr(FormatType, output_type)
)
self.output_type = output_type self.output_type = output_type
self.send_original_frame = send_original_frame self.send_original_frame = send_original_frame
def _set_input_type(self, input_type: Optional[str] = None) -> None:
self.infer_model.input().set_format_type(getattr(FormatType, input_type))
def _set_output_type(
self, output_type_dict: Optional[Dict[str, str]] = None
) -> None:
for output_name, output_type in output_type_dict.items():
self.infer_model.output(output_name).set_format_type(
getattr(FormatType, output_type)
)
def callback( def callback(
self, self,
completion_info, completion_info,