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