vis4d.zoo.base.models.faster_rcnn

Faseter R-CNN base model config.

Functions

get_default_rcnn_box_codec_cfg([...])	Get default config for rcnn box encoder and decoder.
get_default_rpn_box_codec_cfg([...])	Get default config for rpn box encoder and decoder.
get_faster_rcnn_cfg(num_classes, basemodel)	Return default config for faster_rcnn model and loss.

get_default_rcnn_box_codec_cfg(target_means=(0.0, 0.0, 0.0, 0.0), target_stds=(0.1, 0.1, 0.2, 0.2))

Get default config for rcnn box encoder and decoder.

Return type:

tuple[ConfigDict, ConfigDict]

get_default_rpn_box_codec_cfg(target_means=(0.0, 0.0, 0.0, 0.0), target_stds=(1.0, 1.0, 1.0, 1.0))

Get default config for rpn box encoder and decoder.

Return type:

tuple[ConfigDict, ConfigDict]

get_faster_rcnn_cfg(num_classes, basemodel, weights=None)

Return default config for faster_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.

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

Return type:

tuple[ConfigDict, ConfigDict]