Table 1.
Comparison of different neural network models based on the number of layers.
Layer Name | Output Size | 18-Layer | 34-Layer | 50-Layer | 101-Layer | 152-Layer |
---|---|---|---|---|---|---|
conv1 | 112 × 112 | 7 × 7, 64, stride 2 | ||||
3 × 3 max pool, stride 2 | ||||||
conv2_x | 56 × 56 | × 2 | × 3 | × 3 | × 3 | × 3 |
conv3_x | 28 × 28 | × 2 | × 4 | × 4 | × 4 | × 8 |
conv4_x | 14 × 14 | × 2 | × 6 | × 6 | × 23 | × 36 |
conv5_x | 7 × 7 | × 2 | × 3 | × 3 | × 3 | × 3 |
1 × 1 | Average pool, 1000-d fc, softmax | |||||
FLOPs |