Gaussian Bounding Boxes and Probabilistic Intersection-over-Union for Object Detection

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Llerena, Jeffri M., Zeni, Luis Felipe, Kristen, Lucas N., Jung, Claudio
Formato: Preprint
Publicado: 2021
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866913283826515968
author Llerena, Jeffri M.
Zeni, Luis Felipe
Kristen, Lucas N.
Jung, Claudio
author_facet Llerena, Jeffri M.
Zeni, Luis Felipe
Kristen, Lucas N.
Jung, Claudio
contents Most object detection methods use bounding boxes to encode and represent the object shape and location. In this work, we explore a fuzzy representation of object regions using Gaussian distributions, which provides an implicit binary representation as (potentially rotated) ellipses. We also present a similarity measure for the Gaussian distributions based on the Hellinger Distance, which can be viewed as a Probabilistic Intersection-over-Union (ProbIoU). Our experimental results show that the proposed Gaussian representations are closer to annotated segmentation masks in publicly available datasets, and that loss functions based on ProbIoU can be successfully used to regress the parameters of the Gaussian representation. Furthermore, we present a simple mapping scheme from traditional (or rotated) bounding boxes to Gaussian representations, allowing the proposed ProbIoU-based losses to be seamlessly integrated into any object detector.
format Preprint
id arxiv_https___arxiv_org_abs_2106_06072
institution arXiv
publishDate 2021
record_format arxiv
spellingShingle Gaussian Bounding Boxes and Probabilistic Intersection-over-Union for Object Detection
Llerena, Jeffri M.
Zeni, Luis Felipe
Kristen, Lucas N.
Jung, Claudio
Computer Vision and Pattern Recognition
Most object detection methods use bounding boxes to encode and represent the object shape and location. In this work, we explore a fuzzy representation of object regions using Gaussian distributions, which provides an implicit binary representation as (potentially rotated) ellipses. We also present a similarity measure for the Gaussian distributions based on the Hellinger Distance, which can be viewed as a Probabilistic Intersection-over-Union (ProbIoU). Our experimental results show that the proposed Gaussian representations are closer to annotated segmentation masks in publicly available datasets, and that loss functions based on ProbIoU can be successfully used to regress the parameters of the Gaussian representation. Furthermore, we present a simple mapping scheme from traditional (or rotated) bounding boxes to Gaussian representations, allowing the proposed ProbIoU-based losses to be seamlessly integrated into any object detector.
title Gaussian Bounding Boxes and Probabilistic Intersection-over-Union for Object Detection
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2106.06072