结果的save_to_img函数位置

[Linz1248]

请问ocr结果中的save_to_img函数源码在哪个位置呢

[liuhongen1234567]

您好，ocr的 save_to_img 函数实际上是通过 /root/anaconda3/envs/XXX/lib/python3.10/site-packages/paddlex/inference/pipelines/ocr/result.py 下的 _to_img 实现的，可以 pip show paddlex 查看一下paddlex wheel 包的具体位置

[zengjie617789]

那怎么导入呢， 百度的文档怎么就这么难找呢？

[liuhongen1234567]

您好，这个导入具体是指什么呢？这个函数不需要导入呀。直接res.save_to_img("XX") 就可以可视化到具体目录了。OCR文档的站点网站在这里 https://www.paddleocr.ai/latest/
OCR产线的使用文档链接如下： https://www.paddleocr.ai/latest/version3.x/pipeline_usage/OCR.html

[zengjie617789]

好的，谢谢，改成predict就可以了。

[zengjie617789]

使用res.save_to_img()的确很方便，会自动生成res_img.jpg和preprocessed_img.jpg，但是如果想自己绘制函数，如果直接使用results中的rec_boxes，会出现框对不齐的问题，应该是原图与输入图形进行了处理，如何将得到的box正确绘制在原图呢？

[liuhongen1234567]

您好，可以使用 doc_img = res["doc_preprocessor_res"]["output_img"]， poly = res["rec_polys"] 拿到预处理的图和框进行绘制，绘制逻辑可以参考 https://github.com/PaddlePaddle/PaddleX/blob/caa334a9b7b88ef3f711d16fea3177e6e812caf2/paddlex/inference/pipelines/ocr/result.py#L104C9-L104C59 。原图的框映射不回去，因为预处理中图像矫正是基于模型做的。