diff options
Diffstat (limited to 'test.py')
| -rw-r--r-- | test.py | 6 |
1 files changed, 3 insertions, 3 deletions
@@ -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)) |