import argparse
import os
import cv2

parser = argparse.ArgumentParser(description='image_eval')
parser.add_argument('--orig_path', help='path to orig image dataset', default='orig/')
parser.add_argument('--image_format', help='format of the image', default='png')
parser.add_argument('--factor', help='jpeg compression factor, 70, 80, 90', default='70')
opt = parser.parse_args()

def main():
    factor = int(opt.factor)

    num_files = 0
    for i in os.listdir(opt.orig_path):
        num_files += 1

    for idx in range(num_files):
        img = cv2.imread('%s/original%d.%s' % (opt.orig_path, idx, opt.image_format))
        cv2.imwrite('./jpeg_%d/original_%d.jpg' % (factor, idx), img, [int(cv2.IMWRITE_JPEG_QUALITY), factor])

if __name__ == '__main__':
    main()