process sequences

1 files changed, 3 insertions, 1 deletions

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))