1 files changed, 4 insertions, 2 deletions
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))
+