Skip to content

A pytorch reimplementation of Bounding Box Regression using KL-Loss (CVPR '2019)

License

Notifications You must be signed in to change notification settings

souradipp76/box_regression-kl_loss

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

10 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Accurate Prediction of Bounding Boxes in Object Detection

Introduction

This repository contains experiments to reproduce the results of the paper Bounding Box Regression with Uncertainty for Accurate Object Detection (He et al.). For more details, check out this report / slides / presentation.

Running Experiments

  • Run the notebook file kl_loss_MMDet.ipynb with CPU configuration and follow the instructions in the file.
Open in Colab
  • Run the notebook file kl_loss_pytorch.ipynb and follow the instructions in the file. Note that Google Drive has been linked to the notebook for data and output storage.
Open in Colab

Visualization

See a visualization of the objects getting detected using this app.

Contributing

Pull requests are welcome. For major changes, please open an issue first to discuss what you would like to change.

License

MIT