[Resize all imgs in a directory] #Python

resize_all_imgs_in_dir.py

import os
import pandas as pd

from cv2 import imread, resize, imwrite, IMREAD_GRAYSCALE, INTER_AREA

# Extension of files needed:
ext = '.jpg'
label_ext = '.txt'
# Directory of interest:
dir_path = 'dataset'
out_path = 'resized_dataset'
resize_factor = 2
LABEL_SEPARATOR = '\n'

# Loop through files:
for (dirpath, dirnames, filenames) in os.walk(dir_path):
    # Loop through Annotations:
    for i in range(len(filenames)):
        # Open files of interest:
        if filenames[i].endswith(ext):
            img_path = os.path.join(dirpath, filenames[i])
            current_img = imread(img_path, IMREAD_GRAYSCALE)
            label_df = pd.read_csv(os.path.join(img_path.split(ext)[0] +
                                                label_ext),
                                   sep=LABEL_SEPARATOR, header=None)

            # Resize:
            print(int(current_img.shape[0]/resize_factor))
            resized_img = resize(current_img, (int(current_img.shape[0]/resize_factor),
                                               int(current_img.shape[1]/resize_factor)),
                                 interpolation=INTER_AREA)
            label_df = label_df / resize_factor

            # Save:
            imwrite(os.path.join(out_path, 'resized' + filenames[i]), resized_img)
            label_df.to_csv(os.path.join(out_path, 'resized' +
                                    filenames[i].split(ext)[0] + label_ext),
                            header=None, index=None, sep=LABEL_SEPARATOR,
                            encoding='utf-8')