We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent b29f6e0 commit 9849833Copy full SHA for 9849833
src/python/model_api/models/ssd.py
@@ -87,7 +87,12 @@ def __call__(self, outputs) -> DetectionResult:
87
bboxes = np.array(outputs[self.bboxes_layer][0])
88
scores = np.array(outputs[self.scores_layer][0])
89
labels = np.array(outputs[self.labels_layer][0])
90
- return DetectionResult(bboxes, scores, labels)
+
91
+ return DetectionResult(
92
+ bboxes=bboxes,
93
+ labels=labels,
94
+ scores=scores,
95
+ )
96
97
98
class BoxesLabelsParser:
0 commit comments