Object detection a very important problem in computer vision. Here the model is tasked with localizing the objects present in an image, and at the same time, classifying them into different categories. Object detection models can be broadly classified into "single-stage" and "two-stage" detectors. Two-stage detectors are often more accurate but at the cost of being slower. Here in this example, we will implement RetinaNet, a popular single-stage detector, which is accurate and runs fast. RetinaNet uses a feature pyramid network to efficiently detect objects at multiple scales and introduces a new loss, the Focal loss function, to alleviate the problem of the extreme foreground-background class imbalance.

Don't forget to tag @srihari-humbarwadi in your comment, otherwise they may not be notified.

Authors community post
Srihari Humbarwadi | Computer Vision Research Engineer
Share this project
Similar projects
OpenMMLab Computer Vision
MMCV is a python library for CV research and supports many research projects such as object detection, segmentation, pose estimation, action ...
Deep Tutorials for PyTorch
This is a series of in-depth tutorials I'm writing for implementing cool deep learning models on your own with the amazing PyTorch library.
Training a pets detector model with TFOD API (TF 2)
In this notebook, we will be training a custom object detection model using the latest TensorFlow Object Detection (TFOD) API which is based on TensorFlow ...
Computer Vision Pretrained Models
A collection of computer vision pre-trained models.
Top collections