arXiv.itsnick.co

an independent field guide to 23 million papers from arXiv & OpenAlex — not affiliated with arXiv.org

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 map

Abstract

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.

More like this