Based on PyTorch framework, this paper studies the network structure of classical convolutional neural network VGG16 and the process of input image changing in the network. Using VGG16 to train the small dataset CIFAR10 in various ways, it is found that using the Kaiming normal distribution to initialize the weight for training is far less effective than transfer learning (using the training weight of VGG16 on the ImageNet dataset as the initial value of the weight). The input image of VGG16 can be of any size, but after the image of CIFAR10 is resized to 224×224, the accuracy rate on the test set is improved by 6 percentage points.
Access to the requested content is limited to institutions that have purchased or subscribe to SPIE eBooks.
You are receiving this notice because your organization may not have SPIE eBooks access.*
*Shibboleth/Open Athens users─please
sign in
to access your institution's subscriptions.
To obtain this item, you may purchase the complete book in print or electronic format on
SPIE.org.
INSTITUTIONAL Select your institution to access the SPIE Digital Library.
PERSONAL Sign in with your SPIE account to access your personal subscriptions or to use specific features such as save to my library, sign up for alerts, save searches, etc.