国内刊号:51-1196/TP
国际刊号:1001-3695
发布日期:
作者:庞殊杨,芦莎,
单位:1.中冶赛迪重庆信息技术有限公司大数据产品部,重庆401122;2.重庆大学重庆大学—辛辛那提大学联合学院,重庆400044;
关键词:安全帽识别,深度学习,目标检测,多任务级联卷积网络,MobileNet,
基金:重庆市人工智能技术创新重大主题专项重点研发项目(cstc2017rgzn-zdyfX0042);;
针对现有安全帽检测方法对多重叠目标和小目标漏检率较高的问题,提出了一种基于改进MTCNN(multi-task cascaded convolutional neural network)的多尺度安全帽识别方法。首先,删除MTCNN中针对人脸识别的landmark部分以简化网络结构;其次,用普通卷积层替换最大池化层以构成全卷积网络,提升网络检测精度;然后引入MobileNet轻量化网络结构减少计算量;最后,适当调整网络卷积核个数和全连接层神经元个数使模型更适用于不同尺寸的安全帽识别。实验结果表明,与原MTCNN相比,该算法的精确度和召回率分别提高了3.22%和6.73%,就小尺寸安全帽识别而言,F1值提高了8.13%;在无GPU环境下的平均检测速度为29.62 fps,兼顾了多尺度安全帽识别的准确率与实时性。
来源:2021年第6期
《计算机应用研究》期刊编辑部