Defect detection of pins based on RetinaNet and class balanced sampling methods
Author:
Affiliation:

Clc Number:

TM933

Fund Project:

National High Technology Research and Development Program(863 Program)(2015AA050201)

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

    The traditional detection method for the defects of the pin on the power connection fitting in the aerial survey of the drone is dependent on manual marking. Aiming at this problem, the deep learning algorithm RetinaNet is used to automatically extract the features of normal and defective samples and complete the fusion of low-level features and top-level features to achieve automatic labeling of defects. Considering the fact that the number of defective samples is much smaller than the normal number of samples, Firstly, the influence of the category imbalance caused by the deficiency of the defect sample on the recognition result is analyzed. The results show that the trained model in this case will make a large number of defective samples be mistakenly recognized as normal classes. Therefore, at the data level of RetinaNet, class balanced sampling is proposed to ensure that each category has the same opportunity to participate in training. The experimental results show that the proposed method can significantly improve the average precision of defect categories under the premise of maintaining the high recognition rate of normal categories.

    Reference
    Related
    Cited by
Get Citation
Share
Article Metrics
  • Abstract:
  • PDF:
  • HTML:
  • Cited by:
History
  • Received:January 22,2019
  • Revised:March 07,2019
  • Adopted:May 19,2019
  • Online: August 01,2019
  • Published: July 28,2019