用python实现比色功能

老清洁工

python做脚本是很方便的，和模拟器配合简直是绝了。
下面是我写的比色代码，各位觉得如何？在此基础上，要实现找色也是很不错的，请期待我的下一篇文章。

1. # 由于我们习惯性的这样写颜色值，所以写个函数专门转换一下
   
2. # 0xRRGGBB
   
3. def color2array(c):
   
4.     r = (c & 0xFF0000) >> 16
   
5.     g = (c & 0x00FF00) >> 8
   
6.     b = (c & 0x0000FF)
   
7.     return b, g, r  # opencv读入内存的格式是BGR，所以倒着返回
   
8. 
   
9. 
   
10. # 单个点对比，比色和找色都会用到
    
11. def color_cmp(image, x, y, b1, g1, r1, sim):
    
12.     b2 = image.item(y, x, 0)
    
13.     g2 = image.item(y, x, 1)
    
14.     r2 = image.item(y, x, 2)
    
15.     sim = (1 - sim) * 255
    
16.     return abs(b2 - b1) < sim and abs(g2 - g1) < sim and abs(r2 - r1) < sim
    
17. 
    
18. 
    
19. # 多点比色
    
20. def mcc(image, color_list, sim=0.9):
    
21.     if len(color_list) == 0:
    
22.         return False
    
23.     for c in color_list:
    
24.         b1, g1, r1 = color2array(c[2])
    
25.         if not color_cmp(image, c[0], c[1], b1, g1, r1, sim):
    
26.             return -1, -1
    
27.     return color_list[0][0], color_list[0][1]

复制代码

打个小广告：
作者公众号：老清洁工
二维码：
二维码：

老清洁工

看来一楼还是要自己占了

shenxian666

2楼，楼主666