Logits
Use global average pooling to obtain the model's logits.
We'll cover the following...
Chapter Goals:
- Learn about global average pooling and its benefits
- Obtain the per class model logits
A. Channel-based logits
In the CNN section, we applied flattening and a couple of fully-connected layers (with dropout) to obtain our model's logits. However, rather than flattening our data, we can instead use a convolution layer to convert the number of channels in our data to the number of image classes (categories). Then, we apply a special type of average pooling layer to obtain logits for each channel (i.e. image class).
B. Global average pooling
Average pooling follows the exact same concept as max pooling, with the only difference being that the filter performs an average rather than a ...