A method for detecting apple at night based on YOLOv8n with fusion of image enhancement and transfer learning
CSTR:
Author:
Affiliation:

1.Institute of Agricultural Machinery, Hubei University of Technology, Wuhan 430068, China;2.Hubei Engineering Research Center for Intellectualization of Agricultural Equipment, Wuhan 430068, China

Clc Number:

TP391.4

Fund Project:

  • Article
  • |
  • Figures
  • |
  • Metrics
  • |
  • Reference
  • |
  • Related
  • |
  • Cited by
  • |
  • Materials
  • |
  • Comments
    Abstract:

    This paper proposed a method for detecting apple at night based on YOLOv8n with fusion of image enhancement and transfer learning to address the issues of low recognition rate and poor real-time performance of apples in nighttime scenarios. Firstly, embedding a Zero-DCE module in the front-end of YOLOv8n enhanced images of apple at night, presented the contours and details of apples more clearly, and reduced the difficulty of recognizing images of apple at night. Secondly, using SPD-Conv for down-sampling enhanced the ability of the model to extract fine-grained features. On this basis, transfer learning training strategy was used to solve the problem of small sample size in the dataset of apple at night. The MS COCO dataset containing categories of apple was selected as the source domain dataset. In term of the target domain dataset in nighttime scenarios, Zero-DCE was used to increase its similarity with images of apple during the day and finely tune the model of target domain on the model of source domain. Experiments were conducted on the image dataset of apple at night based on the method above. The results showed that the model accuracy P, a recall R, and an average accuracy mean mAP@0.5:0.95 of method proposed was 97.0%,93.4% and 74.6%, being 2.3,1.9, and 4.3 percentages higher than that of the YOLOv8n original model. The inference speed of this model was 22 frames/s, meeting requirements of real-time detection. The results of the ablation experiment showed that the combined effect of image enhancement and transfer learning exceeded the sum of the effects when applied separately. The improved model performed better than the original model in dealing with complex situations including overlap, occlusion, green fruits, and dim lighting, and had good robustness.

    Reference
    Related
    Cited by
Get Citation

仝召茂,陈学海,马志艳,杨光友,张灿. A method for detecting apple at night based on YOLOv8n with fusion of image enhancement and transfer learning[J]. Jorunal of Huazhong Agricultural University,2024,43(5):1-9.

Copy
Related Videos

Share
Article Metrics
  • Abstract:
  • PDF:
  • HTML:
  • Cited by:
History
  • Received:December 01,2023
  • Revised:
  • Adopted:
  • Online: October 08,2024
  • Published:
Article QR Code