def streaming_images(files, convert_func, func_args):
for i, f in enumerate(files):
img = cv2.imread(f)
img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
if convert_func is not None:
# image conversion
img = convert_func(img, func_args)
yield ImageUtils.compress_image(img), f[f.rfind('/') + 1:]
def read_images(path, convert_func, func_args):
imgs = {}
path = os.path.abspath(path)
print('Reading images in %s' % path)
sess = tf.Session()
for _, _, files in os.walk(path):
for i, f in enumerate(files):
img = cv2.imread(os.path.join(path, f))
img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
if convert_func is not None:
# image conversion
img = convert_func(img, func_args)
imgs.update({f[:f.rfind('.')]: ImageUtils.compress_image(img)})
print('\r{:.1%}'.format((i + 1) / len(files)), end='')
print('\n')
sess.close()
return imgs