mirror of
https://github.com/blakeblackshear/frigate.git
synced 2025-12-07 14:04:10 +03:00
[Update] update ModelType to ssd
This commit is contained in:
parent
ad3becfd3a
commit
7568c1da32
@ -53,7 +53,7 @@ class SynapDetector(DetectionApi):
|
||||
output_tensor_obj = self.network.predict()
|
||||
output = self.detector.process(output_tensor_obj, postprocess_data)
|
||||
|
||||
if self.model_type == ModelTypeEnum.yologeneric:
|
||||
if self.model_type == ModelTypeEnum.ssd:
|
||||
detections = np.zeros((20, 6), np.float32)
|
||||
|
||||
for i, item in enumerate(output.items):
|
||||
|
||||
Loading…
Reference in New Issue
Block a user