mirror of
https://github.com/openmv/openmv.git
synced 2025-11-04 14:49:50 +08:00
scripts/libraries: Simplify blazeface/palm list output.
This commit is contained in:
parent
daf26f9fc8
commit
75e16b5738
@ -78,7 +78,7 @@ class mediapipe_detection_postprocess:
|
|||||||
self.detection_post_process(ih, iw, nms, model, inputs, outputs, self.scores[1], self.cords[1],
|
self.detection_post_process(ih, iw, nms, model, inputs, outputs, self.scores[1], self.cords[1],
|
||||||
self.threshold, self.anchors[output_len:])
|
self.threshold, self.anchors[output_len:])
|
||||||
|
|
||||||
return nms.get_bounding_boxes(threshold=self.nms_threshold, sigma=self.nms_sigma)
|
return nms.get_bounding_boxes(threshold=self.nms_threshold, sigma=self.nms_sigma)[0]
|
||||||
|
|
||||||
def detection_post_process(self, ih, iw, nms, model, inputs, outputs, score_idx, cords_idx, t, anchors):
|
def detection_post_process(self, ih, iw, nms, model, inputs, outputs, score_idx, cords_idx, t, anchors):
|
||||||
s_oh, s_ow, s_oc = model.output_shape[score_idx]
|
s_oh, s_ow, s_oc = model.output_shape[score_idx]
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user