Research Article

Research on Classification of Fine-Grained Rock Images Based on Deep Learning

Table 2

Network structure of EfficientNet-B0.

StageOperatorResolutionNumber of channelsNumber of layers
iFiHi × WiCiLi

1Conv3 x 3224 × 224321
2MBConv1, k3 x k3112 × 112161
3MBConv6, k3 x k3112 × 112242
4MBConv6, k5 x k556 × 56402
5MBConv6, k3 x k328 × 28803
6MBConv6, k5 x k514 × 141123
7MBConv6, k5 x k514 × 141924
8MBConv6, k3 x k37 × 73201
9Conv1 x 1&Pooling&FC7 × 712801