基于改进YOLOv5的食品包装袋缺陷检测研究  

Research on Defect Detection of Food Packaging Bags Based on Improved Yolov5

作　　者：罗晨昊 LUO Chen-hao(Guangdong University of Technology,Guangzhou 510006,China)

机构地区：[1]广东工业大学,广东广州510006

出　　处：《绿色包装》2024年第11期15-18,共4页Green Packaging

摘　　要：针对传统的基于人工进行食品包装袋缺陷检测效率低下的问题,提出一种改进YOLOv5模型的食品包装袋缺陷检测方法。添加SE注意力机制,提高模型对于微小缺陷的检测能力;引入CARAFE上采样算子,利用特征图的内容信息来提高重建的质量;将激活函数替换为Mish激活函数,增强网络的准确性和泛化性。在自制的饼干包装袋数据集上,改进后的网络平均精度为88.4%,最终检测模型的mAP相较于原始模型提升了21.76%,参数量下降了9.36%。Aiming at the low efficiency of traditional artificial detection of food packaging bag defects,a food packaging bag defect detection method with improved YOLOv5 model was proposed.SE attention mechanism was added to improve the detection ability of the model for small defects.The CARAFE upsampling operator is introduced to improve the quality of reconstruction by using the content information of feature graph.Replace the activation function with a Mish activation function to enhance the accuracy and generalization of the network.On the self-made cookie bag data set,the average accuracy of the improved network is 88.4%,the mAP of the final detection model is increased by 21.76%,and the number of parameters is decreased by 9.36%compared with the original model.

关 键 词：深度学习 YOLOv5 缺陷检测 食品包装袋 

分 类 号：TB48[一般工业技术—包装工程]