53 lines
1.6 KiB
Python
53 lines
1.6 KiB
Python
import os
|
||
import datetime
|
||
|
||
from paddleocr import PPStructureV3
|
||
|
||
# 安装OCR引擎
|
||
# python -m pip install paddlepaddle paddleocr
|
||
|
||
# 如果使用GPU，请安装GPU版本
|
||
# (1) 卸载CPU版本
|
||
# pip uninstall paddlepaddle
|
||
# pip uninstall paddleocr
|
||
|
||
# (2) 安装GPU版本 【没有成功】
|
||
# pip install paddlepaddle-gpu
|
||
# pip install paddleocr
|
||
from Pptx.Config import ocr_output_dir, markdown_output_dir, time_format
|
||
|
||
# 确保输出目录存在
|
||
os.makedirs(ocr_output_dir, exist_ok=True)
|
||
os.makedirs(markdown_output_dir, exist_ok=True)
|
||
|
||
# 初始化OCR引擎
|
||
pipeline = PPStructureV3()
|
||
|
||
# 图片位置
|
||
img_path = r"D:\logs\img"
|
||
|
||
# 先处理下中文文件名的问题，方法是以_分割，取最后一段即可
|
||
for img_file in os.listdir(img_path):
|
||
if len(img_file.split("_")) > 0:
|
||
os.rename(os.path.join(img_path, img_file), os.path.join(img_path, img_file.split("_")[-1]))
|
||
|
||
# 遍历 image_files目录下所有的文件
|
||
for img_file in os.listdir(img_path):
|
||
# 如果不是图片文件，则跳过
|
||
if not img_file.endswith(".png") and not img_file.endswith(".jpg") and not img_file.endswith(".jpeg"):
|
||
continue
|
||
|
||
# 构建完整的文件路径
|
||
full_path = os.path.join(img_path, img_file)
|
||
|
||
# 输出处理信息
|
||
print(f"{datetime.datetime.now().strftime(time_format)} 正在处理{img_file}")
|
||
|
||
try:
|
||
# 执行OCR识别
|
||
output = pipeline.predict(full_path)
|
||
for res in output:
|
||
res.save_to_markdown(save_path=markdown_output_dir)
|
||
except Exception as e:
|
||
print(f"处理图片{img_file}时出错: {str(e)}")
|