Name
Title | ||
|---|---|---|
Image based fruit category classification by 13-layer deep convolutional neural network and data augmentation. |
Abstract | ||
|---|---|---|
Fruit category identification is important in factories, supermarkets, and other fields. Current computer vision systems used handcrafted features, and did not get good results. In this study, our team designed a 13-layer convolutional neural network (CNN). Three types of data augmentation method was used: image rotation, Gamma correction, and noise injection. We also compared max pooling with average pooling. The stochastic gradient descent with momentum was used to train the CNN with minibatch size of 128. The overall accuracy of our method is 94.94%, at least 5 percentage points higher than state-of-the-art approaches. We validated this 13-layer is the optimal structure. The GPU can achieve a 177× acceleration on training data, and a 175× acceleration on test data. We observed using data augmentation can increase the overall accuracy. Our method is effective in image-based fruit classification. |
Year | DOI | Venue |
|---|---|---|
2019 | 10.1007/s11042-017-5243-3 | Multimedia Tools Appl. |
Keywords | Field | DocType |
Convolutional neural network, Fully connected layer, Softmax, Fruit category identification | Computer vision,Stochastic gradient descent,Pattern recognition,Softmax function,Convolutional neural network,Computer science,Pooling,Data type,Acceleration,Test data,Artificial intelligence,Gamma correction | Journal |
Volume | Issue | ISSN |
78 | 3 | 1573-7721 |
Citations | PageRank | References |
14 | 0.61 | 25 |
Authors | ||
7 | ||
Name | Order | Citations | PageRank |
|---|---|---|---|
| Yudong Zhang | 1 | 251 | 25.00 |
| zhengchao dong | 2 | 346 | 13.60 |
| Xianqing Chen | 3 | 37 | 1.30 |
| Wenjuan Jia | 4 | 54 | 3.39 |
| Sidan Du | 5 | 314 | 31.20 |
| Khan Muhammad | 6 | 986 | 67.67 |
| Shuihua Wang | 7 | 1564 | 87.49 |