Skip to content

Commit 9849833

Browse files
committed
Fix multiple output ssd parser
1 parent b29f6e0 commit 9849833

File tree

1 file changed

+6
-1
lines changed
  • src/python/model_api/models

1 file changed

+6
-1
lines changed

src/python/model_api/models/ssd.py

+6-1
Original file line numberDiff line numberDiff line change
@@ -87,7 +87,12 @@ def __call__(self, outputs) -> DetectionResult:
8787
bboxes = np.array(outputs[self.bboxes_layer][0])
8888
scores = np.array(outputs[self.scores_layer][0])
8989
labels = np.array(outputs[self.labels_layer][0])
90-
return DetectionResult(bboxes, scores, labels)
90+
91+
return DetectionResult(
92+
bboxes=bboxes,
93+
labels=labels,
94+
scores=scores,
95+
)
9196

9297

9398
class BoxesLabelsParser:

0 commit comments

Comments
 (0)