File size: 1,322 Bytes
561c629 |
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 |
import os, sys, cv2, shutil, argparse
if __name__ == "__main__":
# Parse variables available
parser = argparse.ArgumentParser()
parser.add_argument('-i', '--input_dir', type = str)
parser.add_argument('-o', '--store_dir', type = str)
args = parser.parse_args()
input_dir = args.input_dir
store_dir = args.store_dir
print("We are cropping the image for 4x scale such that it is suitable for video compression")
# Check file
if os.path.exists(store_dir):
shutil.rmtree(store_dir)
os.makedirs(store_dir)
# Process
for file_name in sorted(os.listdir(input_dir)):
need_reszie = False
source_path = os.path.join(input_dir, file_name)
destination_path = os.path.join(store_dir, file_name)
img = cv2.imread(source_path)
h, w, c = img.shape
if h % 8 != 0:
print("We need vertical resize")
need_reszie = True
img = img[:8 * (h // 8), :, :]
if w % 8 != 0:
print("We need horizontal resize")
need_reszie = True
img = img[:, :8 * (w // 8), :]
if need_reszie:
cv2.imwrite(destination_path, img, [cv2.IMWRITE_PNG_COMPRESSION, 0])
else:
shutil.copy(source_path, destination_path) |