From ab18a90b1a5c6d12faaa979e56903eaca93a7b0c Mon Sep 17 00:00:00 2001 From: Jules Laplace Date: Fri, 7 Sep 2018 15:21:27 +0200 Subject: augment plus 2 --- augment.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) (limited to 'augment.py') diff --git a/augment.py b/augment.py index a9ba885..b01f04c 100755 --- a/augment.py +++ b/augment.py @@ -94,7 +94,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 - 1 +_len = len(sequence) - data_opt.augment_take - 2 if _len <= 0: print("Got empty sequence...") @@ -160,6 +160,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[index+1]), os.path.abspath(frame_B)) + os.symlink(os.path.abspath(sequence[index+2]), os.path.abspath(frame_B)) -- cgit v1.2.3-70-g09d2