Hands-On Deep Learning Algorithms with Python

Architecture of the Capsule network

Let's suppose our network is trying to predict handwritten digits. We know that capsules in the earlier layers detect basic features, and those in the later layers detect the digit. So, let's call the capsules in the earlier layers primary capsules and those in the later layers digit capsules.

The architecture of a Capsule network is shown here:

In the preceding diagram, we can observe the following:

1. First, we take the input image and feed it to a standard convolution layer, and we call the result convolutional inputs.
2. Then, we feed the convolutional inputs to the primary capsules layer and get the primary ...