用Python写一个水果图像识别

好的，下面是一个简单的 Python 水果图像识别程序：

首先需要安装必要的库：OpenCV 和 Numpy。

pip install opencv-python
pip install numpy

然后，可以使用以下代码来进行水果图像识别：

import cv2
import numpy as np

# 加载图像
img = cv2.imread("fruit.jpg")

# 定义颜色范围
lower_red = np.array([0, 50, 50])
upper_red = np.array([10, 255, 255])
lower_green = np.array([50, 50, 50])
upper_green = np.array([70, 255, 255])
lower_yellow = np.array([25, 50, 50])
upper_yellow = np.array([35, 255, 255])

# 转换颜色空间
hsv_img = cv2.cvtColor(img, cv2.COLOR_BGR2HSV)

# 创建掩码
mask_red = cv2.inRange(hsv_img, lower_red, upper_red)
mask_green = cv2.inRange(hsv_img, lower_green, upper_green)
mask_yellow = cv2.inRange(hsv_img, lower_yellow, upper_yellow)

# 执行位运算
red_fruits = cv2.bitwise_and(img, img, mask=mask_red)
green_fruits = cv2.bitwise_and(img, img, mask=mask_green)
yellow_fruits = cv2.bitwise_and(img, img, mask=mask_yellow)

# 显示结果
cv2.imshow("Original image", img)
cv2.imshow("Red fruits", red_fruits)
cv2.imshow("Green fruits", green_fruits)
cv2.imshow("Yellow fruits", yellow_fruits)
cv2.waitKey(0)
cv2.destroyAllWindows()

这段代码会加载一张名为 "fruit.jpg" 的图像，并将其转换为 HSV 颜色空间。然后，根据颜色范围定义掩码，并使用位运算提取图像中的红色、绿色和黄色水果。最后，将结果显示出来。

注意：这只是一个简单的水果图像识别程序，它只能识别红色、绿色和黄色水果。如果要进行更复杂的图像识别任务，需要使用更高级的算法和工具。