DenseYOLO: Yet Faster, Lighter and More Accurate YOLO - Archive ouverte HAL
Communication Dans Un Congrès Année : 2020

DenseYOLO: Yet Faster, Lighter and More Accurate YOLO

Solomon Negussie Tesema
  • Fonction : Auteur
El-Bay Bourennane

Résumé

As much as an object detector should be accurate, it should be light and fast as well. However, current object detectors tend to be either inaccurate when lightweight or very slow and heavy when accurate. Accordingly, determining tolerable tradeoff between speed and accuracy of an object detector is not a simple task. One of the object detectors that have commendable balance of speed and accuracy is YOLOv2. YOLOv2 performs detection by dividing an input image into grids and training each grid cell to predict certain number of objects. In this paper we propose a new approach to even make YOLOv2 more fast and accurate. We re-purpose YOLOv2 into a dense object detector by using fine-grained grids, where a cell predicts only one object and its corresponding class and objectness confidence score. Our approach also trains the system to learn to pick a best fitting anchor box instead of the fixed anchor assignment during ground-truth annotation as used by YOLOv2. We will also introduce a new loss function to balance the overwhelming imbalance between the number of grids responsible of detecting an object and those that should not.
Fichier non déposé

Dates et versions

hal-04407873 , version 1 (21-01-2024)

Identifiants

Citer

Solomon Negussie Tesema, El-Bay Bourennane. DenseYOLO: Yet Faster, Lighter and More Accurate YOLO. 2020 11th IEEE Annual Information Technology, Electronics and Mobile Communication Conference (IEMCON), Nov 2020, Vancouver, France. pp.0534-0539, ⟨10.1109/IEMCON51383.2020.9284923⟩. ⟨hal-04407873⟩
11 Consultations
0 Téléchargements

Altmetric

Partager

More