Computer Science · Computer Vision and Pattern Recognition
DLDENet: Deep Local Directional Embeddings with Increased Foreground Focal Loss for object detection
Fabian Souto Herrera, José M. Saavedra
See on the mapAbstract
We propose modifications to the RetinaNet architecture and Focal Loss [1] to improve the effectiveness of them in the context of object detection. We show that through normalizing the embeddings generated by each receptive field in the feature map, classifying them using cosine similarity, increasing the loss for the foreground anchors and forcing the dispersion of the classification vectors will improve the performance of the model. We test our proposal with the FlickrLogos-32 [2] dataset that contains 40 examples for each one of the 32 classes, achieving competitive results with respect to the state-of-the-art approaches with a precision of 0.975, a recall of 0.944, an accuracy of 0.949 and improving the mAP over the dataset from 0.657 with RetinaNet [1] to 0.775 using our new architecture DLDENet.