Apple surface defect detection based on improved CNN and data augmentation
CSTR:
Author:
Affiliation:

(1. Hunan Vocational College of Commerce, Changsha, Hunan 410205, China; 2. Hunan University of Finance and Economics, Changsha, Hunan 410205, China; 3. Hubei Polytechnic University, Huangshi, Hubei 435003, China; 4. Wuhan University of Technology, Wuhan, Hubei 430070, China)

Clc Number:

Fund Project:

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

    Objective: Improve the detection accuracy and efficiency of apple surface defects. Methods: An detection method for apple surface defects was established based on an improved convolutional neural network (CNN) and data augmentation method. Firstly, the classical CNN was improved to detect apple surface defects. Then, using the conditional generation adversarial network, the image data of surface defect free and defective apples was augmented with synthetic apple images to improve the detection performance of the improved CNN for apple surface defects. Finally, by pruning the CNN model, the detection accuracy, detection speed and energy saving limits of apple surface defects were balanced reasonably to improve the practicability of the proposed method. Results: When 2 048 interpretive neurons were selected in the interpretation layer of the improved CNN, the average detection accuracy was the highest among the interpretive neuron number situations. Additionally, the diversity of the apple image data sets was enhanced with the synthetic apple images produced by the conditional generation adversarial network. In addition, the accuracy of the proposed method for detecting apple surface defects increased continuously with the increase of the proportion of the enhanced images in the test data set. When the ratio of the pruned model size to the original model size decreased from 100% to 50%, the detection efficiency of apple surface defects was doubled with 6.96% detection accuracy decreasing. Conclusion: This method is expected to realize the automatic defect detection in apple production and processing, and provide a reference for the developing of other fruit surface defect detection methods.

    Reference
    Related
    Cited by
Get Citation

皮 卫,屈喜龙,王绍成,等.基于改进CNN和数据扩充的苹果表面缺陷检测[J].食品与机械英文版,2023,39(8):122-128,226.

Copy
Share
Article Metrics
  • Abstract:
  • PDF:
  • HTML:
  • Cited by:
History
  • Received:January 19,2023
  • Revised:
  • Adopted:
  • Online: October 20,2023
  • Published:
Article QR Code