图像像素梯度

# 编辑梯度损失函数 import numpy as np import matplotlib.pyplot as plt def grad(img): b, h, w, c = img.shape g_y=img[:,1:,:,:]-img[:,:-1,:,:] #计算高度方向梯度 g_x=img[:,:,1:,:]-img[:,:,:-1,:] #计算宽度方向梯度

>>阅读原文<<