How to implement SNR -As a metric function Keras?

Question

I am implementing a denoising autoencoder where I want to use SNR as a metric function. But how can I implement it?

def PSNR(y_true, y_pred):

max_I = 1.0
val = 2