unclestrange / dataaugmentation

Data augmentation using OpenCV

[WIP] Data augmentation tools

This repository provides tools for data augmentation using OpenCV.

Usage

cat = cv2.imread('cat.jpg')
dog = cv2.imread('dog.jpg')

augmented_images = data_augmentation([cat, dog])

for i in range(len(augmented_images)):
	cv2.imwrite('results/' + str(i)+'.jpg', augmented_images[i])

Data augmentation techniques

• Inversion
• Sobel derivative
• Scharr derivative
• Laplacian
• Blur
• Gaussian blur
• Median blur
• Bilateral blur
• Horizontal flips
• Distortions
• Random scales
• Stretching
• Color jittering
• Shearing