Module: tfm.vision.configs.retinanet

RetinaNet configuration definition.

Classes

class Anchor

class AttributeHead

class DataConfig: Input config for training.

class DataDecoder: Data decoder config.

class DetectionGenerator

class ExportConfig

class Losses

class Parser

class RetinaNet

class RetinaNetHead

class RetinaNetTask

class TfExampleDecoder: A simple TF Example decoder config.

class TfExampleDecoderLabelMap: TF Example decoder with label map config.

Functions

retinanet(...): RetinaNet general config.

retinanet_resnetfpn_coco(...): COCO object detection with RetinaNet.

retinanet_spinenet_coco(...): COCO object detection with RetinaNet using SpineNet backbone.

retinanet_spinenet_mobile_coco(...): COCO object detection with mobile RetinaNet.

COCO_INPUT_PATH_BASE 'coco'
COCO_TRAIN_EXAMPLES 118287
COCO_VAL_EXAMPLES 5000