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.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):
|
||||
s_oh, s_ow, s_oc = model.output_shape[score_idx]
|
||||
|
||||
Loading…
Reference in New Issue
Block a user