Table 3.
Layer name | Output size | Layer |
---|---|---|
conv1 | 112 × 112 | 7 × 7, 64, stride 2 |
conv2_x | 56 × 56 | 3 × 3 max pool, stride 2 |
[1 × 1, 64 3 × 3, 64 1 × 1, 256] | ||
conv3_x | 28 × 28 | [1 × 1, 128 3 × 3, 128 1 × 1, 512] |
conv4_x | [, 256 3 × 3, 256 1 × 1, 1024] |
|
conv5_x | 7 × 7 | [1 × 1, 512 3 × 3, 512 1 × 1, 2048] |
fc1 | 1 × 1 | Average pool |
in_features 2048, out_features 2048 | ||
fc2 | 1 × 1 | Dropout 0.5 |
in_features 2048, out_features 2048 | ||
fc3 | 1 × 1 | relu, dropout 0.5 |
in_features 2048, out_features=2 |