python实现PSNR变换

import cv2
import numpy as np
import matplotlib.pyplot as plt

# DCT hyoer-parameter
T = 8
K = 4
channel = 3

# DCT weight
def w（x y u v）:
    cu = 1.
    cv = 1.
    if u == 0:
        cu /= np.sqrt（2）
    if v == 0:
        cv /= np.sqrt（2）
    theta = np.pi / （2 * T）
    return （（ 2 * cu * cv / T） * np.cos（（2*x+1）*u*theta） * np.cos（（2*y+1）*v*theta））

# DCT
def dct（img）:
    H W _ = img.shape

    F = np.zeros（（H W channel） dtype=np.float32）

    for c in range（channel）:
        for yi in range（0 H T）:
            for xi in range（0 W T）:
                for v in range（T）:
                    for u in range（T）:
                        for y in range（T）:
                            for x in range（T）:
                                F[v+yi u+xi c] += img[y+yi x+xi c] * w（xyuv）

    return F


# IDCT
def idct（F）:
    H W _ = F.shape

    out = np.zeros（（H W channel） dtype=np.float32）

    for c in range（channel）:
        for yi in range（0 H T）:
            for x