vis4d.zoo.retinanet.retinanet_coco

RetinaNet COCO training example.

Functions

get_config()

Returns the RetinaNet config dict for the coco detection task.

get_config()[source]

Returns the RetinaNet 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 vis4d/zoo/retinanet/retinanet_rcnn_coco.py –config.num_epochs 100 –config.params.lr 0.001

Returns:

The configuration

Return type:

ExperimentConfig