From de7d9fb18fcc8a43e2d365203904514e89dd414e Mon Sep 17 00:00:00 2001 From: Jules Laplace Date: Thu, 6 Sep 2018 11:24:21 +0200 Subject: fix augment --- augment.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) (limited to 'augment.py') diff --git a/augment.py b/augment.py index 1d44fe2..e34d5cd 100755 --- a/augment.py +++ b/augment.py @@ -87,7 +87,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))) -_len = len(sequence) - data_opt.augment_take +_len = len(sequence) - data_opt.augment_take - 1 if _len <= 0: print("Got empty sequence...") @@ -149,6 +149,6 @@ for m in range(data_opt.augment_take): if os.path.exists(frame_B): os.unlink(frame_B) os.symlink(os.path.abspath(next_path), os.path.abspath(frame_A)) - os.symlink(os.path.abspath(sequence[i+1]), os.path.abspath(frame_B)) + os.symlink(os.path.abspath(sequence[index+1]), os.path.abspath(frame_B)) -- cgit v1.2.3-70-g09d2