利用Python写一个车辆检测代码
时间: 2024-03-06 07:02:28 浏览: 150
优化车辆检测(代码)(python)
要写一个车辆检测代码,可以使用Python中的OpenCV库和机器学习算法。以下是一个简单的车辆检测代码示例:
```python
import cv2
# 读取图片
img = cv2.imread('car.jpg')
# 加载车辆检测分类器
car_cascade = cv2.CascadeClassifier('cars.xml')
# 转换为灰度图片
gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
# 检测车辆
cars = car_cascade.detectMultiScale(gray, 1.1, 1)
# 画框
for (x, y, w, h) in cars:
cv2.rectangle(img, (x, y), (x+w, y+h), (0, 0, 255), 2)
# 显示图片
cv2.imshow('car detection', img)
cv2.waitKey(0)
cv2.destroyAllWindows()
```
在这个示例中,我们首先使用OpenCV的`imread`函数读取一张图片,然后加载了一个预训练的车辆检测分类器,并使用`detectMultiScale`函数进行车辆检测。最后,我们使用`rectangle`函数在车辆周围画了一个矩形框,然后显示结果图片。注意,这个示例只是一个简单的演示,实际应用中需要针对具体场景进行参数调整和优化。
阅读全文