Table 2.
The size, max train accuracy and max validation accuracy of the four candidate models
| Model size (MB) | Max train accuracy | Max validation accuracy | |
|---|---|---|---|
| VGG16 | 512 | 67.34% | 63.04% |
| GoogLeNet | 21.5 | 64.90% | 63.37% |
| DenseNet161 | 102 | 68.04% | 66.22% |
| ResNet152 | 222 | 68.51% | 67.32% |
CNN, convolutional neural network.
Model size: Total parameter size in the CNN model; Max train accuracy: The biggest accuracy on the train set among all the epochs. Max validation accuracy: The biggest accuracy on the validation set among all the epochs.