summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--test-mogrify.py6
1 files changed, 4 insertions, 2 deletions
diff --git a/test-mogrify.py b/test-mogrify.py
index 700f60f..c6deaa3 100644
--- a/test-mogrify.py
+++ b/test-mogrify.py
@@ -56,13 +56,17 @@ if __name__ == '__main__':
cv2.imwrite(start_img_path, A)
numz = re.findall(r'\d+', os.path.basename(opt.start_img))
+ print(numz)
if len(numz) > 0:
A_offset = int(numz[0])
+ print(A_offset)
if A_offset:
print(">> starting offset: {}".format(A_offset))
A_dir = opt.start_img.replace(numz[0], "{:05d}")
+ print(A_dir)
else:
print("Sequence not found")
+ sys.exit(1)
def process_image(im):
img = im[:, :, ::-1].copy()
@@ -177,8 +181,6 @@ if __name__ == '__main__':
tmp_path = render_dir + "frame_{:05d}_tmp.png".format(i+1)
edges_path = render_dir + "frame_{:05d}.png".format(i+1)
render_path = render_dir + "ren_{:05d}.png".format(i+1)
- print(render_path)
- print(edges_path)
if A_dir is not None:
sequence_path = A_dir.format(A_offset+i+1)
print(sequence_path)