OR-FCOS: an enhanced fully convolutional one-stage approach for growth stage identification of Oudemansiella raphanipes
摘要
Accurate identification of Oudemansiella raphanipes growth stages is crucial for understanding its development and optimizing cultivation. However, deep learning methods for this task remain unexplored. This paper introduces OR-FCOS, an enhanced fully convolutional one-stage (FCOS) approach designed to improve accuracy and efficiency in identifying these growth stages. We constructed the ORaph8K dataset, containing 8,000 images of Oudemansiella raphanipes at different growth stages, used for training and validation. The OR-FCOS uses the MobileNetV3-Large backbone with an efficient multi-scale attention (EMA) module, improving feature extraction efficiency without sacrificing accuracy. A neural architecture search (NAS)-enhanced FCOS decoder replaces both the traditional feature pyramid networks (FPN) and prediction head in FCOS, optimizing feature fusion and prediction. Integrating the complete intersection over union (CIoU) loss function addresses standard IoU limitations by factoring in aspect ratio and bounding box center distance. Channel pruning further reduces the decoder’s parameters, decreasing model size and computational requirements while maintaining precision. The enhanced algorithm achieved a mean average precision (mAP) of 89.4% (