残余物
计算机科学
计算机视觉
人工智能
计算机图形学(图像)
算法
作者
Ying Zhao,Xiaoming Wu,Xiangzhi Liu,Hao Chen,Bei Qi,Huomin Dong
标识
DOI:10.1109/smc54092.2024.10831817
摘要
In order to ensure food safety, it is required for personnel to wear hats and masks during food handling processes. To accurately detect the wearing status of kitchen staff, the ARP-YOLO model is proposed. Firstly, images are obtained from multiple kitchens and angles to construct a dataset reflecting the wearing status of hats and masks. To simulate more complex kitchen environments, Gaussian noise is added to the data and lighting conditions are adjusted for data augmentation. Lighting conditions in the kitchen can affect detection, causing the same target to exhibit different shapes and features under different lighting conditions, leading to missed detections. To address the above issues, we propose ARCA (Adaptive-Residual-Coordinate-Attention), which uses residual connections to strengthen attention to important features while preserving original features, and employs adaptive convolution reduction to reduce module parameters. To improve target localization accuracy, P2 detection layers are added in the Neck to obtain more accurate target position information. The ARP-YOLO model demonstrates significant improvements over the baseline model, with a 13.6% increase in Recall, allowing for more effective target capture and reduced missed detection risk. Additionally, mAP@0.5 has increased by 10%, enhancing target localization accuracy. The F1-Score has also increased by 7.4%, better balancing the relationship between Precision and Recall. To validate the effectiveness of the model, comparative experiments with other models are conducted, showing that ARP-YOLO model's Recall and Average Precision(AP) are higher than those of other models.
科研通智能强力驱动
Strongly Powered by AbleSci AI