fastai: faster_rcnn_model – R documentation

Become an expert in R — Interactive courses, Cheat Sheets, certificates and more!

faster_rcnn_model

Faster RSNN model

FasterRCNN model implemented by torchvision.

faster_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 'fastercnn_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. **faster_rcnn_kwargs: Keyword arguments that internally are going to be passed to 'torchvision.models.detection.faster_rcnn.FastRCNN'.

model

Interface to 'fastai'

v2.0.7

Apache License 2.0

Authors

Turgut Abdullayev [ctb, cre, cph, aut]

Initial release