[Update] update ModelType to ssd

This commit is contained in:
GaryHuang-ASUS 2025-09-17 11:14:16 +08:00
parent ad3becfd3a
commit 7568c1da32

View File

@ -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):