STRAug 项目使用教程

STRAug 项目使用教程

straugImage transformations designed for Scene Text Recognition (STR) data augmentation. Published at ICCV 2021 Workshop on Interactive Labeling and Data Augmentation for Vision.项目地址:https://gitcode.com/gh_mirrors/st/straug

1. 项目的目录结构及介绍

straug/
├── straug/
│   ├── __init__.py
│   ├── warp.py
│   ├── geometry.py
│   ├── pattern.py
│   ├── blur.py
│   ├── noise.py
│   ├── weather.py
│   ├── camera.py
│   ├── process.py
│   └── utils.py
├── tests/
│   ├── test_warp.py
│   ├── test_geometry.py
│   ├── test_pattern.py
│   ├── test_blur.py
│   ├── test_noise.py
│   ├── test_weather.py
│   ├── test_camera.py
│   └── test_process.py
├── README.md
├── setup.py
└── requirements.txt

• straug/: 包含所有数据增强功能的模块。
  • warp.py: 包含曲线、扭曲和拉伸变形的函数。
  • geometry.py: 包含透视、旋转和收缩变形的函数。
  • pattern.py: 包含不同网格生成的函数。
  • blur.py: 包含各种模糊效果的函数。
  • noise.py: 包含各种噪声添加的函数。
  • weather.py: 包含模拟天气条件的函数。
  • camera.py: 包含模拟相机传感器调谐和图像压缩/调整大小的函数。
  • process.py: 包含其他图像处理问题的函数。
  • utils.py: 包含辅助函数。
• tests/: 包含所有模块的测试文件。
• README.md: 项目说明文档。
• setup.py: 项目安装文件。
• requirements.txt: 项目依赖文件。

2. 项目的启动文件介绍

项目的启动文件主要是 setup.py，用于安装项目所需的依赖和模块。可以通过以下命令安装：

pip install .

3. 项目的配置文件介绍

项目没有明确的配置文件，但可以通过修改 straug 模块中的各个函数参数来调整数据增强的效果。例如：

from straug.warp import Curve
from PIL import Image

img = Image.open("nokia.png")
img = Curve()(img, mag=3)
img.save("augmented_nokia.png")

在这个例子中，mag=3 是调整曲线变形强度的参数。

straugImage transformations designed for Scene Text Recognition (STR) data augmentation. Published at ICCV 2021 Workshop on Interactive Labeling and Data Augmentation for Vision.项目地址:https://gitcode.com/gh_mirrors/st/straug