由于某些原因，需要計算圖像的曲率。找了半天，網上都是曲率濾波的代碼，而沒有計算圖像曲率的代碼。于是去找曲率的計算公式，發現公式很簡單，所以自己就用python寫了一下。
平均曲率的計算公式如下所示

python代碼如下所示：

import numpy as np import cv2 from skimage import color import matplotlib.pyplot as plt import imageio img = cv2.imread("xxx.jpg") img = color.rgb2gray(img) x , y = np.gradient(img) #一階導數 xx, xy = np.gradient(x)　　#二階偏導數 yx, yy = np.gradient(y)　　　#二階偏導數 Iup = (1+x\*x)\*yy - 2\*x\*y\*xy + (1+y\*y)\*xx　　#公式的分子 Idown = np.power((2\*(1 + x\*x + y\*y)),1.5) #公式的分母 final = Iup/Idown final=abs(final) final = (final-final.min())/(final.max()-final.min())　　　　#將結果歸一化 final = final * 255　　　　　#將像素值擴展為０-255 final = final.astype(np.uint8) plt.imshow(final) imageio.write("xxx.jpg",final) #保存圖像

寫成函數如下所示：

def cal_curvature(img):x , y = np.gradient(img)xx, xy = np.gradient(x)yx, yy = np.gradient(y)Iup = (1+x\*x)\*yy - 2\*x\*y\*xy + (1+y\*y)\*xxIdown = np.power((2\*(1 + x\*x + y\*y)),1.5) final = Iup/Idownfinal=abs(final)final = (final-final.min())/(final.max()-final.min())final = final * 255final = final.astype(np.uint8) return final

總結

以上是生活随笔為你收集整理的python计算图像的曲率的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。