Table 2.
Details about the image sizes at various stages of the proposed model
Layers | Output Size | Details |
---|---|---|
Input Layer | ||
Convolution Layer | conv, stride =2 | |
Max Pooling | Max pool, stride =2 | |
Average Pooling | Avg pool, stride =2 | |
Hybrid Pooling | [ Both max & avg pooling] | |
Dense Block 1 | [ conv, conv] | |
Attention Block 1 | GAP, Dense layers | |
Transition Layer 1 | conv | |
Avg pool, stride =2 | ||
Dense Block 2 | [ conv, conv] | |
Attention Block 2 | GAP, Dense layers | |
Transition Layer 2 | conv | |
Avg pool, stride =2 | ||
Dense Block 3 | [ conv, conv] | |
Attention Block 3 | GAP, Dense layers | |
Transition Layer 3 | conv | |
Avg pool, stride =2 | ||
Dense Block 4 | [ conv, conv] | |
Global Average pooling | global average pooling |