MTSC-Net: A Semi-Supervised Counting Network for Estimating the Number of Slash pine New Shoots

Abstract

The new shoot density of slash pine serves as a vital indicator for assessing its growth and photosynthetic capacity, while the number of new shoots offers an intuitive reflection of this density. With deep learning methods becoming increasingly popular, automated counting of new shoots has greatly improved in recent years but is still limited by tedious and expensive data collection and labeling. To resolve these issues, this paper proposes a semi-supervised counting network (MTSC-Net) for estimating the number of slash pine new shoots. First, based on the mean-teacher framework, we introduce the improved VGG19 to extract multiscale new shoot features. Second, to connect local new shoot feature information with global channel features, attention feature fusion module is introduced to achieve effective feature fusion. Finally, the new shoot density map and density probability distribution are processed in a fine-grained manner through multiscale dilated convolution of the regression head and classification head. In addition, a masked image modeling strategy is introduced to encourage the contextual understanding of global new shoot features and improve the counting performance. The experimental results show that MTSC-Net outperforms other semi-supervised counting models with labeled percentages ranging from 5% to 50%. When the labeled percentage is 5%, the mean absolute error and root mean square error are 17.71 and 25.49, respectively. These findings demonstrate that our work can be used as an efficient semi-supervised counting method to provide automated support for tree breeding and genetic utilization.