mmocr初体验

目录

1. 安装mmocr

1.1 首先安装pytorch

1.2 安装mmcv

1.3 安装mmdetection

1.4 安装mmocr

 2. 训练

2.1 准备数据

2.2 开启训练

2.3 测试

---

1. 安装mmocr

官方依赖：pytorch, mmcv，mmdetection.

1.1 首先安装pytorch

这里默认已经安装好了cuda和cudnn

pip install torch==1.8.0+cu111 torchvision==0.9.0+cu111 torchaudio==0.8.0 -f https://download.pytorch.org/whl/torch_stable.html

1.2 安装mmcv

mmcv安装记录_Mr.Q的博客-CSDN博客_mmcv安装由于官方提供的win版本有限，需要自己源码编译mmcv，特此记录。1 准备提前下载好mmcv-master.zip或者gi clone下来：GitHub - open-mmlab/mmcv: OpenMMLab Computer Vision FoundationOpenMMLab Computer Vision Foundation. Contribute to open-mmlab/mmcv development by creating an account on GitHub.htthttps://blog.csdn.net/jizhidexiaoming/article/details/121402107

1.3 安装mmdetection

git clone https://github.com/open-mmlab/mmdetection.git
cd mmdetection
pip install -r requirements/build.txt
pip install -v -e .  # or "python setup.py develop"

 注意上面的所有操作都要在同一个conda环境中操作，我的conda环境是open-mmlab.

成功安装好前面的三个依赖：

 1.4 安装mmocr

git clone https://github.com/open-mmlab/mmocr.git
cd mmocr
pip3 install -e .

成功安装mmocr

 2. 训练

这里以收据图片的OCR-KIE，即关键信息提取为例。

2.1 准备数据

 Key Information Extraction — MMOCR 0.5.0 documentationhttps://mmocr.readthedocs.io/en/latest/datasets/kie.html

整理好的文件夹： 

 

 2.2 开启训练

修改configs/kie/sdmgr/sdmgr_unet16_60e_wildreceipt.py文件内容，根据自己的需求调整，也可以只修改data_root即可。

 然后命令行输入命令，开启训练：

python tools/train.py configs/kie/sdmgr/sdmgr_unet16_60e_wildreceipt.py

 2.3 测试

python tools/kie_test_imgs.py configs/kie/sdmgr/sdmgr_unet16_60e_wildreceipt.py work_dirs/sdmgr_unet16_60e_wildreceipt/sdmgr_unet16_60e_wildreceipt_20210520-7489e6de.pth --show