File size: 1,656 Bytes
cc095d7 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 |
import os
import argparse
from src.concurrent_task_executor import concurrent_task_executor
from src.utils import crop_and_pad_image, track_files
from typing import NamedTuple, Tuple
parser = argparse.ArgumentParser(description="Crop and Resize Images in a folder")
parser.add_argument("--src", type=str, help="source folder", required=True)
parser.add_argument("--dest", type=str, help="destination folder", required=True)
parser.add_argument("--size", type=Tuple[int,int], help="Size of image in pixels, given as a (width, height) tuple", default=(512, 512))
class FileInfo(NamedTuple):
src: str
dest: str
size: Tuple[int, int] # (width, height) tuple.
def crop_and_save_image(file_info: FileInfo):
try:
cropped_image = crop_and_pad_image(file_info.src, target_size=file_info.size)
cropped_image.save(file_info.dest)
except Exception as e:
print(f"Error processing image: {str(e)}")
if __name__ == "__main__":
args = parser.parse_args()
src_folder = args.src
dst_folder = args.dest
size = args.size
# check if destination folder exists
if not os.path.exists(dst_folder):
print("Destination folder does not exist. Creating folder...")
os.makedirs(dst_folder, exist_ok=True)
files = [
FileInfo(
src_image_path,
os.path.join(dst_folder, os.path.basename(src_image_path)),
size
)
for src_image_path in track_files(src_folder)
]
# cropping and resizing images and saving them to the destination folder
concurrent_task_executor(crop_and_save_image, files, description="Processing images")
|