Locating Image Objects with Probability Distributions

Yu Hsiang Lin, Chih Jen Hsu, Chih Hung Kuo, Ming Der Shieh

Research output: Contribution to journalArticlepeer-review

1 Citation (Scopus)

Abstract

In this letter, we predict the locations as probability distributions for the tasks of image object detection. We adopt the Kullback-Leibler divergences as the regression losses to train the deep neural networks. Since most existing evaluations label the objects with rectangular bounding boxes, we propose the Nearest Distribution Converter to find the closest uniform distributions from the predicted ones. Our proposed method can improve the detected accuracy measured in mAP by 0.57%, 0.75%, and 0.48% on the models YOLOv3, the YOLOv4-tiny, and the YOLOv4, respectively.

Original languageEnglish
Pages (from-to)2722-2726
Number of pages5
JournalIEEE Signal Processing Letters
Volume29
DOIs
Publication statusPublished - 2022

All Science Journal Classification (ASJC) codes

  • Signal Processing
  • Applied Mathematics
  • Electrical and Electronic Engineering

Fingerprint

Dive into the research topics of 'Locating Image Objects with Probability Distributions'. Together they form a unique fingerprint.

Cite this