4.6 Article

Twelve-layer deep convolutional neural network with stochastic pooling for tea category classification on GPU platform

期刊

MULTIMEDIA TOOLS AND APPLICATIONS
卷 77, 期 17, 页码 22821-22839

出版社

SPRINGER
DOI: 10.1007/s11042-018-5765-3

关键词

Convolutional neural network; Stochastic pooling; Data augmentation; Tea category classification; Stochastic gradient descent with momentum

资金

  1. Open fund of Key Laboratory of Guangxi High Schools Complex System and Computational Intelligence [2016CSCI01]
  2. Natural Science Foundation of China [61502254, 61602250]
  3. Program of Natural Science Research of Jiangsu Higher Education Institutions [15KJB470010, 16KJB520025]
  4. Natural Science Foundation of Jiangsu Province [BK20150983]

向作者/读者索取更多资源

Automatic tea-category identification is an important topic in factories and supermarkets. Traditional methods need to extract features from tea images manually, which may not be optimal for tea images classification. To avoid the time consuming efforts of handcrafted features extraction, this study proposed a new method combining convolutional neural network (CNN) with stochastic pooling. We collected 900 tea images of Oolong, green, and black teas, with 300 images for each category. The data augmentation method was used over the training set. We employed stochastic gradient descent with momentum (SGDM) to train the CNN. The experiments showed that a 12-layer CNN gives a good result. The sensitivities of Oolong, green, and black tea are 99.5%, 97.5%, and 98.0%, respectively. The overall accuracy of all three-tea categories is 98.33%. The stochastic pooling gives better results than maximum pooling and average pooling. The optimal number of convolutional layer for this task is 5. In addition, GPU has a 175x acceleration in training set and a 122x acceleration in test set, compared to CPU platform.

作者

我是这篇论文的作者
点击您的名字以认领此论文并将其添加到您的个人资料中。

评论

主要评分

4.6
评分不足

次要评分

新颖性
-
重要性
-
科学严谨性
-
评价这篇论文

推荐

暂无数据
暂无数据