summaryrefslogtreecommitdiff
path: root/augment.py
diff options
context:
space:
mode:
Diffstat (limited to 'augment.py')
-rwxr-xr-xaugment.py2
1 files changed, 1 insertions, 1 deletions
diff --git a/augment.py b/augment.py
index c0f8c7a..e330328 100755
--- a/augment.py
+++ b/augment.py
@@ -100,7 +100,7 @@ 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)))
if data_opt.no_symlinks:
- _len = len(sequence) - data_opt.augment_take - 2
+ _len = len(sequence) - data_opt.augment_make - 2
else:
_len = len(sequence)