yl-ocr-layout

水平框版面分析

基于 Official YOLOv7 和 PaddleYOLO，用于水平框版面分析。

环境

conda env create -f environment.yml

服务端

python run.py --port 8080

镜像打包

make all

模型配置

模型名称	模型配置
Official YOLOv7	./core/detectors/yolov7.py
PaddleYOLO YOLOv7P6-E6	./core/detectors/paddle_yolo/__init__.py

如果更新了模型权重，请同时修改创建镜像时的下载地址，如：

$ cat Dockerfile
...
# 模型权重下载地址
RUN wget ftp://192.168.199.31/yolov7p6_e6_300e/20240314/yolov7p6_e6_300e_coco.zip
...

模型训练

请参考：目标检测模型训练与评估

服务实现

版面分析服务的简化流程为：

图片 + 模型名称
    |
    | (根据模型名称，对图片进行相应的布局检测)
    v
布局检测框集

其中基于 PaddleYOLO 的布局检测实现中还包含图像预处理，模型预测，后处理等。源码：PaddleYOLO/deploy/python/infer.py#L62

注：PaddleYOLO 基于 PaddleDetection，预测部署的代码实现大部分是一致的。