首页
/ Total Text数据集:解锁场景文本检测的全方位解决方案

Total Text数据集:解锁场景文本检测的全方位解决方案

2026-04-25 10:42:21作者:羿妍玫Ivan

价值定位:为何Total Text是任意形状文本识别的理想选择?

在场景文本检测领域,如何应对现实世界中复杂多变的文本形态一直是研究难点。Total Text数据集以1555张图像的规模,提供了水平、多方向和曲线文本的全方位标注,填补了传统数据集在非规则文本研究中的空白。该数据集由Chee Kheng Ch'ng等学者构建,通过精细的多边形标注和丰富的场景覆盖,成为评估任意形状文本识别算法鲁棒性的行业基准。与ICDAR等传统数据集相比,其7.37的单图文本实例密度,更贴近真实世界的文本分布特征。

核心特性:掌握Total Text的三大技术突破

Total Text的独特价值体现在其对复杂文本形态的系统性覆盖。数据集包含11459个文本实例,其中曲线文本占比达23%,这种比例设计专门针对现有算法的薄弱环节。通过多边形标注格式,每个文本实例被精确到像素级边界,支持从字符到文本行的多粒度分析。

Total Text数据集标注示例

图1:Total Text的多边形标注示例,展示了对弯曲文本的精确边界定义,包含坐标点序列和方向属性

数据集采用三级分类体系:水平文本(占比41%)、多方向文本(36%)和曲线文本(23%),这种平衡分布确保模型训练的全面性。标注工具源码Annotation_tools/T3/提供了从初始点标注到多边形调整的完整工作流,支持研究者扩展自定义标注任务。

高效应用:实战场景文本检测的关键技巧

快速上手Total Text需掌握两个核心技巧。首先是数据预处理阶段的坐标归一化:将多边形顶点坐标转换为相对于图像尺寸的比例值,可有效降低不同分辨率图像带来的尺度偏差。其次是标注格式转换,通过Evaluation_Protocol/Python_scripts/polygon_wrapper.py工具,可将原生.mat格式标注转换为COCO格式,无缝对接主流深度学习框架。

import scipy.io as sio
import numpy as np

def convert_mat_to_coco(mat_path, img_id):
    """
    将Total Text的.mat标注转换为COCO格式
    Args:
        mat_path: .mat标注文件路径
        img_id: 图像ID
    Returns:
        coco_annotation: COCO格式标注字典
    """
    # 加载mat文件(Total Text标注采用MATLAB格式存储)
    mat_data = sio.loadmat(mat_path)
    polygons = mat_data['polygt']  # 获取多边形标注数据
    
    coco_annotations = []
    for idx, poly in enumerate(polygons):
        # 提取x,y坐标并转换为整数
        x_coords = [int(p) for p in poly[1][0]]
        y_coords = [int(p) for p in poly[3][0]]
        # 组合坐标点,形成COCO格式的segmentation
        segmentation = [np.column_stack((x_coords, y_coords)).flatten().tolist()]
        
        # 构建COCO标注字典
        coco_annotations.append({
            'id': idx,
            'image_id': img_id,
            'category_id': 1,  # 文本类别ID
            'segmentation': segmentation,
            'bbox': [min(x_coords), min(y_coords), 
                     max(x_coords)-min(x_coords), 
                     max(y_coords)-min(y_coords)],
            'area': (max(x_coords)-min(x_coords)) * (max(y_coords)-min(y_coords)),
            'iscrowd': 0
        })
    return coco_annotations

Total Text数据集统计分布

图2:Total Text与其他主流数据集的统计对比,展示其在文本方向多样性上的优势

生态拓展:构建任意形状文本识别的完整链条

Total Text的价值不仅在于数据集本身,更在于其构建的研究生态。官方提供的Evaluation_Protocol/ComputePrecisionRecall.m实现了基于多边形交并比的评估指标,支持弯曲文本检测结果的量化分析。社区基于该数据集已衍生出多种改进算法,如结合注意力机制的曲线文本检测模型,以及基于贝塞尔曲线的文本形状建模方法。通过将Total Text与预训练模型结合,研究者可快速验证新算法在复杂文本场景下的性能表现,推动场景文本检测技术的持续突破。

Total Text数据集通过其独特的设计理念和丰富的标注信息,为任意形状文本识别研究提供了标准化的测试平台。无论是学术研究还是工业应用,掌握该数据集的使用方法都将为文本检测系统的开发带来显著优势。

登录后查看全文
热门项目推荐
相关项目推荐