Training custom PointRend models
This section describes the steps for training a custom PointRend model for object instance segmentation tasks on the brain tumor dataset (described in the previous chapter). Training custom PointRend models includes steps similar to training instance segmentation models in Detectron2. Therefore, this section focuses more on the discrepancies between the two; the complete source code is available in this book’s GitHub repository.
The source code for downloading the brain tumor segmentation dataset, installing Detectron2, and registering train and test datasets remains the same. Similar to the previous section, before we can get a configuration file, we need to clone the Detectron2 repository to use the configuration files for the PointRend project:
!git clone https://github.com/facebookresearch/detectron2.git detectron2_repo
The code for the initial configuration remains the same:
output_dir = "output/pointrend" os.makedirs...