View source on GitHub |
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.