颜色识别代码,python
时间: 2023-10-17 08:02:17 浏览: 44
以下是Python的颜色识别代码,需要安装OpenCV库和numpy库:
```python
import cv2
import numpy as np
# 定义颜色范围
lower_red = np.array([0,50,50])
upper_red = np.array([10,255,255])
lower_yellow = np.array([26,43,46])
upper_yellow = np.array([34,255,255])
lower_green = np.array([35,43,46])
upper_green = np.array([77,255,255])
lower_blue = np.array([100,43,46])
upper_blue = np.array([124,255,255])
lower_purple = np.array([125,43,46])
upper_purple = np.array([155,255,255])
# 加载图像并转换为HSV格式
img = cv2.imread("test.jpg")
hsv = cv2.cvtColor(img, cv2.COLOR_BGR2HSV)
# 根据颜色范围进行颜色识别
mask_red = cv2.inRange(hsv, lower_red, upper_red)
mask_yellow = cv2.inRange(hsv, lower_yellow, upper_yellow)
mask_green = cv2.inRange(hsv, lower_green, upper_green)
mask_blue = cv2.inRange(hsv, lower_blue, upper_blue)
mask_purple = cv2.inRange(hsv, lower_purple, upper_purple)
# 合并颜色掩模
mask = mask_red + mask_yellow + mask_green + mask_blue + mask_purple
# 根据掩模提取图像中的颜色区域
res = cv2.bitwise_and(img, img, mask=mask)
# 显示原始图像和颜色识别结果
cv2.imshow("Original Image", img)
cv2.imshow("Color Detection Result", res)
cv2.waitKey(0)
cv2.destroyAllWindows()
```
在此代码中,我们将图像转换为HSV格式,并使用`cv2.inRange()`函数根据颜色范围创建颜色掩模。然后,我们将所有颜色掩模合并,并使用`cv2.bitwise_and()`函数根据掩模提取图像中的颜色区域。最后,我们使用`cv2.imshow()`函数显示原始图像和颜色识别结果。
相关推荐
![py](https://img-home.csdnimg.cn/images/20210720083646.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)