Parcourir la source

add stats into extended stats

Jérôme BUISINE il y a 3 ans
Parent
commit
2ec9dc37a9
1 fichiers modifiés avec 2 ajouts et 0 suppressions
  1. 2 0
      data_attributes.py

+ 2 - 0
data_attributes.py

@@ -110,6 +110,8 @@ def get_image_features(data_type, block):
 
         sobel_mag = np.array(np.hypot(sobelx, sobely), 'uint8')  # magnitude
 
+        data = np.append(data, np.std(sobel_mag))
+        
         # add sobel complexity (kernel size of 5)
         sobelx = cv2.Sobel(arr, cv2.CV_64F, 1, 0, ksize=5)
         sobely = cv2.Sobel(arr, cv2.CV_64F, 0, 1,ksize=5)