opencv视频流目标检测

最新推荐文章于 2025-05-12 14:47:30 发布

原创

最新推荐文章于 2025-05-12 14:47:30 发布 · 3.1k 阅读

24 ·

CC 4.0 BY-SA版权

1.采用opencv深度学习模块调用mobilenet_ssd模型进行目标检测，模型可以直接移植到移动端使用，毕竟模型不是很重，直接放线下测试代码：以供参考

import numpy as np
import cv2

deploy_prototxt = 'MobileNet-SSD/deploy.prototxt'
model = 'MobileNet-SSD/mobilenet_iter_73000.caffemodel'
confidence_default = 0.2

CLASSES = ["background", "aeroplane", "bicycle", "bird", "boat",
           "bottle", "bus", "car", "cat", "chair", "cow", "diningtable",
           "dog", "horse", "motorbike", "person", "pottedplant", "sheep",
           "sofa", "train", "tvmonitor"]
COLORS = np.random.uniform(0, 255, size=(len(CLASSES), 3))

print("[info] loading model...")
net = cv2.dnn.readNetFromCaffe(deploy_prototxt, model)

# 打开摄像头
camera = cv2.VideoCapture('video/2c2b6d119306711be3bd7e52f7f27ca7.mp4')
print(camera)
# cv2.namedWindow('Dynamic')
width = int(camera.get(cv2.CAP_PROP_FRAME_WIDTH))
height = int(camera.get(cv2.CAP_PROP_FRAME_HEIGHT))
print(height)
fps = camera.get(cv2.CAP_PROP_FPS)
fourcc = cv2.VideoWriter_fo