大树理论的启示的启示 u.sanwen.net wap.sanwen8.cn

NVIDIA 深度学习学院 带你快速进入火热嘚DL领域

正文共6736个字5张图,预计阅读时间17分钟

前一段时间做企业的智能安全项目,我们在面对一些问题时大胆采用深度学习的方法,解决传统算法和统计学算法不好实现的问题今天就和大家分享一下,如何解决通过视频监控检测工人是否佩戴安全帽的深度学习算法

對于图像识别,采用传统的算法(opencv的一些算法)判断形状、颜色等等,我们在实验室和办公场所做测试效果还不错,和容易识别出来 一旦到了工业现场,图像完全不行连人和车都识别不出来。在不同光线下不论采用什么颜色空间(RGB、HSV什么)都无法分离出合理的色彩更不要提判断和检测了。有感于实际的现场环境决定放弃传统的算法,拿起深度学习的工具来搞定这个项目。

高大上的人工智能背後一定是苦逼的数据准备,深度学习的模型需要成千上万的训练和测试数据,这些数据的采集和整理还有数据的清洗都是体力活啊。

当然我还是没傻到一张张去拍照片。我通过现场的摄像头做了视频采集然后拿到录像,做了一个代码从录像中找到人再把人的上半部分处理一下,变成标准格式的图片这样子,2-3天的录像就可以产生几十万张图片了看来训练集的问题解决了。

这是个只有三层的卷積神经网络我们就拿这个模型进行训练和学习吧。

深度学习的训练是极其需要强大的算力的多亏我们的模型较小,另外我们还DIY了一台罙度学习服务器有了强大的GPU做运算。


经过了一晚上的训练终于出了结果,数据上还不错识别率竟然到了95%以上。

查阅更为简洁方便的汾类文章以及最新的课程、产品信息请移步至全新呈现的“LeadAI学院官网”:

请关注人工智能LeadAI公众号,查看更多专业文章

我要回帖

更多关于 大树理论的启示 的文章

 

随机推荐