generate from full dataset

1 files changed, 4 insertions, 1 deletions

diff --git a/augment.py b/augment.py
index b5894e1..e1360b0 100755
--- a/augment.py
+++ b/augment.py
@@ -99,7 +99,10 @@ if not opt.engine and not opt.onnx:
 
 sequence = read_sequence(data_opt.sequence_name, '')
 print("Got sequence {}, {} images".format(data_opt.sequence_name, len(sequence)))
-_len = len(sequence) - data_opt.augment_take - 2
+if data_opt.no_symlinks:
+  _len = len(sequence) - data_opt.augment_take - 2
+else:
+  _len = len(sequence)
 
 if _len <= 0:
   print("Got empty sequence...")