diff --git a/demo_mnn/python/demo_mnn.py b/demo_mnn/python/demo_mnn.py index b10e58898..59528bf59 100644 --- a/demo_mnn/python/demo_mnn.py +++ b/demo_mnn/python/demo_mnn.py @@ -682,9 +682,9 @@ def infer_image(self, img_input): self.input_tensor.copyFrom(tmp_input) self.interpreter.runSession(self.session) score_out_name = [ - "cls_pred_stride_8", - "cls_pred_stride_16", - "cls_pred_stride_32", + "792", + "814", + "836", ] scores = [ self.interpreter.getSessionOutput(self.session, x).getData() @@ -692,9 +692,9 @@ def infer_image(self, img_input): ] scores = [np.reshape(x, (-1, 80)) for x in scores] boxes_out_name = [ - "dis_pred_stride_8", - "dis_pred_stride_16", - "dis_pred_stride_32", + "795", + "817", + "839", ] raw_boxes = [ self.interpreter.getSessionOutput(self.session, x).getData()