Module: tfm.vision.heads

Heads package definition.

Classes

class DetectionHead: Creates a detection head.

class MaskHead: Creates a mask head.

class RPNHead: Creates a Region Proposal Network (RPN) head.

class RetinaNetHead: Creates a RetinaNet head.

class SegmentationHead: Creates a segmentation head.