PaddleOCR/doc/doc_ch/dataset/ocr_datasets.md

53 lines
2.9 KiB
Markdown
Raw Normal View History

2022-04-26 21:54:06 +08:00
## OCR数据集
- [1. 文本检测](#1)
2022-04-27 14:04:18 +08:00
- [1.1 ICDAR 2015](#11)
2022-04-26 21:54:06 +08:00
- [2. 文本识别](#2)
2022-04-26 21:57:23 +08:00
这里整理了OCR中常用的公开数据集持续更新中欢迎各位小伙伴贡献数据集
2022-04-26 21:54:06 +08:00
<a name="1"></a>
2022-04-27 14:04:18 +08:00
### 1. 文本检测
2022-04-26 21:54:06 +08:00
2022-04-26 22:00:12 +08:00
| 数据集名称 |图片下载地址| PPOCR标注下载地址 |
|---|---|---|
2022-04-26 21:54:06 +08:00
| ICDAR 2015 |https://rrc.cvc.uab.es/?ch=4&com=downloads| [train](https://paddleocr.bj.bcebos.com/dataset/train_icdar2015_label.txt) / [test](https://paddleocr.bj.bcebos.com/dataset/test_icdar2015_label.txt) |
2022-04-26 22:00:12 +08:00
| ctw1500 |https://paddleocr.bj.bcebos.com/dataset/ctw1500.zip| 图片下载地址中已包含 |
2022-04-26 21:54:06 +08:00
| total text |https://paddleocr.bj.bcebos.com/dataset/total_text.tar| 图片下载地址中已包含 |
2022-04-27 14:04:18 +08:00
<a name="11"></a>
#### 1.1 ICDAR 2015
icdar2015 数据集包含1000张训练图像和500张测试图像。icdar2015数据集可以从上表中链接下载首次下载需注册。
注册完成登陆后,下载下图中红色框标出的部分,其中, `Training Set Images`下载的内容保存在`icdar_c4_train_imgs`文件夹下,`Test Set Images` 下载的内容保存早`ch4_test_images`文件夹下
<p align="center">
<img src="../../datasets/ic15_location_download.png" align="middle" width = "700"/>
<p align="center">
将下载到的数据集解压到工作目录下,假设解压在 PaddleOCR/train_data/下。然后从上表中下载转换好的标注文件。
PaddleOCR 也提供了数据格式转换脚本,可以将官网 label 转换支持的数据格式。 数据转换工具在 `ppocr/utils/gen_label.py`, 这里以训练集为例:
```
# 将官网下载的标签文件转换为 train_icdar2015_label.txt
python gen_label.py --mode="det" --root_path="/path/to/icdar_c4_train_imgs/" \
--input_path="/path/to/ch4_training_localization_transcription_gt" \
--output_label="/path/to/train_icdar2015_label.txt"
```
解压数据集和下载标注文件后PaddleOCR/train_data/ 有两个文件夹和两个文件按照如下方式组织icdar2015数据集
```
/PaddleOCR/train_data/icdar2015/text_localization/
└─ icdar_c4_train_imgs/ icdar 2015 数据集的训练数据
└─ ch4_test_images/ icdar 2015 数据集的测试数据
└─ train_icdar2015_label.txt icdar 2015 数据集的训练标注
└─ test_icdar2015_label.txt icdar 2015 数据集的测试标注
```
2022-04-26 21:54:06 +08:00
<a name="2"></a>
2022-04-27 14:04:18 +08:00
### 2. 文本识别
2022-04-26 21:54:06 +08:00
2022-04-26 22:30:22 +08:00
| 数据集名称 | 图片下载地址 | PPOCR标注下载地址 |
|---|---|---------------------------------------------------------------------|
| en benchmark(MJ, SJ, IIIT, SVT, IC03, IC13, IC15, SVTP, and CUTE.) | [DTRB](https://github.com/clovaai/deep-text-recognition-benchmark#download-lmdb-dataset-for-traininig-and-evaluation-from-here) | LMDB格式可直接用[lmdb_dataset.py](../../../ppocr/data/lmdb_dataset.py)加载 |