Module: tfm.vision.models

Stay organized with collections Save and categorize content based on your preferences.

Models under Vision package.

Classes

class ClassificationModel: A classification class builder.

class MaskRCNNModel: The Mask R-CNN(-RS) and Cascade RCNN-RS models.

class RetinaNetModel: The RetinaNet model class.

class SegmentationModel: A Segmentation class model.

class VideoClassificationModel: A video classification class builder.