From 26dcd65a01c00a48671742a404c60e02a0476f74 Mon Sep 17 00:00:00 2001 From: Akatsumekusa <112813970+Akatmks@users.noreply.github.com> Date: Wed, 16 Apr 2025 13:08:35 +0000 Subject: [PATCH] Fix `--skip 1` for vs-zip by removing `SelectEvery` (#13) --- auto-boost_2.5.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/auto-boost_2.5.py b/auto-boost_2.5.py index 6a55044..746ee1e 100755 --- a/auto-boost_2.5.py +++ b/auto-boost_2.5.py @@ -220,8 +220,12 @@ def calculate_ssimu2(src_file, enc_file, ssimu2_txt_path, ranges, skip): iter = 0 with tqdm(total=floor(len(source_clip)), desc=f'Calculating SSIMULACRA 2 scores') as pbar: for i in range(len(ranges) - 1): - cut_source_clip = source_clip[ranges[i]:ranges[i+1]].std.SelectEvery(cycle=skip, offsets=1) - cut_encoded_clip = encoded_clip[ranges[i]:ranges[i+1]].std.SelectEvery(cycle=skip, offsets=1) + if skip > 1: + cut_source_clip = source_clip[ranges[i]:ranges[i+1]].std.SelectEvery(cycle=skip, offsets=1) + cut_encoded_clip = encoded_clip[ranges[i]:ranges[i+1]].std.SelectEvery(cycle=skip, offsets=1) + else: + cut_source_clip = source_clip[ranges[i]:ranges[i+1]] + cut_encoded_clip = encoded_clip[ranges[i]:ranges[i+1]] result = core.vszip.Metrics(cut_source_clip, cut_encoded_clip, mode=0) for index, frame in enumerate(result.frames()): iter += 1