PaddleOCR Docker CPU版一键部署运行

最新推荐文章于 2024-05-30 22:23:32 发布

無糖加冰

最新推荐文章于 2024-05-30 22:23:32 发布

阅读量688

点赞数 8

文章标签： docker 容器运维

本文链接：https://blog.csdn.net/qq_51756940/article/details/136619096

版权

文章描述了如何通过构建镜像部署Paddle_OCR模型，提供测试代码示例，包括初始化API、发送包含Base64编码图像的POST请求，以及解析OCR结果。

摘要由CSDN通过智能技术生成

构建了一个镜像，直接部署运行即可使用

docker run -dp 8866:8866  --name paddle_ocr nowindandmoon/paddle_ocr:latest

测试代码

import requests
import base64
import json


class InvoiceOcr:
    def __init__(self, url):
        self.api_url = url if url else "http://xxxxxx:8866/predict/ocr_system"

    def post(self, parameters):
        headers = {
            'Content-type': 'application/json',
            'Accept': 'application/json'
        }

        try:
            response = requests.post(self.api_url, data=json.dumps(parameters), headers=headers)
            if response.status_code == 200:
                return response.text
            else:
                print(f"Error: {response.status_code}")
                return None
        except requests.exceptions.RequestException as e:
            print(f"Request exception: {e}")
            return None


def image_to_base64(path):
    with open(path, "rb") as image_file:
        return base64.b64encode(image_file.read()).decode("utf-8")


if __name__ == "__main__":
    api_url = "http://xxxxxx:8866/predict/ocr_system"
    ocr = InvoiceOcr(api_url)

    # 替换为你的图片路径
    base64_image = image_to_base64("./test.jpg")

    parameters = {
        "images": [base64_image]
    }

    result = ocr.post(parameters)

    print(result)
    result = json.loads(result)

    for item in result['results'][0]:
        print(item['text'])

效果