基于opencv的安全帽佩戴检测

最新推荐文章于 2024-08-08 07:13:34 发布

weixin_30617737

最新推荐文章于 2024-08-08 07:13:34 发布

阅读量3.7k

点赞数 4

文章标签：人工智能 c/c++ 数据结构与算法

原文链接：http://www.cnblogs.com/shushukui/p/9295590.html

版权

本文介绍了如何使用OpenCV进行安全帽佩戴检测。作者从需求出发，讲述了利用行人检测分类器作为基础，通过调整参数达到满意效果，并自行收集和处理训练数据，训练安全帽检测模型的过程。尽管初期遇到样本问题和训练时间过长的挑战，但最终通过优化，成功实现了在视频流中的安全帽检测功能。

摘要由CSDN通过智能技术生成

最近老板在外面吹牛说我们可以做基于工地的监控，检测工人佩戴安全帽的情况。现在国内做图像识别的公司基本都是顶尖的技术公司了，可以参考的案例并不多，咨询了像海康这样的专门做视频的公司，销售人员说没有产品，可以做项目。那费用至少几十万上百万了。可海康官网明明说有检测安全帽功能的（心里暗骂一句，骗子）。找别人开发的成本太高，没办法只能硬着头皮上了。好在有一个图像处理神器opencv。只不过现在大多数的应用都是围绕在脸部，什么人脸检测啊，人脸识别啊。难得找到一个现成的行人检测分类器，经过测试发现效果还可以接受。附效果图一张。

绿框是检测到的行人。

# -*- coding: utf-8 -*-
# 行人检测
import cv2
from imutils.object_detection import non_max_suppression
from imutils import paths
from imutils.video import FileVideoStream
import numpy as np
import os,imutils 
cap = FileVideoStream(r'E:\test-3_000130-000849.mp4').start()
hog = cv2.HOGDescriptor()   
hog.setSVMDetector(cv2.HOGDescriptor_getDefaultPeopleDetector())
frame_index =