forktown/Av1an
1
0
Fork 0
mirror of https://github.com/master-of-zen/Av1an.git synced 2024-11-25 02:29:40 +00:00
Code Issues Projects Releases Packages Wiki Activity

per frame target quality wip

Browse source
This commit is contained in:
Zen 2020-12-12 05:36:24 +02:00
parent f937839097
commit 7f89fdb2bf
1 changed files with 16 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

18
TargetQuality/per_frame.py
View file

@ -87,7 +87,7 @@ def add_probes_to_frame_list(frame_list, q_list, vmafs):
def per_frame_target_quality(chunk, project): def per_frame_target_quality(chunk, project):
frames = chunk.frames frames = chunk.frames
#pp = pprint.PrettyPrinter(indent=4).pprint pp = pprint.PrettyPrinter(indent=4).pprint
# First q value to make probe at # First q value to make probe at
middle_point = (project.min_q + project.max_q) // 2 middle_point = (project.min_q + project.max_q) // 2
frame_list = [{'frame_number': x, 'probes': []} for x in range(frames)] frame_list = [{'frame_number': x, 'probes': []} for x in range(frames)]
@ -104,10 +104,23 @@ def per_frame_target_quality(chunk, project):
vmafs = per_frame_probe(q_list, 1, chunk, project) vmafs = per_frame_probe(q_list, 1, chunk, project)
frame_list = add_probes_to_frame_list(frame_list, q_list, vmafs) frame_list = add_probes_to_frame_list(frame_list, q_list, vmafs)
#pp(frame_list) print(get_square_error([x['probes'][-1][1] for x in frame_list] ,project.target_quality))
pp(frame_list)
exit() exit()
def get_square_error(ls, target):
total = 0
for i in ls:
dif = i - target
total += dif ** 2
mse = total / len(ls)
return mse
def gen_border_probes_q(frame_list, min_q, max_q, target): def gen_border_probes_q(frame_list, min_q, max_q, target):
q_list = [] q_list = []
@ -143,6 +156,7 @@ def search(q1, v1, q2, v2, target):
def gen_next_q(frame_list, chunk, project): def gen_next_q(frame_list, chunk, project):
q_list = [] q_list = []
for probe in frame_list: for probe in frame_list:
p1, p2 = probe['probes'][-2:] p1, p2 = probe['probes'][-2:]
q_list.append(search(p1[0],p1[1],p2[0],p2[1], project.target_quality)) q_list.append(search(p1[0],p1[1],p2[0],p2[1], project.target_quality))