Trademark detection and classification based on an improved YOLO
摘要
Trademarks are important identifiers for goods and services. They play an increasingly important role in daily life and production. However, with the continuous development of commercial society and the increasing number of goods and services, trademark detection and classification technology has become particularly critical, and it has large application prospects in the fields of commodity image search and trademark infringement. Based on the You Only Look Once (YOLO)-v5 framework, this paper proposes the YOLO-BC network to better detect trademark images. First, the network adopts a new bi-directional feature pyramid network & coordinate attention(BiFC) module to replace the original enhanced feature extraction network (PANet) model to improve the feature extraction ability of the network at different scales and capture key information. Second, a convolutional block attention module (CBAM) is introduced into the C3 module, and the influence of the C3-CBAM module embedded in the backbone and neck of the algorithm network on network performance is explored through experiments. Subsequently, the distance intersection over union non-maximum suppression (DIOU-NMS) is used to replace the NMS in the original network as the bounding box non-maximum suppression method. Finally, we replace generalized IOU (GIOU) with