精度要求高: 涉及安全问题,检测精度需非常高才能保证对施工场所人员的安全督导,因此需要专门针对此目标的检测算法进行优化。
存在小目标情况: 实际使用过程中,人员离镜头较远,因此需要模型对小目标的检测有较低的漏检率。
方案设计
本项目利用开源数据集基于PaddleX中单阶段目标检测模型实现安全帽检测任务。使用PaddleX提供的paddlex.det.coco_error_analysis接口对基线模型在验证集上预测错误的原因进行分析,采用了一系列优化步骤如IoU-aware、更换backbone等方法提升模型精度,最后在NV Jetson上部署应用。
模型优化策略和效果
本项目采用YOLOv3模型作为基线模型,backbone为DarkNet53。优化策略包括backbone更换、模型架构更换、输入图像大小改变、IoU-aware、label smooth等优化技巧。
模型部署
在模型优化结束后,根据我们实际应用需求,选择最合适的模型进行部署和应用。首先,将模型导出成静态图模型,然后采用PaddleX提供的C++ inference部署方案,在该方案中提供了在端侧设备Jetson部署Demo,用户可根据实际情况自行参考:
范例使用工具介绍
PaddleX是百度飞桨全流程开发工具,以低代码的形式支持开发者快速实现产业实际项目落地。PaddleX 集成飞桨智能视觉领域图像分类、目标检测、语义分割、实例分割任务能力,将深度学习开发全流程从数据准备、模型训练与优化到多端部署端到端打通,并提供统一任务API接口及图形化开发界面Demo。开发者无需分别安装不同套件,以低代码的形式即可快速完成飞桨全流程开发。
飞桨产业实践范例
助力企业跨越AI落地鸿沟
飞桨产业实践范例,致力于加速AI在产业落地的前进路径,减少理论技术与产业应用的差距。范例来源于产业真实业务场景,通过完整的代码实现,提供从数据准备到模型部署的方案过程解析,堪称产业落地的“自动导航”。
真实产业场景:与实际具有AI应用的企业合作共建,选取企业高频需求的AI应用场景如智慧城市-安全帽检测、智能制造-表计读数等;
完整代码实现:提供可一键运行的代码,在“AI Studio一站式开发平台”上使用免费算力一键Notebook运行;
详细过程解析:深度解析从数据准备和处理、模型选择、模型优化和部署的AI落地全流程,共享可复用的模型调参和优化经验;
直达项目落地:百度高工手把手教用户进行全流程代码实践,轻松直达项目POC阶段。
精彩课程预告
为了让小伙伴们更便捷地应用安全帽检测范例教程,百度高级研发工程师于5月12日19:00为大家深度解析从数据准备、方案设计到模型优化部署的开发全流程,手把手教大家进行代码实践。
相关阅读
引用说明
关注【飞桨PaddlePaddle】公众号
获取更多技术内容~