andsonder/document-scanner

基于PaddlePaddle实现的扫描全能王

Document Scanner

这是一个基于Paddle的扫描全能王，它的检测步骤主要分为两步：

1. 进行图像分割，提取出文本所在的位置
2. 根据分割结果提取四个角点并进行后处理

环境

安装环境

pip install -r requirements.txt

下载model

从release中下载模型，并放到项目根目录下，从下面的链接下载：

https://github.com/AndSonder/Document-Scanner/releases/tag/model

运行

运行demo.py, 修改图片路径后运行

# demo.py
from predict import *

result = predict_one_image('images/IMG_0817.JPG')
cv2.imwrite('./output.jpg', result)
result = cv2.cvtColor(result, cv2.COLOR_BGR2RGB)
show(result)

效果图