Exciting news! Gradient has launched a FREE GPU plan. Read More
Project Details

Object Detection With Detectron2

Train Detectron2 on custom object detection data

By
The Gradient Team

Description

This tutorial implements the new Detectron2 Library by facebook, and shows how to train on your own custom objects for object detection.

We use a public blood cell detection dataset, which is open source and free to use. You can also use this notebook on your own data.

To train our detector we take the following steps:

  • Install Detectron2 dependencies
  • Download custom Detectron2 object detection data
  • Visualize Detectron2 training data
  • Write our Detectron2 training configuration
  • Run Detectron2 training
  • Evaluate Detectron2 performance
  • Run inference on test images