📊RapidUnWrap

English | 简体中文

最近更新

• 2024.11.15
  • 完成初版代码，转换 UVDoc 模型为onnx,完善前后处理

简介

本仓库用于进行文档扭曲的修正，同时能一定程度缓解透视和旋转问题

在线体验

modelscope huggingface

效果展示

安装

# 建议使用清华源安装 https://pypi.tuna.tsinghua.edu.cn/simple
pip install rapid-unwrap

快速使用

import cv2

from rapid_unwrap.inference import DocUnwrapper
img_path = "img/demo4.jpg"
doc_unwrapper = DocUnwrapper(img_path)
unwrapped_img, elapse = doc_unwrapper(img_path)
print(f"doc unwrap elapse:{elapse}")
cv2.imwrite("unwarped.png", unwrapped_img)

致谢

UVDoc

贡献指南

欢迎提交请求。对于重大更改，请先打开issue讨论您想要改变的内容。

有其他的好建议和集成场景，作者也会积极响应支持

开源许可证

该项目采用Apache 2.0 开源许可证。