Potato defects classification and localization with convolutional neural networks - Archive ouverte HAL Access content directly
Conference Papers Year :

Potato defects classification and localization with convolutional neural networks

(1) , (1) , (1)
1

Abstract

Various defects can appear on the surface of a potato, producing an adverse effect on their market price. For several years, manual methods have been applied to classify this tuber, which caused certain drawbacks such as: high-cost, high-processing time and subjective results. In this paper we introduce a deep-learning based method to classify and localize defects in potatoes with the aim to automate the quality control task. An extensive dataset was created including six potato categories: healthy, damaged, greening, black dot, common scab and black scurf. Then, a convolutional neural network (CNN) was trained with this dataset in order to achieve the classification task. We also propose to leverage the localization capability of the trained network to localize the region of the classified defect. Finally, a global evaluation was done in a test set, where 4 different sides images were taken into account to represent one tuber. Experimental results with different CNN architectures are shown. We achieved an average F1-score of 0.94 for the classification task. The localization performance is measured qualitatively by a heat map output, which shows that the proposed method accurately localize the defects.
Not file

Dates and versions

hal-02358746 , version 1 (12-11-2019)

Identifiers

Cite

Sofia Marino, André Smolarz, Pierre Beauseroy. Potato defects classification and localization with convolutional neural networks. Fifteenth International Conference on Quality Control by Artificial Vision, May 2019, Mulhouse, France. pp.28, ⟨10.1117/12.2521264⟩. ⟨hal-02358746⟩
56 View
0 Download

Altmetric

Share

Gmail Facebook Twitter LinkedIn More