pip install pillow
augmentImages.py
# ---
#1 Amplify the image file to 20 files and store it in the generated folder
# ---
from PIL import Image, ImageOps
import glob, os
#Specify the file to be amplified
files = glob.glob('images/**/*.JPG', recursive=True)
for i, file in enumerate(files):
im = Image.open(file)
print(i + 1, '/', len(files), file, im.format, im.size, im.mode)
#Create output directory
new_dir = 'generated/' + os.path.dirname(file)
os.makedirs(new_dir, exist_ok=True)
#Resized to be 1000px wide
resize_ratio = float(1000) / im.width
im = im.resize((int(im.width * resize_ratio), int(im.height * resize_ratio)))
#Save original image
im.save(new_dir + '/' + os.path.splitext(os.path.basename(file))[0] + '.JPG', quality=95)
#-5〜+Rotate up to 5 degrees and save each
im.rotate(1).save(new_dir + '/' + os.path.splitext(os.path.basename(file))[0] + '-p1.JPG', quality=95)
im.rotate(2).save(new_dir + '/' + os.path.splitext(os.path.basename(file))[0] + '-p2.JPG', quality=95)
im.rotate(3).save(new_dir + '/' + os.path.splitext(os.path.basename(file))[0] + '-p3.JPG', quality=95)
im.rotate(4).save(new_dir + '/' + os.path.splitext(os.path.basename(file))[0] + '-p4.JPG', quality=95)
im.rotate(5).save(new_dir + '/' + os.path.splitext(os.path.basename(file))[0] + '-p5.JPG', quality=95)
im.rotate(-1).save(new_dir + '/' + os.path.splitext(os.path.basename(file))[0] + '-n1.JPG', quality=95)
im.rotate(-2).save(new_dir + '/' + os.path.splitext(os.path.basename(file))[0] + '-n2.JPG', quality=95)
im.rotate(-3).save(new_dir + '/' + os.path.splitext(os.path.basename(file))[0] + '-n3.JPG', quality=95)
im.rotate(-4).save(new_dir + '/' + os.path.splitext(os.path.basename(file))[0] + '-n4.JPG', quality=95)
im.rotate(-5).save(new_dir + '/' + os.path.splitext(os.path.basename(file))[0] + '-n5.JPG', quality=95)
#Invert-5〜+Rotate up to 5 degrees and save each
ImageOps.mirror(im).rotate(1).save(new_dir + '/' + os.path.splitext(os.path.basename(file))[0] + '-mp1.JPG', quality=95)
ImageOps.mirror(im).rotate(2).save(new_dir + '/' + os.path.splitext(os.path.basename(file))[0] + '-mp2.JPG', quality=95)
ImageOps.mirror(im).rotate(3).save(new_dir + '/' + os.path.splitext(os.path.basename(file))[0] + '-mp3.JPG', quality=95)
ImageOps.mirror(im).rotate(4).save(new_dir + '/' + os.path.splitext(os.path.basename(file))[0] + '-mp4.JPG', quality=95)
ImageOps.mirror(im).rotate(5).save(new_dir + '/' + os.path.splitext(os.path.basename(file))[0] + '-mp5.JPG', quality=95)
ImageOps.mirror(im).rotate(-1).save(new_dir + '/' + os.path.splitext(os.path.basename(file))[0] + '-mn1.JPG', quality=95)
ImageOps.mirror(im).rotate(-2).save(new_dir + '/' + os.path.splitext(os.path.basename(file))[0] + '-mn2.JPG', quality=95)
ImageOps.mirror(im).rotate(-3).save(new_dir + '/' + os.path.splitext(os.path.basename(file))[0] + '-mn3.JPG', quality=95)
ImageOps.mirror(im).rotate(-4).save(new_dir + '/' + os.path.splitext(os.path.basename(file))[0] + '-mn4.JPG', quality=95)
ImageOps.mirror(im).rotate(-5).save(new_dir + '/' + os.path.splitext(os.path.basename(file))[0] + '-mn5.JPG', quality=95)
Python code for learning and testing with Custom Vision of Cognitive Service --Qiita
Recommended Posts