gray_image = cv2.cvtColor(color,cv2.COLOR_RGB2GRAY)
gray_image = gray_image/255.0
rows,cols = gray_image.shape
M1 = np.float32([ [1,0, 1], [0,1, 0] ])
M2 = np.float32([ [1,0,-1], [0,1, 0] ])
M3 = np.float32([ [1,0, 0], [0,1,1] ])
M4 = np.float32([ [1,0, 0], [0,1,-1] ])
temp1 = cv2.warpAffine(gray_image,M1,(cols,rows), borderMode = cv2.BORDER_WRAP)
temp2 = cv2.warpAffine(gray_image,M2,(cols,rows), borderMode = cv2.BORDER_WRAP)
temp3 = cv2.warpAffine(gray_image,M3,(cols,rows), borderMode = cv2.BORDER_WRAP)
temp4 = cv2.warpAffine(gray_image,M4,(cols,rows), borderMode = cv2.BORDER_WRAP)
dx = cv2.subtract(temp1, temp2)
dy = cv2.subtract(temp3, temp4)
dxNeg = dx * -1
dyNeg = dy * -1
dxSquare = np.power(dx, 2)
dySquare = np.power(dy, 2)
nxSquareRoot = np.sqrt(dxSquare + dxSquare + 1)
nySquareRoot = np.sqrt(dySquare + dySquare + 1)
nzSquareRoot = np.sqrt(dxSquare + dxSquare + 1)
nx = np.divide(dxNeg,nxSquareRoot)
ny = np.divide(dyNeg,nySquareRoot)
nz = np.divide(dxNeg,nzSquareRoot)
R = np.divide(nx +1,2)
G = np.divide(ny +1,2)
B = np.ones(nx.shape)
r = np.floor(R * 255.0).astype('int')
g = np.floor(G * 255.0).astype('int')
b = np.floor(B * 255.0).astype('int')
new_rgb = cv2.merge((b, g, r))
return new_rgb
def newBump(color):
测试了一下,还可以。