Module: tfm.vision.heads
Stay organized with collections
Save and categorize content based on your preferences.
Heads package definition.
Classes
class DetectionHead: Creates a detection head.
class MaskHead: Creates a mask head.
class MaskScoring: Creates a mask scoring layer.
class RPNHead: Creates a Region Proposal Network (RPN) head.
class RetinaNetHead: Creates a RetinaNet head.
class SegmentationHead: Creates a segmentation head.