Training custom segmentation models
The previous section described the architecture of an object segmentation application that extends the object detection application. Therefore, the training steps and configurations of these two applications are similar. This section utilizes the source code provided in the Putting it all together section in swswswswswswswswsw. Ideally, we should perform all the steps in that chapter to get a better set of hyperparameters for this specific dataset. However, for simplicity, this section reuses all the source code as it is and focuses on the differences for training object segmentation applications. The following code snippet downloads the dataset prepared in the previous steps with train
and test
sets:
!wget -q <url_to/segbraintumors_coco.zip> !unzip -q segbraintumors_coco.zip
Once the dataset is extracted, the source code for Detectron2 installation, registering datasets, getting a training configuration, building a custom trainer with...