vis4d.zoo.base.models.mask_rcnn

Mask RCNN base model config.

Functions

get_mask_rcnn_cfg(num_classes, basemodel[, ...])

Return default config for mask_rcnn model and loss.
get_mask_rcnn_cfg(num_classes, basemodel, no_overlap=False, weights=None)[source]

Return default config for mask_rcnn model and loss.

This is an example for setting every component of the model and loss. Everything is the same as the default args.

Parameters:

  • num_classes (FieldReference | int) – Number of classes.

  • basemodel (ConfigDict) – Base model config.

  • no_overlap (bool, optional) – Whether to remove overlapping pixels between masks. Defaults to False.

  • weights (str | None, optional) – Weights to load. Defaults to None.

Return type:

tuple[ConfigDict, ConfigDict]