Update file main.py

phs/main.py

@@ -23,7 +23,7 @@ def hue2rgb_abs(hue):
 # and calculate average, prevailing hue and its percentage
 # if whole image is gray, image doesn't participate in sorting and plotting
 def get_image_info(image_path):
-    bgr_sequence = cv2.imdecode(np.fromfile(image_path, dtype=np.uint8), cv2.IMREAD_UNCHANGED).reshape(-1, 3)
+    bgr_sequence = cv2.imdecode(np.fromfile(image_path, dtype=np.uint8), cv2.IMREAD_UNCHANGED)[:, :, 0:3].reshape(-1, 3)
     rgb_average_color, hsv_average_color = seq2mean(bgr_sequence)
 
     gray_indices = np.all(bgr_sequence == bgr_sequence[:, 0][:, np.newaxis], axis=1)