标题:Research on Data Augmentation for Image Classification Based on Convolution Neural Networks
作者:Jia Shijie; Wang Ping; Jia Peiyi; Hu Siping
通讯作者:Jia, SJ
作者机构:[Jia Shijie; Wang Ping; Hu Siping] Dalian Jiaotong Univ, Elect & Informat Coll, Dalian, Peoples R China.; [Jia Peiyi] Shandong Univ, Sch Mechatron & 更多
会议名称:Chinese Automation Congress (CAC)
会议日期:OCT 20-22, 2017
来源:2017 CHINESE AUTOMATION CONGRESS (CAC)
出版年:2017
页码:4165-4170
关键词:Image Classification; Data Augmentation; Convolution Neural network
摘要:The performance of deep convolution neural networks will be further enhanced with the expansion of the training data set. For the image classification tasks, it is necessary to expand the insufficient training image samples through various data augmentation methods. This paper explores the impact of various data augmentation methods on image classification tasks with deep convolution Neural network, in which Alexnet is employed as the pre-training network model and a subset of CIFAR10 and ImageNet (10 categories) are selected as the original data set. The data augmentation methods used in this paper include: GAN/WGAN, Flipping, Cropping, Shifting, PCA jittering, Color jittering, Noise, Rotation, and some combinations. Experimental results show that, under the same condition of multiple increasing, the performance evaluation on small-scale data sets is more obvious, the four individual methods (Cropping, Flipping, WGAN, Rotation) perform generally better than others, and some appropriate combination methods are slightly more effective than the individuals.
收录类别:CPCI-S
资源类型:会议论文
TOP