38 lines
973 B
Python
38 lines
973 B
Python
|
|
import glob
|
|
import logging
|
|
import os
|
|
import cv2
|
|
import sys
|
|
|
|
|
|
def resize(input_glob, output_dir, factor=2):
|
|
"""
|
|
Resize images using multiplicative factor
|
|
"""
|
|
list_im = glob.glob(input_glob)
|
|
|
|
|
|
for idx, path_in in enumerate(list_im):
|
|
|
|
basename, _ = os.path.splitext(os.path.basename(path_in))
|
|
im = cv2.imread(path_in)
|
|
assert im is not None, "Image not found"
|
|
|
|
# Paddle the image if requested and resized the dataset to a fixed dataset
|
|
h_im = im.shape[0]
|
|
w_im = im.shape[1]
|
|
w_new = round(factor * w_im)
|
|
h_new = round(factor * h_im)
|
|
|
|
print("resizing image {} to: {} x {}".format(basename, w_new, h_new))
|
|
im_new = cv2.resize(im, (w_new, h_new))
|
|
|
|
# Save the image
|
|
name_im = basename + '.png'
|
|
path_out = os.path.join(output_dir, name_im)
|
|
cv2.imwrite(path_out, im_new)
|
|
sys.stdout.write('\r' + 'Saving image number: {}'.format(idx) + '\t')
|
|
|
|
|