CBAM attention mechanism tomato detection method based on YOLOv7

• Main Authors: LI YINUO, WANG JIAN, LUO GAIFANG, HOU YIJUN, CAI JI, SHI YAN, WANG KUN, SUN SHENG, GUO WENHUI, SONG GUOZHU, JING CHAO
• Format: Patent
• Language: Chinese; English
• Published: 12.05.2023
• Subjects: CALCULATING; COMPUTER SYSTEMS BASED ON SPECIFIC COMPUTATIONAL MODELS; COMPUTING; COUNTING; IMAGE DATA PROCESSING OR GENERATION, IN GENERAL; PHYSICS

The invention discloses a CBAM attention mechanism tomato detection method based on YOLOv7. The CBAM attention mechanism tomato detection method has the advantages of reducing noise, improving recognition accuracy and the like. Firstly, labeling, data enhancement and data set division are carried out on an acquired greenhouse string tomato image, secondly, a Gaussian filter is introduced before the image is input into a target detection backbone network for feature extraction, noise reduction processing is carried out on the image, then a CBAM module is introduced into a YOLOv7 model after feature extraction of the backbone network is finished, and a CBAM model is introduced into the YOLOv7 model. And feature enhancement is sequentially carried out from a channel domain and a spatial domain, so that the identification accuracy of the string-type tomatoes is further improved. According to data display, the recognition accuracy can reach 95.8%, the field recognition rate can reach 91%, the method meets practica