/AugStatic

AugStatic - A Light-Weight Image Augmentation Library

Primary LanguagePython

AugStatic - A Light-Weight Image Augmentation Library

PWC

Abstract

[ Paper Link ]

The rapid exponential increase in the data led to an abrupt mix of various data types, leading to a deficiency of helpful information. Creating new data with the existing different types of data are presented in this paper. Augmentation is adding up or modifying the dataset with extra data. There are many types of augmentation done for various kinds of datasets. Augmentation has been widely used in multiple pre-processing steps of diverse machine learning pipelines. Many libraries or packages are made for augmentation called augmentation libraries. There are many salient features that each library supports. This paper seeks to enhance the library that makes the AugStatic library much more lightweight and efficient. AugStatic is a custom-built image augmentation library with lower computation costs and more extraordinary salient features compared to other image augmentation libraries. This framework can be used for NumPy array and tensors too.

Background Research Work

Aug_types

Various types of augmentations were researched and compiled into a compact, lightweight, and practical library.

  • Imgaug [ Library-GitHub-Link | Documentation-Link ]

    The salient feature are –

    • It contains over forty image augmentation techniques.
    • Functionality to augment images with masks, key points, bounding boxes, and heat maps.
    • Easier to augment the image dataset for object detection and segmentation problems.
    • Complex augmentation pipelines.
    • Many helper functions for augmentation visualization, conversion, and more.

  • Augmentor [ Library-GitHub-Link | Documentation-Link ]

    The salient feature are –

    • It has fewer possible augmentations compared to other packages.
    • It supports extra features like size-preserving shearing, size-preserving rotations, and cropping, which is beneficial for machine learning pipelines.
    • It supports to compose augmentation pipelines.
    • It supports usage with PyTorch [8] and Tensorflow [1].

  • Albumentations [ Library-GitHub-Link | Documentation-Link ]

    The salient feature are –

    • It contains over forty image augmentation techniques.
    • Functionality to augment images with masks, key points, bounding boxes, and heat maps.
    • Easier to augment the image dataset for object detection and segmentation problems.
    • Complex augmentation pipelines.
    • Many helper functions for augmentation visualization, conversion, and more.

Methods & Results

Augmentation Technique Input Image Output Image
Blur Input_image Blur
CLAHE (Contrast Stretched Adaptive Histogram Equalization) Input_image CLAHE
ChannelDropout Input_image ChannelDropout
ChannelShuffle Input_image ChannelShuffle
ColorJitter Input_image ColorJitter
Downscale Input_image Downscale
Emboss Input_image Emboss
FancyPCA Input_image FancyPCA
GaussNoise Input_image GaussNoise
GaussianBlur Input_image GaussianBlur
GlassBlur Input_image GlassBlur
HueSaturationValue Input_image HueSaturationValue
ISONoise Input_image ISONoise
InvertImg Input_image InvertImg
MedianBlur Input_image MedianBlur
MotionBlur Input_image MotionBlur
MultiplicativeNoise Input_image MultiplicativeNoise
Posterize Input_image Posterize
RGBShift Input_image RGBShift
Sharpen Input_image Sharpen
Solarize Input_image Solarize
Superpixels Input_image Superpixels
ToGray Input_image ToGray
ToSepia Input_image ToSepia
VerticalFlip Input_image VerticalFlip
HorizontalFlip Input_image HorizontalFlip
Transpose Input_image Transpose
OpticalDistortion Input_image OpticalDistortion
GridDistortion Input_image GridDistortion
JpegCompression Input_image JpegCompression
Cutout Input_image Cutout
CoarseDropout Input_image CoarseDropout
GridDropout Input_image GridDropout

Conclusion:

  • In this research, An light weight Efficient Augmentation library has been developed, named AugStatic
  • This framework can be used for NumPy arrays and tensors too.
  • It supports all the augmentations of PyTorch, Keras, Imgaug, Albumentations and Augmentor.
  • AugStatic is a custom-built image augmentation library with lower computation costs and efficiency compared to other image augmentation libraries.
  • It is built on python and is easily understandable and flexible enough to keep adding features. Hence, making it more scalable
  • With the advancement in augmentation, there is a lot of scope in making the AugStatic library for audio, NLP, and time-series data.

To cite my paper:

Citing Text
"AugStatic - A Light-Weight Image Augmentation Library", International Journal of Emerging Technologies and Innovative Research (www.jetir.org), ISSN:2349-5162, Vol.9, Issue 5, page no.b735-b742, May-2022, Available :http://www.jetir.org/papers/JETIR2205199.pdf