paths

1 files changed, 3 insertions, 2 deletions

diff --git a/augment.py b/augment.py
index 4aab5f1..1b32483 100755
--- a/augment.py
+++ b/augment.py
@@ -153,6 +153,9 @@ for m in range(data_opt.augment_take):
     image_pil.save(tmp_path)
     os.rename(tmp_path, next_path)
 
+    if data_opt.no_symlink:
+      continue
+
     frame_A = os.path.join("./datasets/", data_opt.sequence_name, "train_A", "recur_{}_{:05d}_{:05d}.png".format(epoch_id, m, index+1))
     frame_B = os.path.join("./datasets/", data_opt.sequence_name, "train_B", "recur_{}_{:05d}_{:05d}.png".format(epoch_id, m, index+1))
     if os.path.exists(frame_A):
@@ -161,5 +164,3 @@ for m in range(data_opt.augment_take):
       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))
-
-