← all repositories

ethanhe42/KL-Loss

A PyTorch implementation of KL-Loss for improved bounding box regression in object detection using learned localization uncertainty.

721 stars Python Computer VisionML Frameworks
KL-Loss
Velocity · 7d
+0.3
★ / day
Trend
steady
star history

This repository implements a CVPR 2019 paper that proposes a novel bounding box regression loss enabling simultaneous learning of bounding box transformation and localization variance. The approach leverages learned variance during non-maximum suppression to merge neighboring boxes, improving localization accuracy. The implementation supports Faster R-CNN with VGG-16 and ResNet-50-FPN Mask R-CNN architectures, achieving significant AP improvements on MS-COCO.

heatdrop uses Google Analytics to see which pages get read — nothing else. Your call. How we handle data.