mirror of
https://github.com/blakeblackshear/frigate.git
synced 2025-12-09 14:55:43 +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_tensor_obj = self.network.predict()
|
||||||
output = self.detector.process(output_tensor_obj, postprocess_data)
|
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)
|
detections = np.zeros((20, 6), np.float32)
|
||||||
|
|
||||||
for i, item in enumerate(output.items):
|
for i, item in enumerate(output.items):
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user