python 图像处理库具有用于生成数据集的滤镜(失真、灯光)

python image processing library has filters(distortion, lights) used for generate dataset

我正在尝试生成一个用于对象检测的 数据集 (关于 1e6 图片)。我有物体和背景的图像。但我认为在对象&对象+背景上添加一些过滤器对模型训练有好处。经过调查,我发现我必须使用以下过滤器:

  1. 变形,包括曲线剪切、球面变形

  2. 给 img 投光。

  3. 其他简单滤镜,如旋转、调整大小、模糊、噪点、颜色渐变...

我想知道有没有什么库,使用简单(input args + [w, h, 4], return [w, h, 4]),自然,快速,最重要的是,它可以在 img 上添加灯光和镜头灯光,并扭曲它。

或任何可以在 mxnet 训练期间使用 python 的库。

ORG IMG1

Distortion1

ORG2

Distortion2

Lights

在观察Photoshop之后,我尝试用PIL、opencv和scimage用球体、椭圆和para-curve公式制作自己的滤镜,但是这些滤镜并不自然,不如Photoshop好

我也试过gimp-python,但我想在纯python程序中使用它,我想在训练过程中处理图片。也许速度真的很重要。

这些博客好像是个实用的乱码方式:http://paulbourke.net/miscellaneous/imagewarp/