ayo
7c5d3c9157
Add native K/M/RR inference engines, HIP kernels, replay and validation harnesses, and Windows DLSS setup and runtime support. Include GPU profiling tools, build and packaging scripts, platform integration patches, and implementation documentation.
16 lines
871 B
Python
16 lines
871 B
Python
#!/usr/bin/env python3
|
|
"""psnr.py DIR_A DIR_B: per-frame PSNR of rgba16f outputs (watermark box excluded, values clipped to [0, 1] after tonemap-free clamp)."""
|
|
import sys, glob, os, numpy as np
|
|
a, b = sys.argv[1:3]
|
|
vals = []
|
|
for f in sorted(glob.glob(a + '/frame-*.rgba16f')):
|
|
g = os.path.join(b, os.path.basename(f))
|
|
if not os.path.exists(g): continue
|
|
x = np.fromfile(f, np.float16).reshape(1440, 2560, 4)[..., :3].astype(np.float64)
|
|
y = np.fromfile(g, np.float16).reshape(1440, 2560, 4)[..., :3].astype(np.float64)
|
|
m = np.ones(x.shape[:2], bool); m[1370:1392, 1850:2460] = False
|
|
x, y = np.clip(x[m], 0, 1), np.clip(y[m], 0, 1)
|
|
mse = np.mean((x - y) ** 2)
|
|
vals.append(10 * np.log10(1.0 / mse) if mse > 0 else 99.0)
|
|
print('frames', len(vals), 'PSNR dB min %.2f mean %.2f' % (min(vals), np.mean(vals)), ' '.join('%.1f' % v for v in vals))
|