process sequences

1 files changed, 3 insertions, 2 deletions

diff --git a/run.py b/run.py
index 4127cb3..e9f5de1 100644
--- a/run.py
+++ b/run.py
@@ -84,8 +84,8 @@ def recurse_videos(moduleNetwork, tensorOutput, a, b, a_offset, b_offset, count,
 	if (index % 10) == 0:
 		print("{}...".format(index))
 	step /= 2
-	a_img = load_image(os.path.join(a, "frame_{:0>5}.png".format(frame_index + a_offset)))
-	b_img = load_image(os.path.join(b, "frame_{:0>5}.png".format(frame_index + b_offset)))
+	a_img = load_image(os.path.join(a, "frame_{:0>5}.png".format(int(frame_index + a_offset))))
+	b_img = load_image(os.path.join(b, "frame_{:0>5}.png".format(int(frame_index + b_offset))))
 	frame = process_two_frames(moduleNetwork, tensorOutput, a, b, a_offset, b_offset, a_img, b_img, frame_index, count, count / 2)
 	if step < 2:
 		return [frame]
@@ -144,6 +144,7 @@ if arguments_strVideo and arguments_strVideoOut:
 	writer.close()
 elif arguments_mixVideos:
 	# Morph two videos
+	print("morph two videos...")
 	print("{} => {}".format(arguments_strFirst, arguments_strSecond))
 	frames = process_two_videos(moduleNetwork, tensorOutput, arguments_strFirst, arguments_strSecond, arguments_aOffset, arguments_bOffset, arguments_steps)
 	outputPath = './renders/' + arguments_strVideoOut