Hands-On Computer Vision with Detectron2

5

Training Custom Object Detection Models

This chapter starts with an introduction to the dataset and dataset preprocessing steps. It then continues with steps to train an object detection model using the default trainer and an option for saving and resuming training. This chapter then describes the steps to select the best from a set of trained models. It also provides the steps to perform object detection tasks on images with discussions on establishing appropriate inferencing thresholds and classification confidences. Additionally, it details the development process of a custom trainer by extending the default trainer and incorporating a hook into the training process.

By the end of this chapter, you will be able to train Detectron2’s models ...