View source on GitHub
|
TensorFlow Models Vision Libraries.
Modules
anchor module: Anchor box and labeler definition.
anchor_generator module: Multi scale anchor generator definition.
augment module: Augmentation policies for enhanced image/video preprocessing.
backbones module: Backbones package definition.
box_matcher module: Box matcher implementation.
box_ops module: Box related ops.
classification_model module: Build classification models.
configs module: Configs package definition.
decoders module: Decoders package definition.
factory module: Factory methods to build models.
factory_3d module: Factory methods to build models.
heads module: Heads package definition.
image_classification module: Image classification task definition.
iou_similarity module: Region Similarity Calculators.
layers module: Layers package definition.
mask_ops module: Utility functions for segmentations.
maskrcnn module: MaskRCNN task definition.
maskrcnn_model module: R-CNN(-RS) models.
models module: Models under Vision package.
nms module: Tensorflow implementation of non max suppression.
preprocess_ops module: Preprocessing ops.
preprocess_ops_3d module: Utils for processing video dataset features.
retinanet module: RetinaNet task definition.
retinanet_model module: RetinaNet.
sampling_ops module: Class to subsample minibatches by balancing positives and negatives.
segmentation_model module: Build segmentation models.
semantic_segmentation module: Image segmentation task definition.
serving module: Tools for exporting models.
spatial_transform_ops module: Spatial transform ops.
target_gather module: Definition of target gather, which gathers targets from indices.
video_classification module: Video classification task definition.
video_classification_model module: Build video classification models.
Classes
class ImageClassificationTask: A task for image classification.
class MaskRCNNTask: A single-replica view of training procedure.
class RetinaNetTask: A single-replica view of training procedure.
class SemanticSegmentationTask: A task for semantic segmentation.
class VideoClassificationTask: A task for video classification.