论文原文,答辩PPT,读书报告,实现代码增强学习:文本识别的联合数据增强和网络优化.-CSDN下载
CVPR2020论文:
《Learn to Augment: Joint Data Augmentation and Network Optimization for Text》
原文链接
官方代码实现
重写Python代码实现(非本人编码)
摘要
手写文本和场景文本存在各种形状和扭曲的图案。因此,训练一个健壮的识别模型需要大量的数据来尽可能地覆盖多样性。与数据收集和注释相比,数据增强是一种低成本的方式。本文提出了一种新的文本图像增强方法。与传统的旋转、缩放和透视变换等增强方法不同,我们提出的增强方法是为了学习适当和高效的数据增强,这对于训练一个健壮的识别器来说更加有效和具体。通过使用一组自定义基准点,所提出的增强方法具有灵活性和可控性。此外,我们还通过联合学习在孤立的数据扩充过程和网络优化过程之间架起了一座桥梁。代理网络从识别网络的输出中学习,并控制基准点来为识别网络生成更合适的训练样本。在包括规则场景文本、不规则场景文本和手写文本在内的各种基准上的大量实验表明,所提出的增强和联合学习方法显著提高了识别网络的性能。提供了用于几何增强的通用工具包。
答辩PPT
代码实现
参考链接
输入图片:
输出图片:
使用开发环境:Python3.7 Pycharm2019
下载代码,运行即可。
安装 NumPy 最简单的方法就是使用 pip 工具:
pip3 install --user numpy scipy matplotlib
--user 选项可以设置只安装在当前的用户下,而不是写入到系统目录。
默认情况使用国外线路,国外太慢,我们使用清华的镜像就可以:
pip3 install numpy scipy matplotlib -i https://pypi.tuna.tsinghua.edu.cn/simple
没有找到cv2,应该安装opencv-python
pip install opencv-python scipy matplotlib -i https://pypi.tuna.tsinghua.edu.cn/simple
没有找到imageio,应该安装imageio
pip install imageio scipy matplotlib -i https://pypi.tuna.tsinghua.edu.cn/simple