Merge branch 'master' of asdf.us:pix2pixhd

1 files changed, 3 insertions, 2 deletions

diff --git a/augment.py b/augment.py
index 87d5717..81dfc8c 100755
--- a/augment.py
+++ b/augment.py
@@ -157,6 +157,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):
@@ -165,5 +168,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))
-
-