Face detection with SSD

Single Shot Detection (SSD) is another fast and accurate deep learning object-detection method with a similar concept to YOLO, in which the object and bounding box are predicted in the same architecture.

SSD model architecture

The SSD algorithm is called single shot because it predicts the bounding box and the class simultaneously as it processes the image in the same deep learning model. Basically, the architecture is summarized in the following steps:

1. A 300 x 300 image is input into the architecture.
2. The input image is passed through multiple convolutional layers, obtaining different features at different scales.
3. For each feature map obtained in 2, we use a 3 x 3 convolutional filter to evaluate small set of default bounding boxes.
4. For each default box evaluated, the bounding box offsets and class probabilities are predicted.

The model architecture looks like this:

SSD is used for predicting multiple classes similar to that in YOLO, but it can be modified to detect a single...