← all repositories

fizyr/keras-retinanet

A Keras/TensorFlow implementation of RetinaNet for object detection, training on bounding box predictions with focal loss.

4.4k stars Python Computer VisionML Frameworks
keras-retinanet
Velocity · 7d
+1.4
★ / day
Trend
steady
star history

This repository provides a Keras implementation of the RetinaNet object detection architecture as described in the original paper by Lin et al. It implements the focal loss mechanism to address class imbalance in dense object detection and uses a Feature Pyramid Network (FPN) backbone for multi-scale feature extraction. The model accepts image inputs and outputs bounding box coordinates, confidence scores, and class labels for detected objects.

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