pat-e/auto-boost-algorithm
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Add custom encoder params (#9)

Browse Source
This commit is contained in:
EdgeLord69-dev
2025-02-22 00:52:44 +05:30
committed by GitHub
parent 1ba1b63eaf
commit 8626360906
1 changed files with 23 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

39
auto-boost_2.5.py
View File

@@ -42,6 +42,7 @@ parser.add_argument("-m", "--metrics", help = "Select metrics: 1 = SSIMU2, 2 = X
parser.add_argument("-S", "--skip", help = "SSIMU2 skip value, every nth frame's SSIMU2 is calculated | Default: 1 for turbo-metrics, 3 for vs-zip")
parser.add_argument("-z", "--zones", help = "Zones calculation method: 1 = SSIMU2, 2 = XPSNR, 3 = Multiplication, 4 = Lowest Result | Default: 1", default=1)
parser.add_argument("-a", "--aggressive", action='store_true', help = "More aggressive boosting | Default: not active")
parser.add_argument("-v","--video_params", help="Custom encoder parameters for av1an")
args = parser.parse_args()
stage = int(args.stage)
src_file = Path(args.input).resolve()
@@ -52,6 +53,7 @@ scenes_file = tmp_dir / "scenes.json"
br = float(args.deviation)
skip = int(args.skip) if args.skip is not None else default_skip
aggressive = args.aggressive
video_params = args.video_params
def get_ranges(scenes: str) -> list[int]:
"""
@@ -71,7 +73,7 @@ def get_ranges(scenes: str) -> list[int]:
return ranges
def fast_pass(
input_file: str, output_file: str, tmp_dir: str, preset: int, crf: float, workers: int
input_file: str, output_file: str, tmp_dir: str, preset: int, crf: float, workers: int,video_params: str
):
"""
Quick fast-pass using Av1an
@@ -88,6 +90,8 @@ def fast_pass(
:type crf: float
:param workers: number of workers
:type workers: int
:param video_params: custom encoder params for av1an
:type video_prams: str
"""
fast_av1an_command = [
@@ -95,16 +99,16 @@ def fast_pass(
'-i', input_file,
'--temp', tmp_dir,
'-y',
'--verbose',
'--verbose',
'--keep',
'-m', 'lsmash',
'-c', 'mkvmerge',
'--min-scene-len', '24',
'--sc-downscale-height', '720',
'--sc-downscale-height', '720',
'--set-thread-affinity', '2',
'-e', 'svt-av1',
'--force',
'-v', f'--preset {preset} --crf {crf:.2f} --lp 2 --scm 0 --keyint 0 --fast-decode 1 --color-primaries 1 --transfer-characteristics 1 --matrix-coefficients 1',
'-v', f'--preset {preset} --crf {crf:.2f} --lp 2 --keyint 0 --scm 0 --fast-decode 1 --color-primaries 1 --transfer-characteristics 1 --matrix-coefficients 1 {video_params}',
'-w', str(workers),
'-o', output_file
]
@@ -292,7 +296,7 @@ def calculate_std_dev(score_list: list[int]):
percentile_95 = sorted_score_list[int (len(filtered_score_list)//(20/19))]
return (average, percentile_5, percentile_95)
def generate_zones(ranges: list, percentile_5_total: list, average: int, crf: float, zones_txt_path: str):
def generate_zones(ranges: list, percentile_5_total: list, average: int, crf: float, zones_txt_path: str, video_params: str):
"""
Appends a scene change to the ``zones_txt_path`` file in Av1an zones format.
@@ -309,6 +313,8 @@ def generate_zones(ranges: list, percentile_5_total: list, average: int, crf: fl
:type crf: int
:param zones_txt_path: Path to the zones.txt file
:type zones_txt_path: str
:param video_params: custom encoder params for av1an
:type video_prams: str
"""
zones_iter = 0
for i in range(len(ranges)-1):
@@ -329,7 +335,7 @@ def generate_zones(ranges: list, percentile_5_total: list, average: int, crf: fl
f'Adjusted CRF: {new_crf:.2f}\n')
with zones_txt_path.open("w" if zones_iter == 1 else "a") as file:
file.write(f"{ranges[i]} {ranges[i+1]} svt-av1 --crf {new_crf:.2f}\n")
file.write(f"{ranges[i]} {ranges[i+1]} svt-av1 --crf {new_crf:.2f} --lp 2 {video_params}\n")
def calculate_metrics(src_file, output_file, tmp_dir, ranges, skip, metrics):
match metrics:
@@ -345,7 +351,7 @@ def calculate_metrics(src_file, output_file, tmp_dir, ranges, skip, metrics):
calculate_xpsnr(src_file, output_file, xpsnr_txt_path)
calculate_ssimu2(src_file, output_file, ssimu2_txt_path, ranges, skip)
def calculate_zones(tmp_dir, ranges, zones, cq):
def calculate_zones(tmp_dir, ranges, zones, cq, video_params):
match zones:
case 1:
ssimu2_txt_path = output_dir / f"{src_file.stem}_ssimu2.log"
@@ -373,7 +379,7 @@ def calculate_zones(tmp_dir, ranges, zones, cq):
print(f'Median score: {ssimu2_average}')
print(f'5th Percentile: {ssimu2_percentile_5}')
print(f'95th Percentile: {ssimu2_percentile_95}\n')
generate_zones(ranges, ssimu2_percentile_5_total, ssimu2_average, cq, ssimu2_zones_txt_path)
generate_zones(ranges, ssimu2_percentile_5_total, ssimu2_average, cq, ssimu2_zones_txt_path, video_params)
case 2:
xpsnr_txt_path = output_dir / f"{src_file.stem}_xpsnr.log"
@@ -399,7 +405,7 @@ def calculate_zones(tmp_dir, ranges, zones, cq):
print(f'Median score: {xpsnr_average}')
print(f'5th Percentile: {xpsnr_percentile_5}')
print(f'95th Percentile: {xpsnr_percentile_95}\n')
generate_zones(ranges, xpsnr_percentile_5_total, xpsnr_average, cq, xpsnr_zones_txt_path)
generate_zones(ranges, xpsnr_percentile_5_total, xpsnr_average, cq, xpsnr_zones_txt_path, video_params)
case 3:
ssimu2_txt_path = output_dir / f"{src_file.stem}_ssimu2.log"
@@ -433,7 +439,7 @@ def calculate_zones(tmp_dir, ranges, zones, cq):
print(f'Median score: {multiplied_average}')
print(f'5th Percentile: {multiplied_percentile_5}')
print(f'95th Percentile: {multiplied_percentile_95}\n')
generate_zones(ranges, multiplied_percentile_5_total, multiplied_average, cq, multiplied_zones_txt_path)
generate_zones(ranges, multiplied_percentile_5_total, multiplied_average, cq, multiplied_zones_txt_path, video_params)
case 4:
@@ -474,33 +480,34 @@ def calculate_zones(tmp_dir, ranges, zones, cq):
print(f'Median score: {minimum_average}')
print(f'5th Percentile: {minimum_percentile_5}')
print(f'95th Percentile: {minimum_percentile_95}\n')
generate_zones(ranges, minimum_percentile_5_total, minimum_average, cq, minimum_zones_txt_path)
generate_zones(ranges, minimum_percentile_5_total, minimum_average, cq, minimum_zones_txt_path, video_params)
match stage:
case 0:
workers = args.workers
crf = float(args.quality)
preset = args.preset
fast_pass(src_file, output_file, tmp_dir, preset, crf, workers)
video_params = args.video_params
fast_pass(src_file, output_file, tmp_dir, preset, crf, workers, video_params)
ranges = get_ranges(scenes_file)
metrics = int(args.metrics)
calculate_metrics(src_file, output_file, tmp_dir, ranges, skip, metrics)
zones = int(args.zones)
calculate_zones(tmp_dir, ranges, zones, crf)
calculate_zones(tmp_dir, ranges, zones, crf, video_params)
case 1:
workers = args.workers
crf = float(args.quality)
preset = args.preset
fast_pass(src_file, output_file, tmp_dir, preset, crf, workers)
fast_pass(src_file, output_file, tmp_dir, preset, crf, workers, video_params)
case 2:
ranges = get_ranges(scenes_file)
metrics = int(args.metrics)
calculate_metrics(src_file, output_file, tmp_dir, ranges, skip, metrics)
calculate_metrics(src_file, output_file, tmp_dir, ranges, skip, metrics, video_params)
case 3:
ranges = get_ranges(scenes_file)
zones = int(args.zones)
crf = float(args.quality)
calculate_zones(tmp_dir, ranges, zones, crf)
calculate_zones(tmp_dir, ranges, zones, crf, video_params)
case _:
print(f"Stage argument invalid, exiting.")
exit(-2)