vis4d.zoo.mask_rcnn.mask_rcnn_coco

Mask RCNN COCO training example.

Functions

get_config()

Returns the Mask-RCNN config dict for the coco detection task.

get_config()[source]

Returns the Mask-RCNN config dict for the coco detection task.

This is an example that shows how to set up a training experiment for the COCO detection task.

Note that the high level params are exposed in the config. This allows to easily change them from the command line. E.g.: >>> python -m vis4d.engine.run fit –config configs/faster_rcnn/faster_rcnn_coco.py –config.params.lr 0.001

Returns:

The configuration

Return type:

ExperimentConfig