Added missing processor exports to pip package.

PiperOrigin-RevId: 447477914

Author: khanhlvg

tensorflow_lite_support/python/task/processor/proto/BUILD

@@ -42,6 +42,7 @@ py_library(
     name = "classifications_pb2",
     srcs = ["classifications_pb2.py"],
     deps = [
+        ":class_pb2",
         "//tensorflow_lite_support/cc/task/processor/proto:classifications_py_pb2",
     ],
 )
@@ -58,6 +59,7 @@ py_library(
     name = "detections_pb2",
     srcs = ["detections_pb2.py"],
     deps = [
+        ":class_pb2",
         "//tensorflow_lite_support/cc/task/processor/proto:detections_py_pb2",
     ],
 )

tensorflow_lite_support/tools/pip_package/task_processor.__init__.py

@@ -19,31 +19,40 @@
 """
 
 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 segmentation_options_pb2
-from tensorflow_lite_support.python.task.processor.proto import segmentations_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
 
 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
-SegmentationOptions = segmentation_options_pb2.SegmentationOptions
-Segmentation = segmentations_pb2.Segmentation
+EmbeddingResult = embedding_pb2.EmbeddingResult
 SearchOptions = search_options_pb2.SearchOptions
 SearchResult = search_result_pb2.SearchResult
+NearestNeighbor = search_result_pb2.NearestNeighbor
+SegmentationOptions = segmentation_options_pb2.SegmentationOptions
+Segmentation = segmentations_pb2.Segmentation
+SegmentationResult = segmentations_pb2.SegmentationResult
 
 # 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