Classification and Regression Heads
The SSDHead class is responsible for initializing the Classification and Regression parts of the network. Here are a few notable details about their code:

Both the Classification and the Regression head inherit from the same class which is responsible for making the predictions for each feature map.
