#导入numpy库和opencv库 import numpy as np import cv2
#读取图片,并对图片进行分割有效区域 img = cv2.imread("rrrrrr.PNG",1) roi = img[220:260, 300:380]
#展式图片 cv2.imshow("img", roi)
#统计rgb的值 bb = np.sum(roi,1) cc = np.sum(bb,0) print(cc) #打印rgb各个值
#进行比较判断,比较bgr的值。 if cc[1]<cc[2]: print("红灯") elif cc[1]>cc[2]: print("绿灯") else: print("黄灯")
#关闭
cv2.waitKey(0)
cv2.destroyAllWindows()
最后结果