import argparse
import scipy.misc
import numpy as np
import sys
import glob
sys.path.append('./')
parser = argparse.ArgumentParser()
parser.add_argument("--src", help="src image path", type=str)
parser.add_argument("--dst", help="dst image path", type=str)
parser.add_argument("--sample_number", help="sample number", type=int)
def process_img_origin(file):
	try:
		t = scipy.misc.imread(file)
	except:
		return None
	return t
if __name__ == '__main__':
	args = parser.parse_args()
	imgs = glob.glob(args.src+'/*.jpeg')
	np.random.shuffle(imgs)
	for i,m in enumerate(imgs[:arg.sample_number]):
		img = process_img_origin(m)
		if img != None and len(img.shape) != 0:
			scipy.misc.imsave(args.dst+'/'+str(i).zfill(6)+'.jpg',img)