CenterNet-LW-SE net: integrating lightweight CenterNet and channel attention mechanism for the detection of Camellia oleifera fruits

Camellia oleifera is a typical dense and small economic oil fruit. Its traditional harvesting method relies on manual labor, which is inefficient and costly. It is of significance to develop automatic harvesting machinery, which relies on rapid and accurate detection of Camellia oleifera in natural and complex environment. In this study, a novel model named CenterNet-LW-SE was proposed. Based on the structure of CenterNet, the lightweight Deep Layer Aggregation was used as the feature extraction backbone and the Squeeze-and-Excitation module was embedded to enhance the ability of feature extracting. A total of 4700 images were used to train and test the algorithms. The precision, recall, AP_75 , F_1 , model size and detection speed of CenterNet-LW-SE were 94.70