PaddleOCR 多线程版本(python)

    封装成http服务进行调用

import threading
from flask import Flask, request, jsonify
from paddleocr import PaddleOCR

app = Flask(__name__)

# Initialize 5 PaddleOCR instances and locks
ocr_instances = [PaddleOCR(enable_mkldnn=True, use_angle_cls=True, lang='en') for _ in range(5)]
locks = [threading.Lock() for _ in range(5)]


@app.route('/ocr', methods=['POST'])
def ocr_handler():
    if 'img' not in request.files:
        return jsonify({'error': 'No image file provided'})

    image_file = request.files['img']
    image_path = 'C:\\Users\\dukun\\Downloads\\ppocr_img\\ppocr_img\\imgs_en\\' + image_file.filename
    # 创建一个空白的图像文件
    with open(image_path, 'wb') as f:
        # 仅写入一个字节作为占位符
        f.write(b'\x00')
    # Specify the path to save the image
    image_file.save(image_path)
    # image_path = 'C:\\Users\\dukun\\AppData\\Local\\Temp\\16853451072921665941938794632449408042page_2.jpg'

    # Get the index of the OCR instance to use
    model_index = id(request._get_current_object()) % len(ocr_instances)
    ocr = ocr_instances[model_index]

    # Acquire the lock for the OCR instance
    lock = locks[model_index]
    lock.acquire()
    print("model_index", model_index)
    try:
        result = ocr.ocr(image_path, cls=True)

        # Process the OCR result
        ocr_result = []
        for res in result:
            for line in res:
                ocr_result.append(line)

        return jsonify({'result': ocr_result})
    finally:
        # Release the lock after OCR processing is complete
        lock.release()


if __name__ == '__main__':
    app.run()

  • 0
    点赞
  • 8
    收藏
    觉得还不错? 一键收藏
  • 3
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 3
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值