运用unet实现对文档表格的自动检测,表格重建
实现功能
- [x] 支持GPU,CPU(opencv dnn加速);
- [ ] 整合darknet-ocr完成对表格的重建,输出json\excel
编译对GPU的支持
git clone https://github.com/pjreddie/darknet.git ../darknet cp Makefile ../darknet cd ../darknet && make
下载text.weights模型文件
模型文件地址: http://gofile.me/4Nlqh/fNHlWzVWo
拷贝table.weights文件到models目录
训练(tensorflow2.0版本训练)
https://github.com/chineseocr/table-detect.git
测试
python3 table.py -jpgPath test/dd.jpg
识别结果展示
横线竖线检测
单元格输出
参考
- darket https://github.com/pjreddie/darknet.git
- darknet-ocr https://github.com/chineseocr/darknet-ocr.git
- chineseocr https://github.com/chineseocr/chineseocr.git