rotate_img.py

rotate_img.py

# import packages
from PIL import Image
import os
import argparse # build command-line interfaces


# construct the argument parse and parse the arguments
ap = argparse.ArgumentParser()
ap.add_argument("-d",
                "--directory",
                required=True,
                help="path to input directory")
args = vars(ap.parse_args())


# load the directory path
directory = args["directory"]

for filename in os.listdir(directory):
	if filename.endswith(".jpg") or filename.endswith(".jpeg"): 
		image = os.path.join(directory, filename)        
		# print(image)
		# Load the original image:
		img = Image.open(image)

		# couterclockwise rotation
		img2 = img.rotate(90, expand=True)
		img2.save(image)
	else:
		pass





# Resampling filters
# ==================
# 
# Nearest Neighbor (default):
# img2 = img.rotate(45, respample=Image.NEAREST)
#
# Linear interpolation:
# img2 = img.rotate(45, respample=Image.BILINEAR)
#
# Cubic-spline interpolation:
# img2 = img.rotate(45, respample=Image.BICUBIC)