diff options
| -rw-r--r-- | crop_frames.py | 2 | ||||
| -rw-r--r-- | test.py | 2 |
2 files changed, 2 insertions, 2 deletions
diff --git a/crop_frames.py b/crop_frames.py index ee879a3..8314816 100644 --- a/crop_frames.py +++ b/crop_frames.py @@ -8,7 +8,7 @@ from PIL import Image parser = argparse.ArgumentParser() parser.add_argument('--step', default=256, type=int) parser.add_argument('--overlap', default=4, type=int) -parser.add_argument('--scale', default=1, type=int) +parser.add_argument('--scale', default=4, type=int) parser.add_argument('cmd', metavar='crop', help='Command') parser.add_argument('dir', metavar='dir', help='Directory to process') args = parser.parse_args() @@ -57,7 +57,7 @@ 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_{:03d}.png'.format(args.L, scene_name, i+1)) def G(x, is_train): # shape of x: [B,T_in,H,W,C] |