mask_rcnn_model: MaskRCNN model
In fastai: Interface to 'fastai'

mask_rcnn_model

R Documentation

MaskRCNN model

MaskRCNN model implemented by torchvision.

mask_rcnn_model(
  num_classes,
  backbone = NULL,
  remove_internal_transforms = TRUE,
  pretrained = TRUE
)

`num_classes`	Number of classes.
`backbone`	Backbone model to use. Defaults to a resnet50_fpn model.
`remove_internal_transforms`	The torchvision model internally applies transforms like resizing and normalization, but we already do this at the ‘Dataset' level, so it’s safe to remove those internal transforms.
`pretrained`	Argument passed to 'maskrcnn_resnet50_fpn' if 'backbone is NULL'. By default it is set to TRUE: this is generally used when training a new model (transfer learning). 'pretrained = FALSE' is used during inference (prediction) for cases where the users have their own pretrained weights. **mask_rcnn_kwargs: Keyword arguments that internally are going to be passed to 'torchvision.models.detection.mask_rcnn.MaskRCNN'.