用 python写一段识别苹果病虫害的代码
时间: 2023-05-16 16:07:47 浏览: 62
以下是一段使用 Python 识别苹果病虫害的代码:
```python
import cv2
import numpy as np
# 加载图像
img = cv2.imread('apple.jpg')
# 转换为 HSV 颜色空间
hsv = cv2.cvtColor(img, cv2.COLOR_BGR2HSV)
# 定义绿色和红色的范围
lower_green = np.array([25, 52, 72])
upper_green = np.array([102, 255, 255])
lower_red = np.array([0, 50, 50])
upper_red = np.array([10, 255, 255])
# 创建掩膜
mask_green = cv2.inRange(hsv, lower_green, upper_green)
mask_red = cv2.inRange(hsv, lower_red, upper_red)
# 合并掩膜
mask = cv2.bitwise_or(mask_green, mask_red)
# 应用掩膜
result = cv2.bitwise_and(img, img, mask=mask)
# 显示结果
cv2.imshow('result', result)
cv2.waitKey(0)
cv2.destroyAllWindows()
```
这段代码使用 OpenCV 库,将图像转换为 HSV 颜色空间,然后根据绿色和红色的范围创建掩膜,最后将掩膜应用到原图像上,得到识别苹果病虫害的结果。