PaddleOCR中PP-Structure使用例程img_show.save保存路径问题

最新推荐文章于 2024-10-01 16:02:57 发布

xiewk2015

最新推荐文章于 2024-10-01 16:02:57 发布

阅读量1.2k

点赞数

分类专栏：深度学习 OCR 文章标签： python ocr 深度学习

本文链接：https://blog.youkuaiyun.com/xie_wenkai/article/details/121430639

版权

深度学习同时被 2 个专栏收录

5 篇文章

订阅专栏

OCR

3 篇文章

订阅专栏

这篇博客介绍了如何利用PaddleOCR的PPStructure模块进行表格结构识别。首先，对图像进行放大和锐化处理，然后用DB检测算法进行表格检测，最后将结果保存为jpg图像并打印关键信息。代码示例中还包括了图像处理步骤和结果展示。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

# 网上代码im_show.save(MapPath("./myImage")+"\\a.gif", ImageFormat.Gif)
im_show.save(os.path.join(save_folder, os.path.basename(img_path).split('.')[0])+"\\result.jpg")

import os
import cv2
from paddleocr import PPStructure, draw_structure_result, save_structure_res

# 自己添加**********************************
import numpy as np

# import warnings
# warnings.filterwarnings("ignore")

# 自己添加************************************

table_engine = PPStructure(det_algorithm='DB', show_log=False, use_gpu=False, use_angle_cls=False)

save_folder = 'F:\\PaddleOCR-release-2.2\\output\\table'
img_path = 'F:\\PaddleOCR-release-2.2\\doc\\table\\1.png'
img = cv2.imread(img_path)

# 自己添加**********************************
# 放大2倍
img = cv2.resize(img, None, fx=1, fy=1, interpolation=cv2.INTER_CUBIC)
# 图像锐化
kernel = np.array([[0, -1, 0], [-1, 5, -1], [0, -1, 0]], np.float32)
img = cv2.filter2D(img, -1, kernel=kernel)
# 转回为灰度图片
img = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
# 自己添加************************************


result = table_engine(img)
save_structure_res(result, save_folder, os.path.basename(img_path).split('.')[0])

for line in result:
    line.pop('img')
    print(line)

from PIL import Image
font_path = 'F:\\PaddleOCR-release-2.2\\doc\\fonts\\simfang.ttf'  # PaddleOCR下提供字体包
image = Image.open(img_path).convert('RGB')
im_show = draw_structure_result(image, result, font_path=font_path)
im_show = Image.fromarray(im_show)

# 网上代码im_show.save(MapPath("./myImage")+"\\a.gif", ImageFormat.Gif)
im_show.save(os.path.join(save_folder, os.path.basename(img_path).split('.')[0])+"\\result.jpg")