chromium/third_party/tflite_support/src/tensorflow_lite_support/tools/pip_package/task_processor.__init__.py

# Copyright 2022 The TensorFlow Authors. All Rights Reserved.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#     http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
# ==============================================================================
"""TensorFlow Lite Task Library's processor module.

This module contains classes related to the pre-processing and post-processing
steps of the Task Library.
"""

from tensorflow_lite_support.python.task.processor.proto import bounding_box_pb2
from tensorflow_lite_support.python.task.processor.proto import class_pb2
from tensorflow_lite_support.python.task.processor.proto import classification_options_pb2
from tensorflow_lite_support.python.task.processor.proto import classifications_pb2
from tensorflow_lite_support.python.task.processor.proto import detection_options_pb2
from tensorflow_lite_support.python.task.processor.proto import detections_pb2
from tensorflow_lite_support.python.task.processor.proto import embedding_options_pb2
from tensorflow_lite_support.python.task.processor.proto import embedding_pb2
from tensorflow_lite_support.python.task.processor.proto import search_options_pb2
from tensorflow_lite_support.python.task.processor.proto import search_result_pb2
from tensorflow_lite_support.python.task.processor.proto import segmentation_options_pb2
from tensorflow_lite_support.python.task.processor.proto import segmentations_pb2
from tensorflow_lite_support.python.task.processor.proto import qa_answers_pb2
from tensorflow_lite_support.python.task.processor.proto import clu_pb2
from tensorflow_lite_support.python.task.processor.proto import clu_annotation_options_pb2

BoundingBox = bounding_box_pb2.BoundingBox
Category = class_pb2.Category
ClassificationOptions = classification_options_pb2.ClassificationOptions
Classifications = classifications_pb2.Classifications
ClassificationResult = classifications_pb2.ClassificationResult
DetectionOptions = detection_options_pb2.DetectionOptions
Detection = detections_pb2.Detection
DetectionResult = detections_pb2.DetectionResult
EmbeddingOptions = embedding_options_pb2.EmbeddingOptions
FeatureVector = embedding_pb2.FeatureVector
Embedding = embedding_pb2.Embedding
EmbeddingResult = embedding_pb2.EmbeddingResult
SearchOptions = search_options_pb2.SearchOptions
SearchResult = search_result_pb2.SearchResult
NearestNeighbor = search_result_pb2.NearestNeighbor
OutputType = segmentation_options_pb2.OutputType
SegmentationOptions = segmentation_options_pb2.SegmentationOptions
ColoredLabel = segmentations_pb2.ColoredLabel
ConfidenceMask = segmentations_pb2.ConfidenceMask
Segmentation = segmentations_pb2.Segmentation
SegmentationResult = segmentations_pb2.SegmentationResult
Pos = qa_answers_pb2.Pos
QaAnswer = qa_answers_pb2.QaAnswer
QuestionAnswererResult = qa_answers_pb2.QuestionAnswererResult
CluRequest = clu_pb2.CluRequest
CluResponse = clu_pb2.CluResponse
Mention = clu_pb2.Mention
CategoricalSlot = clu_pb2.CategoricalSlot
MentionedSlot = clu_pb2.MentionedSlot
BertCluAnnotationOptions = clu_annotation_options_pb2.BertCluAnnotationOptions

# Remove unnecessary modules to avoid duplication in API docs.
del bounding_box_pb2
del class_pb2
del classification_options_pb2
del classifications_pb2
del detection_options_pb2
del detections_pb2
del embedding_options_pb2
del embedding_pb2
del segmentation_options_pb2
del segmentations_pb2
del search_options_pb2
del search_result_pb2
del qa_answers_pb2
del clu_pb2
del clu_annotation_options_pb2