用于图像增强的最快批量透视变换

数据挖掘 火炬 图像预处理 数据增强 scipy
2022-02-15 16:18:40

我需要为神经网络的训练图像做一些增强。
问题是即使在并行加载批次时,增强的执行时间也比网络训练要长。

问题在于我正在应用的透视变换。和(使用 PIL 图像)
花费了大致相同的时间。cv2.warpPerspectivetorchvision.transforms.functional.perspective

但是他们俩都是按图像来做的(这对我来说似乎效率很低)。

那么,执行透视变换的最佳(最快和批量)方法是什么?

  • 我想直接用 PyTorch 函数来做,但我不知道算法。(接受一个好的算法作为答案)
  • 还有其他可以更快完成的软件包吗?
1个回答

图像增强在很大程度上依赖于您的DataGeneratorDataLoader设计,主要是与您正在使用的硬件资源一起。除此之外,这里有一个快速比较图表,可帮助您进行转换以及库链接。

这些数字代表它每秒处理的图像数量

在此处输入图像描述

以下是所有必需库的链接:

  1. 相册
  2. 伊姆高格
  3. 增强器
  4. 索尔特

参考:Albulentations 基准测试结果