项目场景: pytorch 训练EAST OCR文本检测模型 问题描述: 加载数据集并做数据增强时,出现exception,导致训练集加载失败。 原因分析: 从异常信息推断,数据类型不匹配导致了这一问题,通过debug发现如下数据类型不一致: 解决方案: 修改传入实体数据"xy_in_poly"的类型为“np.int64”,或者修改函数“gen_geo_map()”第二个参数的数据类型为“np.int32_t”,保证输入实体数据参数与函数定义参数数据类型一致即可。 如下为第一种修改方案: