From 57f041f7e9e08180c9539a172197e110aa486cd5 Mon Sep 17 00:00:00 2001 From: "jules@lens" Date: Fri, 23 Nov 2018 18:18:25 +0100 Subject: upload check --- augment.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) (limited to 'augment.py') diff --git a/augment.py b/augment.py index c0f8c7a..03b8140 100755 --- a/augment.py +++ b/augment.py @@ -172,5 +172,7 @@ for m in range(data_opt.augment_take): os.unlink(frame_A) if os.path.exists(frame_B): os.unlink(frame_B) - os.symlink(os.path.abspath(next_path), os.path.abspath(frame_A)) - os.symlink(os.path.abspath(sequence[index+2]), os.path.abspath(frame_B)) + if index + 2 < len(sequence): + os.symlink(os.path.abspath(next_path), os.path.abspath(frame_A)) + os.symlink(os.path.abspath(sequence[index+2]), os.path.abspath(frame_B)) + -- cgit v1.2.3-70-g09d2