From c40b92b7a480188a6e32d58f5488ba8449e168ab Mon Sep 17 00:00:00 2001 From: Jules Laplace Date: Sun, 18 Nov 2018 21:22:02 +0100 Subject: frame count --- test.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/test.py b/test.py index 3347063..b91297f 100644 --- a/test.py +++ b/test.py @@ -120,7 +120,7 @@ with tf.Session(config=config) as sess: out_H = sess.run(GH, feed_dict={H: in_H, is_train: False}) out_H = np.clip(out_H, 0, 1) - Image.fromarray(np.around(out_H[0,0]*255).astype(np.uint8)).save('./results/{}L/G/{}/Frame{:03d}.png'.format(args.L, scene_name, i+1)) + Image.fromarray(np.around(out_H[0,0]*255).astype(np.uint8)).save('./results/{}L/G/{}/frame_{:05d}.png'.format(args.L, scene_name, i+1)) out_Hs.append(out_H[0, 0]) out_Hs = np.asarray(out_Hs) @@ -142,7 +142,7 @@ def process_dir(v): scene_name = v.split('/')[-1] os.mkdir('./results/{}L/L/{}/'.format(args.L, scene_name)) - dir_frames = glob.glob(v + '/*.png') + dir_frames = sorted(glob.glob(v + '/*.png')) dir_frames.sort() frames = [] @@ -159,6 +159,6 @@ def process_dir(v): out_H = sess.run(GH, feed_dict={L: in_L, is_train: False}) out_H = np.clip(out_H, 0, 1) - Image.fromarray(np.around(out_H[0,0]*255).astype(np.uint8)).save('./results/{}L/L/{}/Frame{:03d}.png'.format(args.L, scene_name, i+1)) + Image.fromarray(np.around(out_H[0,0]*255).astype(np.uint8)).save('./results/{}L/L/{}/frame_{:05d}.png'.format(args.L, scene_name, i+1)) -- cgit v1.2.3-70-g09d2