Example Python3 script showing the parsing of libvmaf JSON format output logs. All AOM CTC metrics are included.

parse_vmaf_log.py

import json
import sys

for vmaf_log in sys.argv[1:]:
    with open(vmaf_log) as f:
        data = json.load(f)

        per_frame_metrics = {}

        per_frame_metrics["psnr_y"] = \
            [frame['metrics']['psnr_y'] for frame in data['frames']]
        per_frame_metrics["psnr_cb"] = \
            [frame['metrics']['psnr_cb'] for frame in data['frames']]
        per_frame_metrics["psnr_cr"] = \
            [frame['metrics']['psnr_cr'] for frame in data['frames']]
        per_frame_metrics["psnr_hvs"] = \
            [frame['metrics']['psnr_hvs'] for frame in data['frames']]
        per_frame_metrics["ssim"] = \
            [frame['metrics']['float_ssim'] for frame in data['frames']]
        per_frame_metrics["ms_ssim"] = \
            [frame['metrics']['float_ssim'] for frame in data['frames']]
        per_frame_metrics["ciede2000"] = \
            [frame['metrics']['ciede2000'] for frame in data['frames']]
        per_frame_metrics["vmaf"] = \
            [frame['metrics']['vmaf'] for frame in data['frames']]

        pooled_metrics = {}

        pooled_metrics["psnr_y"] = \
            data["pooled_metrics"]["psnr_y"]["mean"]
        pooled_metrics["psnr_cb"] = \
            data["pooled_metrics"]["psnr_cb"]["mean"]
        pooled_metrics["psnr_cr"] = \
            data["pooled_metrics"]["psnr_cr"]["mean"]
        pooled_metrics["psnr_hvs"] = \
            data["pooled_metrics"]["psnr_hvs"]["mean"]
        pooled_metrics["float_ssim"] = \
            data["pooled_metrics"]["float_ssim"]["mean"]
        pooled_metrics["float_ms_ssim"] = \
            data["pooled_metrics"]["float_ms_ssim"]["mean"]
        pooled_metrics["ciede2000"] = \
            data["pooled_metrics"]["ciede2000"]["mean"]
        pooled_metrics["vmaf"] = \
            data["pooled_metrics"]["vmaf"]["mean"]

        aggregate_metrics = {}

        aggregate_metrics["aspnr_y"] = \
            data["aggregate_metrics"]["apsnr_y"]
        aggregate_metrics["aspnr_cb"] = \
            data["aggregate_metrics"]["apsnr_cb"]
        aggregate_metrics["aspnr_cr"] = \
            data["aggregate_metrics"]["apsnr_cr"]

        print(per_frame_metrics)
        print(pooled_metrics)
        print(aggregate_metrics)