diff options
Diffstat (limited to 'test.py')
| -rw-r--r-- | test.py | 4 |
1 files changed, 1 insertions, 3 deletions
@@ -58,8 +58,6 @@ def process_dir(v): 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)) - - def G(x, is_train): # shape of x: [B,T_in,H,W,C] @@ -144,7 +142,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) |