期刊
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
类别
资金
- Open fund of Key Laboratory of Guangxi High Schools Complex System and Computational Intelligence [2016CSCI01]
- Natural Science Foundation of China [61502254, 61602250]
- Program of Natural Science Research of Jiangsu Higher Education Institutions [15KJB470010, 16KJB520025]
- 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.
作者
我是这篇论文的作者
点击您的名字以认领此论文并将其添加到您的个人资料中。
推荐
暂无数据