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

Vmaf plot improvements

Browse source
This commit is contained in:
Zen 2020-08-25 23:45:08 +03:00
parent e289896a4d
commit 8bf69cfd7b
1 changed files with 15 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

16
Av1an/vmaf.py
View file

@ -8,6 +8,7 @@ from pathlib import Path
from subprocess import PIPE, STDOUT
import numpy as np
from math import log10
from matplotlib import pyplot as plt
from .bar import process_pipe
@ -107,7 +108,20 @@ def plot_vmaf_score_file(scores: Path, plot_path: Path):
vmafs = [x['metrics']['vmaf'] for x in file['frames']]
plot_size = len(vmafs)
plt.figure(figsize=(15, 4))
figure_width = 3 + round((4 * log10(plot_size)))
plt.figure(figsize=(figure_width, 5))
plt.plot([1, plot_size], [perc_1, perc_1], '-', color='red')
plt.annotate(f'1%: {perc_1}', xy=(0, perc_1), color='red')
plt.plot([1, plot_size], [perc_25, perc_25], ':', color='orange')
plt.annotate(f'25%: {perc_25}', xy=(0, perc_25), color='orange')
plt.plot([1, plot_size], [perc_75, perc_75], ':', color='green')
plt.annotate(f'75%: {perc_75}', xy=(0, perc_75), color='green')
plt.plot([1, plot_size], [mean, mean], ':', color='black')
plt.annotate(f'Mean: {mean}', xy=(0, mean), color='black')
for i in range(0, 100):
plt.axhline(i, color='grey', linewidth=0.4)