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

optimized target_vmaf

Browse source
This commit is contained in:
Zen 2020-07-22 20:35:59 +03:00
parent 05621ddc1f
commit b5e665f428
1 changed files with 6 additions and 13 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

19
Av1an/target_vmaf.py
View file

@ -94,7 +94,6 @@ def plot_probes(args, vmaf_cq, vmaf_target, probe, frames):
plt.title(f'Chunk: {probe.stem}, Frames: {frames}') plt.title(f'Chunk: {probe.stem}, Frames: {frames}')
plt.tight_layout() plt.tight_layout()
temp = args.temp / probe.stem temp = args.temp / probe.stem
plt.tight_layout()
plt.savefig(f'{temp}.png', dpi=200, format='png') plt.savefig(f'{temp}.png', dpi=200, format='png')
plt.close() plt.close()
@ -160,23 +159,17 @@ def target_vmaf_search(probe, source, frames, args):
vmaf_cq = [] vmaf_cq = []
q_list = [args.min_cq, args.max_cq] q_list = [args.min_cq, args.max_cq]
vmaf_steps = args.vmaf_steps - 2
score = 0 score = 0
last_q = args.min_cq
next_q = args.max_cq
for _ in range(args.vmaf_steps - 2 ):
for i in range(vmaf_steps): new_point= get_q(last_q, next_q)
if i == 0: last_q = new_point
middle_point = (args.min_cq + args.max_cq) // 2
score = vmaf_probe(probe, middle_point, args )
vmaf_cq.append((score, middle_point))
q_list.append(middle_point)
continue
last_q = vmaf_cq[-1][1]
next_q = get_closest(q_list, last_q, positive=score >= args.vmaf_target)
new_point = get_q(last_q, next_q)
q_list.append(new_point) q_list.append(new_point)
score = vmaf_probe(probe, new_point, args) score = vmaf_probe(probe, new_point, args)
next_q = get_closest(q_list, last_q, positive=score >= args.vmaf_target)
vmaf_cq.append((score, new_point)) vmaf_cq.append((score, new_point))
return vmaf_cq return vmaf_cq