基于深度学习的厨师帽和口罩佩戴的检测方法
摘要:
本发明提供了一种基于深度学习的厨师帽和口罩佩戴的检测方法,包括:采集包括厨房场景内的人头图像;对所述人头图像进行预处理,构建人头检测数据集;将所述训练集放入卷积特征器提取包括所述厨师帽和所述口罩相关的特征,通过K-means聚类方法产生Anchor box个数来生成预测的人头边界框,并通过候选框与真实框的交并比为评价标准;将所述训练集数据输入YOLOv3网络进行重复训练,获得卷积层的权重值和偏置值,输出所述训练集数据的损失函数值;将所述训练好的模型进行模型压缩到满足实时检测条件时进行检测,根据所述训练好的模型检测所述人头图像,分别输出包括厨师帽和口罩的预测边界框和类别的结果。
0/0