YOLOv8-CAS: An improvement of multi-class target detection algorithm based on YOLO

Yifeng Du,Junmei Guo,Fanbing Li,Shengqi Yang
DOI: https://doi.org/10.1109/ICETCI61221.2024.10594299
2024-05-24
Abstract:In this paper, a new target detection algorithm, YOLOv8-CAS, seeks to further enhance the effectiveness of the multi-class target detection algorithm. By incorporating the CBAM into the Neck layer, we aim to overcome the drawbacks of traditional convolutional neural networks in processing information of diverse scales, shapes, and orientations. This enhancement not only strengthens the feature representation but also boosts the network's ability to concentrate on distinct target classes. The ASFF method is also employed to enhance the accuracy of the YOLOv8 decoupling header and its capability in handling complex scenes. By effectively filtering out conflicting information, it further improves scale invariance. The experimental results show that the improved YOLOv8-CAS achieves a significant improvement in mAP, which indicates that the new feature fusion approach and attention mechanism can effectively enhance the algorithm's detection performance for multi-class targets. This paper provides effective ideas and methods for enhancing the performance of multi-class target detection algorithms, which have practical application potential.
Computer Science
What problem does this paper attempt to address?