TreeSegNet: multi-scale query-based instance segmentation with frequency-aware and gated feature enhancement
摘要
Accurate instance segmentation of tree species is crucial for forest ecosystem monitoring and biodiversity assessment. However, UAV-based forest imagery presents unique challenges due to the substantial variations in tree scales, intricate branch-leaf structures, and densely overlapping canopies. Recent query-based instance segmentation methods have achieved remarkable progress, but their direct application to forest scenes remains suboptimal. These models often fail to fully exploit cross-scale feature interactions and rely on randomly initialized query embeddings that lack image-aware priors. To address these issues, we proposed TreeSegNet, a query-based instance segmentation method built upon the Mask2Former framework, incorporating the following key modules: (1) Multi-scale Feature Segmentation Fusion module, which integrates multi-scale features via global feature fusion at the feature level and contextual fusion at the window level. (2) Multi-scale Frequency Mean Query Generator, which combines low-frequency information from multi-scale feature maps with content query vectors. (3) Gated-enhanced Queries Transformer Decoder, which selectively integrates query information from different levels by dynamically adjusting gating values. Experimental results show that TreeSegNet achieves significant improvements on a self-collected dataset of tree species instances, with