Skip to content

ShuiyuanSJTU/discourse-image-enhancement-service

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

3 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Discourse Image Enhancement Service

简体中文 | English

Discourse 图片增强插件 提供后端 AI 推理服务。

安装

  1. 安装 Python 3.10 及以上版本。
  2. 安装 pyTorch
  3. 安装 PaddlePaddle
  4. 安装其他依赖:
    pip install -r requirements.txt

运行

  • 直接运行(不推荐,仅用于调试):

    python app.py

    初次运行后自动生成配置文件config.json,请根据实际情况修改配置。

  • 使用 uvicorn

    uvicorn app:app --port 8000

    由于 paddleOCR 已知有内存泄漏问题,不建议采用该方法部署到生产环境。

  • 使用 gunicorn

    gunicorn -w 2 -k uvicorn.workers.UvicornWorker --max-requests 1000 --timeout 120 --bind 0.0.0.0:8000 app:app

API

访问/docs以查看 API 文档。

GET /health

检查服务是否正常运行。如果服务正常运行,返回{"status": "ok"}

POST /analyze

分析图片的 Endpoint,需要将此 Endpoint 完整 URL 填入Discourse 图片增强插件image enhancement analyze service endpoint设置。

请求参数 (application/json):

{
  "images": [
    {
      "sha1": "string",
      "url": "string",
      "id": 0
    }
  ],
  "lang": "string"
}

响应结果 (application/json):

{
  "images": [
    {
      "sha1": "string",
      "success": true,
      "error": "string",
      "ocr_result": [
        "string"
      ],
      "description": "string"
    }
  ],
  "message": ""
}

About

AI inference services for the Discourse Image Enhancement plugin.

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages