【个人开源】提供预训练结果的ocr标注工具

ocr标注工具

提供预训练模型的标注结果
采用python-flask框架开发图像标注工具,主要思路是在标注的过程中先调用百度/阿里等免费的api进行标注,并进行人工的验证或目标矩形框的修正,同时支持单张图片多个目标的标注。整理想法很简单,但是实现起来还是很麻烦的。大概用了2周的时间(而且是在开源项目上修改来的)。

原特点

  • B/S方式交互
  • 支持多人同时标注(可分配不同标注人员的标注范围,或不同人员标注不同类别)
  • 类别采用选择方式,免去手工输入类别工作
  • 支持拖拽方式修正标注区域
  • 支持键盘方向键切换标注样本

增加特性

  • 增加百度api/自用api调用的接口
  • 支持百度api检测结果的调整
  • 采用lmdb做数据库,用图片名做key进行保存
  • 支持单张图片,多目标的标注

使用方法

  1. 根据requirements.txt安装环境依赖,其中有些依赖可能没有使用,也可以装了相应python与flask版本后,根据报错安装其他依赖
$ cd od-annotation
$ pip3 install -r requirements.txt       # use pip
# or create a new conda enviroment
conda env create -f enviroment.yaml
  1. 将需要标注的图片放到:‘dataset/images/’

  2. 启动/停止/重启标注工具:

$ cd od-annotation
$ python3 app.py --start|stop|restart  # 前台进程方式运行
$ python3 app.py --start|restart --daemon  # 以后台进程方式(重新)启动
  1. 访问http://localhost:5000开始标注。整个标注流程是:1.先调用百度api; 2.将百度的结果画出来,并调整; 3.保存结果/下一张。其中可以用左键移动矩形框,并可以在矩形框右下角进行大小调整。右键点击矩形框可以删除当前标注的矩形框。当前样本标注状态会同步更新标注信息。 所有样本标注状态用的原作者的,没有调整其显示,可能会有错误,没有关系。

  2. 点击左右方向按钮或通过键盘方向键切换标注样本。切换时自动提交标注结果,或手动点击保存按钮提交标注结果。

  3. 标注文件在dataset/labels/中,是lmdb数据库。标注格式为:{'img_name': 'x1,y1,x2,y2,label\nx12,y12,x22,y22,label2\n'},标注的一个矩形,采用左上和右下的坐标

更改预训练API

按同样的返回格式,修改app.py中的第81行到102行

reference

borrowed code from od-annotation,thanks!

代码

github:https://github.com/chenjun2hao/ocr_annotation
CSDN:https://download.csdn.net/download/u011622208/12086826

发布了244 篇原创文章 · 获赞 147 · 访问量 28万+

猜你喜欢

转载自blog.csdn.net/u011622208/article/details/103873152