From ecef678539202b2ad66c2a05995e26ccb7d4bf07 Mon Sep 17 00:00:00 2001 From: Jules Laplace Date: Mon, 24 Sep 2018 14:57:57 +0200 Subject: generate from full dataset --- augment.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) 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...") -- cgit v1.2.3-70-g09d2