CODEWITHSUNDEEP
tensorflowkerasdeep-learningconv-neural-networkfaster-rcnn
Mcstang94
Mcstang94

Reputation: 31

Can someone give me an explanation for Multibox loss function?

I have found some expression for SSD Multibox-loss function as follows:

multibox_loss = confidence_loss + alpha * location_loss

Can someone explains what are the explanations for those terms?

Upvotes: 2

Views: 1459

Answers (1)

Agost Biro
Agost Biro

Reputation: 2839

SSD Multibox (short for Single Shot Multibox Detector) is a neural network that can detect and locate objects in an image in a single forward pass. The network is trained in a supervised manner on a dataset of images where a bounding box and a class label is given for each object of interest. The loss term

multibox_loss = confidence_loss + alpha * location_loss

is made up of two parts:

Confidence loss is a categorical cross-entropy loss for classifying the detected objects. The purpose of this term is to make sure that correct label is assigned to each detected object.

Location loss is a regression loss (either the smooth L1 or the L2 loss) on the parameters (width, height and corner offset) of the detected bounding box. The purpose of this term is to make sure that the correct region of the image is identified for the detected objects. The alpha term is a hyper parameter used to scale the location loss.

The precise formulation of the loss is given in Equation 1 of the SSD: Single Shot MultiBox Detector paper.

Upvotes: 2

Related Questions