东方耀AI技术分享

标题: 如何区分彩色图片和灰度图片（同样都是3通道的情况） [打印本页]

---

作者: 东方耀    时间: 2020-9-23 17:14
标题: 如何区分彩色图片和灰度图片（同样都是3通道的情况）


如何区分彩色图片和灰度图片（同样都是3通道的情况）

1. # -*- coding: utf-8 -*-
   
2. __author__ = u'江锦炬 微信：715781570'
   
3. __date__ = '2020/9/23 下午4:42'
   
4. __product__ = 'PyCharm'
   
5. __filename__ = 'dfy_test_true_person_pass_rate'
   
6. 
   
7. 
   
8. import os
   
9. import cv2
   
10. import numpy as np
    
11. 
    
12. # 如何区分彩色图片和灰度图片（同样都是3通道的情况）
    
13. 
    
14. def is_Gray(chip):
    
15.     # chip_gray = cv2.cvtColor(chip,cv2.COLOR_BGR2GRAY)
    
16.     r, g, b = cv2.split(chip)
    
17.     r = r.astype(np.float32)
    
18.     g = g.astype(np.float32)
    
19.     b = b.astype(np.float32)
    
20.     s_w, s_h = r.shape[:2]
    
21.     x = (r + b + g) / 3
    
22. 
    
23.     area_s = s_w * s_h
    
24.     # x = chip_gray
    
25.     r_gray = abs(r - x)
    
26.     g_gray = abs(g - x)
    
27.     b_gray = abs(b - x)
    
28.     r_sum = np.sum(r_gray) / area_s
    
29.     g_sum = np.sum(g_gray) / area_s
    
30.     b_sum = np.sum(b_gray) / area_s
    
31.     gray_degree = (r_sum + g_sum + b_sum) / 3
    
32.     if gray_degree < 1.0:
    
33.         # 灰度图片
    
34.         return True, gray_degree
    
35.     else:
    
36.         # 彩色图片
    
37.         return False, gray_degree
    
38. 
    
39. 
    
40. 
    
41. img_path = "/media/jiang/AI_DataSets/DataSets/nir_rgb_original_data/sw-hdr-sm/rgb_sw_hdr_sm_20200607/001X29/1.jpg"
    
42. img_path1 = "/media/jiang/AI_DataSets/DataSets/nir_rgb_original_data/sw-hdr-sm/rgb_sw_hdr_sm_20200607/001X29/2.jpg"
    
43. 
    
44. img = cv2.imread(img_path)
    
45. 
    
46. print(img.shape)
    
47. 
    
48. cv2.imshow("ori", img)
    
49. cv2.waitKey(0)
    
50. 
    
51. 
    
52. ret, value = is_Gray(img)
    
53. # 彩色图片 结果： False 8.359733153292181
    
54. # 灰度图片 结果： True 0.03208879886831276
    
55. print("结果：", ret, value)
    
56. 
    
57. 
    
58. 
    

复制代码

---

欢迎光临 东方耀AI技术分享 (http://www.ai111.vip/)

Powered by Discuz! X3.4