使用引擎推理时,如何写preprocess和postprocess
收藏
我是初学者,我在参考快速上手C推理 — Paddle-Inference documentation使用JAVA调用引擎推理PPOCRv2的det,cls,rec模型时,遇到了不知道该如何写preprocess和postprocess的困境,不知道有没有好心人能指点一下:
我在visualDL中观察到det的输入是float32[?,3,?,?],输出是float32[?,1,?,?],cls的输入是float32[?,3,?,?],输出是float32[?,2],rec的输入是float32[?,3,32,?],输出是float32[?,?,6625],我做了一个32X32的单字图片,preprocess都是将图像的红绿蓝拆分到数组中后调用PD_TensorCopyFromCpuFloat(input_tensor, input_data)传入,然后det,cls,rec也都通过PD_TensorCopyToCpuFloat(output_tensor, out_data);收到了返回,返回值的大小与模型接口定义相符。最后我就不知道该如何解析这个返回数组了。
0
收藏
请登录后评论
还是自己回答自己的问题吧。借鉴DJL的处理思路解决了这个问题。