frigate/detectors/plugins/memryx.py aktualisiert
Some checks failed
CI / AMD64 Build (push) Has been cancelled
CI / ARM Build (push) Has been cancelled
CI / Jetson Jetpack 5 (push) Has been cancelled
CI / Jetson Jetpack 6 (push) Has been cancelled
CI / AMD64 Extra Build (push) Has been cancelled
CI / ARM Extra Build (push) Has been cancelled
CI / Assemble and push default build (push) Has been cancelled
Some checks failed
CI / AMD64 Build (push) Has been cancelled
CI / ARM Build (push) Has been cancelled
CI / Jetson Jetpack 5 (push) Has been cancelled
CI / Jetson Jetpack 6 (push) Has been cancelled
CI / AMD64 Extra Build (push) Has been cancelled
CI / ARM Extra Build (push) Has been cancelled
CI / Assemble and push default build (push) Has been cancelled
This commit is contained in:
@@ -25,6 +25,9 @@ logger = logging.getLogger(__name__)
|
||||
|
||||
DETECTOR_KEY = "memryx"
|
||||
|
||||
# Check for env variable
|
||||
mxserver_addr = os.getenv('MXA_MANAGER_ADDRESS', 'gateway.docker.internal')
|
||||
|
||||
# Configuration class for model settings
|
||||
class ModelConfig(BaseModel):
|
||||
path: str = Field(default=None, title="Model Path") # Path to the DFP file
|
||||
@@ -129,7 +132,7 @@ class MemryXDetector(DetectionApi):
|
||||
# Load MemryX Model with a unique device target
|
||||
self.accl = AsyncAccl(
|
||||
self.memx_model_path,
|
||||
mxserver_addr="gateway.docker.internal",
|
||||
mxserver_addr=mxserver_addr,
|
||||
group_id=device_id, # AsyncAccl device id
|
||||
)
|
||||
|
||||
|
Reference in New Issue
Block a user