From b5cb2dab354ba999c2dc5de7ec820997535cb3ed Mon Sep 17 00:00:00 2001 From: Jules Laplace Date: Mon, 25 Jun 2018 18:12:23 +0200 Subject: process sequences --- run.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/run.py b/run.py index 39aeab3..d234b9d 100644 --- a/run.py +++ b/run.py @@ -61,9 +61,9 @@ def recurse_two_frames(moduleNetwork, tensorOutput, a_np, b_np, frame_index, mor tensorInputFirst = torch.FloatTensor(a_np) tensorInputSecond = torch.FloatTensor(b_np) + print("{} {}".format(tensorInputFirst.size(1), tensorInputSecond.size(1))) process(moduleNetwork, tensorInputFirst, tensorInputSecond, tensorOutput) middle_np = tensorOutput.clamp(0.0, 1.0).numpy() - middle_img = (numpy.rollaxis(middle_np, 0, 3)[:,:,::-1] * 255.0).astype(numpy.uint8) if morph_index == frame_index: print("frame {}, depth {}".format(frame_index, depth)) @@ -126,8 +126,10 @@ def store_frames(frames, outputPath): for frame in frames: writer.write_frame(frame) writer.write_frame(inputSecond) + def load_image(path): return numpy.rollaxis(numpy.asarray(PIL.Image.open(path))[:,:,::-1], 2, 0).astype(numpy.float32) / 255.0 + def load_image_tensor(path): return torch.FloatTensor(load_image(path)) -- cgit v1.2.3-70-g09d2