def load_images(o_dir, i_dir, n_images=3000, size=128):
prepare_inception_data(o_dir, i_dir)
image_list = []
done = False
cnt = 0
bar = progressbar.ProgressBar(redirect_stdout=True,
max_value=progressbar.UnknownLength)
for root, dirs, files in os.walk(o_dir):
if files:
for f in files:
cnt += 1
file_dir = os.path.join(root, f)
image_list.append(ip.load_image_inception(file_dir, 0))
bar.update(cnt)
if len(image_list) == n_images:
done = True
break
if done:
break
bar.finish()
print('Finished Loading Files')
return image_list
评论列表
文章目录