diff options
| -rw-r--r-- | run.py | 4 |
1 files changed, 3 insertions, 1 deletions
@@ -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)) |