/TunAugmentor

A Python library for image augmentation

Primary LanguagePythonMIT LicenseMIT

TunAugmentor

TunAugmentor is a python library for image data augmentation. Image augmentation is an effective technique widely used for Machine Learning and Computer Vision tasks. The aim of image augmentation is to apply different transformations on existing images to create more data for the model hence increasing the performance of the model. Therefore, it is the process of increasing the training dataset without collecting new data. You can find the documentation on https://ahmedbelgacem.github.io/TunAugmentor/ .

Table of contents

About TunAugmentor

  • The aim of this project was to reimplement different image augmentation techniques using only Python and Numpy. We do not claim we reinvented the wheel. We know theese techniques are already available in different libraries with better implementations and you should probably use theese instead. The goal was to learn to implement this techniques and to try to distribute a Python Library. Thus, we didn't do any benchmarking for this work.
  • Albumentations was a great source of inspiration concerning which transformation to implement and for the documentation. Try to check their work you may find what you really need.

Authors

Installation

pip install TunAugmentor

Example

from TunAugmentor.transformations import Flip,RandomRotation90,CenterCrop
from TunAugmentor.pipeline import Pipeline
from TunAugmentor.utils import read_images,export

Augmentor=Pipeline([Flip(V=-1,H=-1),RandomRotation90(),CenterCrop(300,300)])
images=read_images('./image_folder')
images=Augmentor.apply(images)
res=export(images,'./res')

Getting Started

You can check the Getting Started with TunAugmentor Notebook for a concrete example.

Transformation List