aaappp7878 commited on
Commit
c68418d
·
verified ·
1 Parent(s): f16fd05

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +12 -3
app.py CHANGED
@@ -66,9 +66,18 @@ def analyze_image_features(image):
66
  # 纹理分析 - 使用灰度共生矩阵
67
  if len(img_array.shape) == 3:
68
  gray = cv2.cvtColor(img_cv, cv2.COLOR_BGR2GRAY)
69
- glcm = cv2.createGLCM(gray, 5) # 创建灰度共生矩阵
70
- features["texture_contrast"] = cv2.GLCMContrast(glcm)
71
- features["texture_homogeneity"] = cv2.GLCMHomogeneity(glcm)
 
 
 
 
 
 
 
 
 
72
 
73
  # 噪声分析
74
  if len(img_array.shape) == 3:
 
66
  # 纹理分析 - 使用灰度共生矩阵
67
  if len(img_array.shape) == 3:
68
  gray = cv2.cvtColor(img_cv, cv2.COLOR_BGR2GRAY)
69
+ from skimage.feature import graycomatrix, graycoprops
70
+
71
+ # 计算GLCM
72
+ distances = [5]
73
+ angles = [0, np.pi/4, np.pi/2, 3*np.pi/4]
74
+ glcm = graycomatrix(gray, distances=distances, angles=angles, symmetric=True, normed=True)
75
+
76
+ # 计算GLCM属性
77
+ features["texture_contrast"] = float(np.mean(graycoprops(glcm, 'contrast')[0]))
78
+ features["texture_homogeneity"] = float(np.mean(graycoprops(glcm, 'homogeneity')[0]))
79
+
80
+
81
 
82
  # 噪声分析
83
  if len(img_array.shape) == 3: