"""

This program augments (performs rotate/shear operations) on all images with the .jpg extension in the specified

folder (the extension type can be changed to .png below) and puts them in a folder called augmented.

If no file path is given as an argument when running the program, it will just augment images in the current working directory.

I used this script to generate extra images for my Data Mining project so that we would have a larger dataset to work with.

Designed for Python 3.6.

"""

from skimage import transform as tf

from skimage import io

import numpy as np

import math

import sys

import os

import random

DIR_NAME = 'augmented' #name of the folder augmented images are placed in

FILL_MODE = 'wrap' #options: constant, edge, symmetric, reflect, wrap

MAX_ROTATE_ANGLE = math.pi/float(6) #radians

MAX_SHEAR_ANGLE = 0.25 #radians\

IMG_TYPE = ".jpg" #image extension type

def rotate(name,img_arr):

io.imsave(name+"_backrotated" + IMG_TYPE,back_rotated)

def shear(name,img_arr):

io.imsave(name+"_sheared_inverse" + IMG_TYPE,sheared_inv)

def flip(name,img_arr):

flipped = np.fliplr(img_arr)

io.imsave(name+"_flipped"+IMG_TYPE,flipped)

return flipped

def swirl(name,img_arr):

swirled = tf.swirl(img_arr,mode=FILL_MODE)

io.imsave(name+"_swirled"+IMG_TYPE,swirled)

def main():

swirl(aug_filename+"_flipped",flipped)

if __name__ == "__main__":

main()