gimlet-api 0.0.1__py3-none-any.whl → 0.0.4__py3-none-any.whl

gml/pipelines.py

@@ -0,0 +1,149 @@
+# Copyright 2023- Gimlet Labs, Inc.
+#
+# 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.
+#
+# SPDX-License-Identifier: Apache-2.0
+
+import abc
+from typing import List
+
+
+class Pipeline:
+    @abc.abstractmethod
+    def to_yaml(self, models: List[str], org_name: str) -> str:
+        pass
+
+
+class SingleModelPipeline(Pipeline):
+    def to_yaml(self, models: List[str], org_name: str) -> str:
+        if len(models) != 1:
+            raise ValueError(
+                "{} only supports a single model".format(type(self).__qualname__)
+            )
+        return self._to_yaml(models[0], org_name)
+
+    @abc.abstractmethod
+    def _to_yaml(self, model_name: str, org_name: str) -> str:
+        pass
+
+
+class SimpleDetectionPipeline(SingleModelPipeline):
+    def __init__(self, add_tracking_id: bool = False):
+        self.add_tracking_id = add_tracking_id
+
+    def _to_yaml(self, model_name: str, org_name: str):
+        add_tracking_id = "true" if self.add_tracking_id else "false"
+        # editorconfig-checker-disable
+        return f"""---
+nodes:
+- name: camera_source
+  kind: CameraSource
+  outputs:
+  - frame
+- name: detect
+  kind: Detect
+  attributes:
+    add_tracking_id: {add_tracking_id}
+    model:
+      model:
+        name: {model_name}
+        org: {org_name}
+    frame_rate_limit: 30
+  inputs:
+    frame: .camera_source.frame
+  outputs:
+  - detections
+- name: frame_metrics_sink
+  kind: FrameMetricsSink
+  attributes:
+    frame_rate_limit: 30
+  inputs:
+    frame: .camera_source.frame
+  outputs:
+  - frame_metrics
+- name: detection_metrics_sink
+  kind: DetectionsMetricsSink
+  attributes:
+  inputs:
+    detections: .detect.detections
+- name: pipeline_latency_metrics_sink
+  kind: LatencyMetricsSink
+  attributes:
+    name: model
+  inputs:
+    reference: .camera_source.frame
+    detections: .detect.detections
+    frame_metrics: .frame_metrics_sink.frame_metrics
+- name: video_stream_sink
+  kind: VideoStreamSink
+  attributes:
+    frame_rate_limit: 30
+  inputs:
+    frame: .camera_source.frame
+    detections: .detect.detections
+    frame_metrics: .frame_metrics_sink.frame_metrics
+"""
+
+
+# editorconfig-checker-enable
+
+
+class SimpleSegmentationPipeline(SingleModelPipeline):
+    def _to_yaml(self, model_name: str, org_name: str):
+        # editorconfig-checker-disable
+        return f"""---
+nodes:
+- name: camera_source
+  kind: CameraSource
+  outputs:
+  - frame
+- name: segment
+  kind: Segment
+  attributes:
+    model:
+      model:
+        name: {model_name}
+        org: {org_name}
+    frame_rate_limit: 30
+  inputs:
+    frame: .camera_source.frame
+  outputs:
+  - segmentation
+- name: frame_metrics_sink
+  kind: FrameMetricsSink
+  attributes:
+    frame_rate_limit: 30
+  inputs:
+    frame: .camera_source.frame
+  outputs:
+  - frame_metrics
+- name: pipeline_latency_metrics_sink
+  kind: LatencyMetricsSink
+  attributes:
+    name: model
+  inputs:
+    reference: .camera_source.frame
+    segmentation: .segment.segmentation
+    frame_metrics: .frame_metrics_sink.frame_metrics
+- name: video_stream_sink
+  kind: VideoStreamSink
+  attributes:
+    frame_rate_limit: 30
+  inputs:
+    frame: .camera_source.frame
+    frame_metrics: .frame_metrics_sink.frame_metrics
+    segmentation: .segment.segmentation
+"""
+
+
+# editorconfig-checker-enable

gml/preprocessing.py

@@ -0,0 +1,78 @@
+# Copyright 2023- Gimlet Labs, Inc.
+#
+# 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.
+#
+# SPDX-License-Identifier: Apache-2.0
+
+import abc
+from typing import List
+
+import gml.proto.src.api.corepb.v1.model_exec_pb2 as modelexecpb
+
+
+class ImagePreprocessingStep(abc.ABC):
+    @abc.abstractmethod
+    def to_proto(self) -> modelexecpb.ImagePreprocessingStep:
+        pass
+
+
+class LetterboxImage(ImagePreprocessingStep):
+    def to_proto(self) -> modelexecpb.ImagePreprocessingStep:
+        return modelexecpb.ImagePreprocessingStep(
+            kind=modelexecpb.ImagePreprocessingStep.IMAGE_PREPROCESSING_KIND_RESIZE,
+            resize_params=modelexecpb.ImagePreprocessingStep.ImageResizeParams(
+                kind=modelexecpb.ImagePreprocessingStep.ImageResizeParams.IMAGE_RESIZE_KIND_LETTERBOX,
+            ),
+        )
+
+
+class ResizeImage(ImagePreprocessingStep):
+    """ResizeImage resizes the image to the target size without preserving aspect ratio."""
+
+    def to_proto(self) -> modelexecpb.ImagePreprocessingStep:
+        return modelexecpb.ImagePreprocessingStep(
+            kind=modelexecpb.ImagePreprocessingStep.IMAGE_PREPROCESSING_KIND_RESIZE,
+            resize_params=modelexecpb.ImagePreprocessingStep.ImageResizeParams(
+                kind=modelexecpb.ImagePreprocessingStep.ImageResizeParams.IMAGE_RESIZE_KIND_STRETCH,
+            ),
+        )
+
+
+class StandardizeTensor(ImagePreprocessingStep):
+    """StandardizeTensor standardizes the tensor with the given means and standard deviations."""
+
+    def __init__(self, means: List[float], stddevs: List[float]):
+        self.means = means
+        self.stddevs = stddevs
+
+    def to_proto(self) -> modelexecpb.ImagePreprocessingStep:
+        return modelexecpb.ImagePreprocessingStep(
+            kind=modelexecpb.ImagePreprocessingStep.IMAGE_PREPROCESSING_KIND_STANDARDIZE,
+            standardize_params=modelexecpb.ImagePreprocessingStep.ImageStandardizeParams(
+                means=self.means,
+                stddevs=self.stddevs,
+            ),
+        )
+
+
+class ImageToFloatTensor(ImagePreprocessingStep):
+    def __init__(self, scale: bool = True):
+        self.scale = scale
+
+    def to_proto(self) -> modelexecpb.ImagePreprocessingStep:
+        return modelexecpb.ImagePreprocessingStep(
+            kind=modelexecpb.ImagePreprocessingStep.IMAGE_PREPROCESSING_KIND_CONVERT_TO_TENSOR,
+            conversion_params=modelexecpb.ImagePreprocessingStep.ImageConversionParams(
+                scale=self.scale
+            ),
+        )

gml/proto/gogoproto/gogo_pb2.py

@@ -0,0 +1,101 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: gogoproto/gogo.proto
+"""Generated protocol buffer code."""
+from google.protobuf.internal import builder as _builder
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import symbol_database as _symbol_database
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+from google.protobuf import descriptor_pb2 as google_dot_protobuf_dot_descriptor__pb2
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x14gogoproto/gogo.proto\x12\tgogoproto\x1a google/protobuf/descriptor.proto:N\n\x13goproto_enum_prefix\x12\x1c.google.protobuf.EnumOptions\x18\xb1\xe4\x03 \x01(\x08R\x11goprotoEnumPrefix:R\n\x15goproto_enum_stringer\x12\x1c.google.protobuf.EnumOptions\x18\xc5\xe4\x03 \x01(\x08R\x13goprotoEnumStringer:C\n\renum_stringer\x12\x1c.google.protobuf.EnumOptions\x18\xc6\xe4\x03 \x01(\x08R\x0c\x65numStringer:G\n\x0f\x65num_customname\x12\x1c.google.protobuf.EnumOptions\x18\xc7\xe4\x03 \x01(\tR\x0e\x65numCustomname::\n\x08\x65numdecl\x12\x1c.google.protobuf.EnumOptions\x18\xc8\xe4\x03 \x01(\x08R\x08\x65numdecl:V\n\x14\x65numvalue_customname\x12!.google.protobuf.EnumValueOptions\x18\xd1\x83\x04 \x01(\tR\x13\x65numvalueCustomname:N\n\x13goproto_getters_all\x12\x1c.google.protobuf.FileOptions\x18\x99\xec\x03 \x01(\x08R\x11goprotoGettersAll:U\n\x17goproto_enum_prefix_all\x12\x1c.google.protobuf.FileOptions\x18\x9a\xec\x03 \x01(\x08R\x14goprotoEnumPrefixAll:P\n\x14goproto_stringer_all\x12\x1c.google.protobuf.FileOptions\x18\x9b\xec\x03 \x01(\x08R\x12goprotoStringerAll:J\n\x11verbose_equal_all\x12\x1c.google.protobuf.FileOptions\x18\x9c\xec\x03 \x01(\x08R\x0fverboseEqualAll:9\n\x08\x66\x61\x63\x65_all\x12\x1c.google.protobuf.FileOptions\x18\x9d\xec\x03 \x01(\x08R\x07\x66\x61\x63\x65\x41ll:A\n\x0cgostring_all\x12\x1c.google.protobuf.FileOptions\x18\x9e\xec\x03 \x01(\x08R\x0bgostringAll:A\n\x0cpopulate_all\x12\x1c.google.protobuf.FileOptions\x18\x9f\xec\x03 \x01(\x08R\x0bpopulateAll:A\n\x0cstringer_all\x12\x1c.google.protobuf.FileOptions\x18\xa0\xec\x03 \x01(\x08R\x0bstringerAll:?\n\x0bonlyone_all\x12\x1c.google.protobuf.FileOptions\x18\xa1\xec\x03 \x01(\x08R\nonlyoneAll:;\n\tequal_all\x12\x1c.google.protobuf.FileOptions\x18\xa5\xec\x03 \x01(\x08R\x08\x65qualAll:G\n\x0f\x64\x65scription_all\x12\x1c.google.protobuf.FileOptions\x18\xa6\xec\x03 \x01(\x08R\x0e\x64\x65scriptionAll:?\n\x0btestgen_all\x12\x1c.google.protobuf.FileOptions\x18\xa7\xec\x03 \x01(\x08R\ntestgenAll:A\n\x0c\x62\x65nchgen_all\x12\x1c.google.protobuf.FileOptions\x18\xa8\xec\x03 \x01(\x08R\x0b\x62\x65nchgenAll:C\n\rmarshaler_all\x12\x1c.google.protobuf.FileOptions\x18\xa9\xec\x03 \x01(\x08R\x0cmarshalerAll:G\n\x0funmarshaler_all\x12\x1c.google.protobuf.FileOptions\x18\xaa\xec\x03 \x01(\x08R\x0eunmarshalerAll:P\n\x14stable_marshaler_all\x12\x1c.google.protobuf.FileOptions\x18\xab\xec\x03 \x01(\x08R\x12stableMarshalerAll:;\n\tsizer_all\x12\x1c.google.protobuf.FileOptions\x18\xac\xec\x03 \x01(\x08R\x08sizerAll:Y\n\x19goproto_enum_stringer_all\x12\x1c.google.protobuf.FileOptions\x18\xad\xec\x03 \x01(\x08R\x16goprotoEnumStringerAll:J\n\x11\x65num_stringer_all\x12\x1c.google.protobuf.FileOptions\x18\xae\xec\x03 \x01(\x08R\x0f\x65numStringerAll:P\n\x14unsafe_marshaler_all\x12\x1c.google.protobuf.FileOptions\x18\xaf\xec\x03 \x01(\x08R\x12unsafeMarshalerAll:T\n\x16unsafe_unmarshaler_all\x12\x1c.google.protobuf.FileOptions\x18\xb0\xec\x03 \x01(\x08R\x14unsafeUnmarshalerAll:[\n\x1agoproto_extensions_map_all\x12\x1c.google.protobuf.FileOptions\x18\xb1\xec\x03 \x01(\x08R\x17goprotoExtensionsMapAll:X\n\x18goproto_unrecognized_all\x12\x1c.google.protobuf.FileOptions\x18\xb2\xec\x03 \x01(\x08R\x16goprotoUnrecognizedAll:I\n\x10gogoproto_import\x12\x1c.google.protobuf.FileOptions\x18\xb3\xec\x03 \x01(\x08R\x0fgogoprotoImport:E\n\x0eprotosizer_all\x12\x1c.google.protobuf.FileOptions\x18\xb4\xec\x03 \x01(\x08R\rprotosizerAll:?\n\x0b\x63ompare_all\x12\x1c.google.protobuf.FileOptions\x18\xb5\xec\x03 \x01(\x08R\ncompareAll:A\n\x0ctypedecl_all\x12\x1c.google.protobuf.FileOptions\x18\xb6\xec\x03 \x01(\x08R\x0btypedeclAll:A\n\x0c\x65numdecl_all\x12\x1c.google.protobuf.FileOptions\x18\xb7\xec\x03 \x01(\x08R\x0b\x65numdeclAll:Q\n\x14goproto_registration\x12\x1c.google.protobuf.FileOptions\x18\xb8\xec\x03 \x01(\x08R\x13goprotoRegistration:G\n\x0fmessagename_all\x12\x1c.google.protobuf.FileOptions\x18\xb9\xec\x03 \x01(\x08R\x0emessagenameAll:R\n\x15goproto_sizecache_all\x12\x1c.google.protobuf.FileOptions\x18\xba\xec\x03 \x01(\x08R\x13goprotoSizecacheAll:N\n\x13goproto_unkeyed_all\x12\x1c.google.protobuf.FileOptions\x18\xbb\xec\x03 \x01(\x08R\x11goprotoUnkeyedAll:J\n\x0fgoproto_getters\x12\x1f.google.protobuf.MessageOptions\x18\x81\xf4\x03 \x01(\x08R\x0egoprotoGetters:L\n\x10goproto_stringer\x12\x1f.google.protobuf.MessageOptions\x18\x83\xf4\x03 \x01(\x08R\x0fgoprotoStringer:F\n\rverbose_equal\x12\x1f.google.protobuf.MessageOptions\x18\x84\xf4\x03 \x01(\x08R\x0cverboseEqual:5\n\x04\x66\x61\x63\x65\x12\x1f.google.protobuf.MessageOptions\x18\x85\xf4\x03 \x01(\x08R\x04\x66\x61\x63\x65:=\n\x08gostring\x12\x1f.google.protobuf.MessageOptions\x18\x86\xf4\x03 \x01(\x08R\x08gostring:=\n\x08populate\x12\x1f.google.protobuf.MessageOptions\x18\x87\xf4\x03 \x01(\x08R\x08populate:=\n\x08stringer\x12\x1f.google.protobuf.MessageOptions\x18\xc0\x8b\x04 \x01(\x08R\x08stringer:;\n\x07onlyone\x12\x1f.google.protobuf.MessageOptions\x18\x89\xf4\x03 \x01(\x08R\x07onlyone:7\n\x05\x65qual\x12\x1f.google.protobuf.MessageOptions\x18\x8d\xf4\x03 \x01(\x08R\x05\x65qual:C\n\x0b\x64\x65scription\x12\x1f.google.protobuf.MessageOptions\x18\x8e\xf4\x03 \x01(\x08R\x0b\x64\x65scription:;\n\x07testgen\x12\x1f.google.protobuf.MessageOptions\x18\x8f\xf4\x03 \x01(\x08R\x07testgen:=\n\x08\x62\x65nchgen\x12\x1f.google.protobuf.MessageOptions\x18\x90\xf4\x03 \x01(\x08R\x08\x62\x65nchgen:?\n\tmarshaler\x12\x1f.google.protobuf.MessageOptions\x18\x91\xf4\x03 \x01(\x08R\tmarshaler:C\n\x0bunmarshaler\x12\x1f.google.protobuf.MessageOptions\x18\x92\xf4\x03 \x01(\x08R\x0bunmarshaler:L\n\x10stable_marshaler\x12\x1f.google.protobuf.MessageOptions\x18\x93\xf4\x03 \x01(\x08R\x0fstableMarshaler:7\n\x05sizer\x12\x1f.google.protobuf.MessageOptions\x18\x94\xf4\x03 \x01(\x08R\x05sizer:L\n\x10unsafe_marshaler\x12\x1f.google.protobuf.MessageOptions\x18\x97\xf4\x03 \x01(\x08R\x0funsafeMarshaler:P\n\x12unsafe_unmarshaler\x12\x1f.google.protobuf.MessageOptions\x18\x98\xf4\x03 \x01(\x08R\x11unsafeUnmarshaler:W\n\x16goproto_extensions_map\x12\x1f.google.protobuf.MessageOptions\x18\x99\xf4\x03 \x01(\x08R\x14goprotoExtensionsMap:T\n\x14goproto_unrecognized\x12\x1f.google.protobuf.MessageOptions\x18\x9a\xf4\x03 \x01(\x08R\x13goprotoUnrecognized:A\n\nprotosizer\x12\x1f.google.protobuf.MessageOptions\x18\x9c\xf4\x03 \x01(\x08R\nprotosizer:;\n\x07\x63ompare\x12\x1f.google.protobuf.MessageOptions\x18\x9d\xf4\x03 \x01(\x08R\x07\x63ompare:=\n\x08typedecl\x12\x1f.google.protobuf.MessageOptions\x18\x9e\xf4\x03 \x01(\x08R\x08typedecl:C\n\x0bmessagename\x12\x1f.google.protobuf.MessageOptions\x18\xa1\xf4\x03 \x01(\x08R\x0bmessagename:N\n\x11goproto_sizecache\x12\x1f.google.protobuf.MessageOptions\x18\xa2\xf4\x03 \x01(\x08R\x10goprotoSizecache:J\n\x0fgoproto_unkeyed\x12\x1f.google.protobuf.MessageOptions\x18\xa3\xf4\x03 \x01(\x08R\x0egoprotoUnkeyed:;\n\x08nullable\x12\x1d.google.protobuf.FieldOptions\x18\xe9\xfb\x03 \x01(\x08R\x08nullable:5\n\x05\x65mbed\x12\x1d.google.protobuf.FieldOptions\x18\xea\xfb\x03 \x01(\x08R\x05\x65mbed:?\n\ncustomtype\x12\x1d.google.protobuf.FieldOptions\x18\xeb\xfb\x03 \x01(\tR\ncustomtype:?\n\ncustomname\x12\x1d.google.protobuf.FieldOptions\x18\xec\xfb\x03 \x01(\tR\ncustomname:9\n\x07jsontag\x12\x1d.google.protobuf.FieldOptions\x18\xed\xfb\x03 \x01(\tR\x07jsontag:;\n\x08moretags\x12\x1d.google.protobuf.FieldOptions\x18\xee\xfb\x03 \x01(\tR\x08moretags:;\n\x08\x63\x61sttype\x12\x1d.google.protobuf.FieldOptions\x18\xef\xfb\x03 \x01(\tR\x08\x63\x61sttype:9\n\x07\x63\x61stkey\x12\x1d.google.protobuf.FieldOptions\x18\xf0\xfb\x03 \x01(\tR\x07\x63\x61stkey:=\n\tcastvalue\x12\x1d.google.protobuf.FieldOptions\x18\xf1\xfb\x03 \x01(\tR\tcastvalue:9\n\x07stdtime\x12\x1d.google.protobuf.FieldOptions\x18\xf2\xfb\x03 \x01(\x08R\x07stdtime:A\n\x0bstdduration\x12\x1d.google.protobuf.FieldOptions\x18\xf3\xfb\x03 \x01(\x08R\x0bstdduration:?\n\nwktpointer\x12\x1d.google.protobuf.FieldOptions\x18\xf4\xfb\x03 \x01(\x08R\nwktpointerBE\n\x13\x63om.google.protobufB\nGoGoProtosZ\"github.com/gogo/protobuf/gogoproto')
+
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, globals())
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'gogoproto.gogo_pb2', globals())
+if _descriptor._USE_C_DESCRIPTORS == False:
+  google_dot_protobuf_dot_descriptor__pb2.EnumOptions.RegisterExtension(goproto_enum_prefix)
+  google_dot_protobuf_dot_descriptor__pb2.EnumOptions.RegisterExtension(goproto_enum_stringer)
+  google_dot_protobuf_dot_descriptor__pb2.EnumOptions.RegisterExtension(enum_stringer)
+  google_dot_protobuf_dot_descriptor__pb2.EnumOptions.RegisterExtension(enum_customname)
+  google_dot_protobuf_dot_descriptor__pb2.EnumOptions.RegisterExtension(enumdecl)
+  google_dot_protobuf_dot_descriptor__pb2.EnumValueOptions.RegisterExtension(enumvalue_customname)
+  google_dot_protobuf_dot_descriptor__pb2.FileOptions.RegisterExtension(goproto_getters_all)
+  google_dot_protobuf_dot_descriptor__pb2.FileOptions.RegisterExtension(goproto_enum_prefix_all)
+  google_dot_protobuf_dot_descriptor__pb2.FileOptions.RegisterExtension(goproto_stringer_all)
+  google_dot_protobuf_dot_descriptor__pb2.FileOptions.RegisterExtension(verbose_equal_all)
+  google_dot_protobuf_dot_descriptor__pb2.FileOptions.RegisterExtension(face_all)
+  google_dot_protobuf_dot_descriptor__pb2.FileOptions.RegisterExtension(gostring_all)
+  google_dot_protobuf_dot_descriptor__pb2.FileOptions.RegisterExtension(populate_all)
+  google_dot_protobuf_dot_descriptor__pb2.FileOptions.RegisterExtension(stringer_all)
+  google_dot_protobuf_dot_descriptor__pb2.FileOptions.RegisterExtension(onlyone_all)
+  google_dot_protobuf_dot_descriptor__pb2.FileOptions.RegisterExtension(equal_all)
+  google_dot_protobuf_dot_descriptor__pb2.FileOptions.RegisterExtension(description_all)
+  google_dot_protobuf_dot_descriptor__pb2.FileOptions.RegisterExtension(testgen_all)
+  google_dot_protobuf_dot_descriptor__pb2.FileOptions.RegisterExtension(benchgen_all)
+  google_dot_protobuf_dot_descriptor__pb2.FileOptions.RegisterExtension(marshaler_all)
+  google_dot_protobuf_dot_descriptor__pb2.FileOptions.RegisterExtension(unmarshaler_all)
+  google_dot_protobuf_dot_descriptor__pb2.FileOptions.RegisterExtension(stable_marshaler_all)
+  google_dot_protobuf_dot_descriptor__pb2.FileOptions.RegisterExtension(sizer_all)
+  google_dot_protobuf_dot_descriptor__pb2.FileOptions.RegisterExtension(goproto_enum_stringer_all)
+  google_dot_protobuf_dot_descriptor__pb2.FileOptions.RegisterExtension(enum_stringer_all)
+  google_dot_protobuf_dot_descriptor__pb2.FileOptions.RegisterExtension(unsafe_marshaler_all)
+  google_dot_protobuf_dot_descriptor__pb2.FileOptions.RegisterExtension(unsafe_unmarshaler_all)
+  google_dot_protobuf_dot_descriptor__pb2.FileOptions.RegisterExtension(goproto_extensions_map_all)
+  google_dot_protobuf_dot_descriptor__pb2.FileOptions.RegisterExtension(goproto_unrecognized_all)
+  google_dot_protobuf_dot_descriptor__pb2.FileOptions.RegisterExtension(gogoproto_import)
+  google_dot_protobuf_dot_descriptor__pb2.FileOptions.RegisterExtension(protosizer_all)
+  google_dot_protobuf_dot_descriptor__pb2.FileOptions.RegisterExtension(compare_all)
+  google_dot_protobuf_dot_descriptor__pb2.FileOptions.RegisterExtension(typedecl_all)
+  google_dot_protobuf_dot_descriptor__pb2.FileOptions.RegisterExtension(enumdecl_all)
+  google_dot_protobuf_dot_descriptor__pb2.FileOptions.RegisterExtension(goproto_registration)
+  google_dot_protobuf_dot_descriptor__pb2.FileOptions.RegisterExtension(messagename_all)
+  google_dot_protobuf_dot_descriptor__pb2.FileOptions.RegisterExtension(goproto_sizecache_all)
+  google_dot_protobuf_dot_descriptor__pb2.FileOptions.RegisterExtension(goproto_unkeyed_all)
+  google_dot_protobuf_dot_descriptor__pb2.MessageOptions.RegisterExtension(goproto_getters)
+  google_dot_protobuf_dot_descriptor__pb2.MessageOptions.RegisterExtension(goproto_stringer)
+  google_dot_protobuf_dot_descriptor__pb2.MessageOptions.RegisterExtension(verbose_equal)
+  google_dot_protobuf_dot_descriptor__pb2.MessageOptions.RegisterExtension(face)
+  google_dot_protobuf_dot_descriptor__pb2.MessageOptions.RegisterExtension(gostring)
+  google_dot_protobuf_dot_descriptor__pb2.MessageOptions.RegisterExtension(populate)
+  google_dot_protobuf_dot_descriptor__pb2.MessageOptions.RegisterExtension(stringer)
+  google_dot_protobuf_dot_descriptor__pb2.MessageOptions.RegisterExtension(onlyone)
+  google_dot_protobuf_dot_descriptor__pb2.MessageOptions.RegisterExtension(equal)
+  google_dot_protobuf_dot_descriptor__pb2.MessageOptions.RegisterExtension(description)
+  google_dot_protobuf_dot_descriptor__pb2.MessageOptions.RegisterExtension(testgen)
+  google_dot_protobuf_dot_descriptor__pb2.MessageOptions.RegisterExtension(benchgen)
+  google_dot_protobuf_dot_descriptor__pb2.MessageOptions.RegisterExtension(marshaler)
+  google_dot_protobuf_dot_descriptor__pb2.MessageOptions.RegisterExtension(unmarshaler)
+  google_dot_protobuf_dot_descriptor__pb2.MessageOptions.RegisterExtension(stable_marshaler)
+  google_dot_protobuf_dot_descriptor__pb2.MessageOptions.RegisterExtension(sizer)
+  google_dot_protobuf_dot_descriptor__pb2.MessageOptions.RegisterExtension(unsafe_marshaler)
+  google_dot_protobuf_dot_descriptor__pb2.MessageOptions.RegisterExtension(unsafe_unmarshaler)
+  google_dot_protobuf_dot_descriptor__pb2.MessageOptions.RegisterExtension(goproto_extensions_map)
+  google_dot_protobuf_dot_descriptor__pb2.MessageOptions.RegisterExtension(goproto_unrecognized)
+  google_dot_protobuf_dot_descriptor__pb2.MessageOptions.RegisterExtension(protosizer)
+  google_dot_protobuf_dot_descriptor__pb2.MessageOptions.RegisterExtension(compare)
+  google_dot_protobuf_dot_descriptor__pb2.MessageOptions.RegisterExtension(typedecl)
+  google_dot_protobuf_dot_descriptor__pb2.MessageOptions.RegisterExtension(messagename)
+  google_dot_protobuf_dot_descriptor__pb2.MessageOptions.RegisterExtension(goproto_sizecache)
+  google_dot_protobuf_dot_descriptor__pb2.MessageOptions.RegisterExtension(goproto_unkeyed)
+  google_dot_protobuf_dot_descriptor__pb2.FieldOptions.RegisterExtension(nullable)
+  google_dot_protobuf_dot_descriptor__pb2.FieldOptions.RegisterExtension(embed)
+  google_dot_protobuf_dot_descriptor__pb2.FieldOptions.RegisterExtension(customtype)
+  google_dot_protobuf_dot_descriptor__pb2.FieldOptions.RegisterExtension(customname)
+  google_dot_protobuf_dot_descriptor__pb2.FieldOptions.RegisterExtension(jsontag)
+  google_dot_protobuf_dot_descriptor__pb2.FieldOptions.RegisterExtension(moretags)
+  google_dot_protobuf_dot_descriptor__pb2.FieldOptions.RegisterExtension(casttype)
+  google_dot_protobuf_dot_descriptor__pb2.FieldOptions.RegisterExtension(castkey)
+  google_dot_protobuf_dot_descriptor__pb2.FieldOptions.RegisterExtension(castvalue)
+  google_dot_protobuf_dot_descriptor__pb2.FieldOptions.RegisterExtension(stdtime)
+  google_dot_protobuf_dot_descriptor__pb2.FieldOptions.RegisterExtension(stdduration)
+  google_dot_protobuf_dot_descriptor__pb2.FieldOptions.RegisterExtension(wktpointer)
+
+  DESCRIPTOR._options = None
+  DESCRIPTOR._serialized_options = b'\n\023com.google.protobufB\nGoGoProtosZ\"github.com/gogo/protobuf/gogoproto'
+# @@protoc_insertion_point(module_scope)

gml/proto/mediapipe/framework/calculator_contract_test_pb2.py

@@ -0,0 +1,32 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: mediapipe/framework/calculator_contract_test.proto
+"""Generated protocol buffer code."""
+from google.protobuf.internal import builder as _builder
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import symbol_database as _symbol_database
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+from gml.proto.mediapipe.framework import calculator_pb2 as mediapipe_dot_framework_dot_calculator__pb2
+try:
+  mediapipe_dot_framework_dot_calculator__options__pb2 = mediapipe_dot_framework_dot_calculator__pb2.mediapipe_dot_framework_dot_calculator__options__pb2
+except AttributeError:
+  mediapipe_dot_framework_dot_calculator__options__pb2 = mediapipe_dot_framework_dot_calculator__pb2.mediapipe.framework.calculator_options_pb2
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n2mediapipe/framework/calculator_contract_test.proto\x12\tmediapipe\x1a$mediapipe/framework/calculator.proto\"\x9f\x01\n\x1d\x43\x61lculatorContractTestOptions\x12!\n\ntest_field\x18\x01 \x01(\x01:\x02-1R\ttestField2[\n\x03\x65xt\x12\x1c.mediapipe.CalculatorOptions\x18\xb7\xd5\x80Z \x01(\x0b\x32(.mediapipe.CalculatorContractTestOptionsR\x03\x65xtB1Z/github.com/google/mediapipe/mediapipe/framework')
+
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, globals())
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'mediapipe.framework.calculator_contract_test_pb2', globals())
+if _descriptor._USE_C_DESCRIPTORS == False:
+  mediapipe_dot_framework_dot_calculator__options__pb2.CalculatorOptions.RegisterExtension(_CALCULATORCONTRACTTESTOPTIONS.extensions_by_name['ext'])
+
+  DESCRIPTOR._options = None
+  DESCRIPTOR._serialized_options = b'Z/github.com/google/mediapipe/mediapipe/framework'
+  _CALCULATORCONTRACTTESTOPTIONS._serialized_start=104
+  _CALCULATORCONTRACTTESTOPTIONS._serialized_end=263
+# @@protoc_insertion_point(module_scope)

gml/proto/mediapipe/framework/calculator_options_pb2.py

@@ -0,0 +1,28 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: mediapipe/framework/calculator_options.proto
+"""Generated protocol buffer code."""
+from google.protobuf.internal import builder as _builder
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import symbol_database as _symbol_database
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n,mediapipe/framework/calculator_options.proto\x12\tmediapipe\"F\n\x11\x43\x61lculatorOptions\x12%\n\x0cmerge_fields\x18\x01 \x01(\x08\x42\x02\x18\x01R\x0bmergeFields*\n\x08\xa0\x9c\x01\x10\x80\x80\x80\x80\x02\x42\x65\n\x1a\x63om.google.mediapipe.protoB\x16\x43\x61lculatorOptionsProtoZ/github.com/google/mediapipe/mediapipe/framework')
+
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, globals())
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'mediapipe.framework.calculator_options_pb2', globals())
+if _descriptor._USE_C_DESCRIPTORS == False:
+
+  DESCRIPTOR._options = None
+  DESCRIPTOR._serialized_options = b'\n\032com.google.mediapipe.protoB\026CalculatorOptionsProtoZ/github.com/google/mediapipe/mediapipe/framework'
+  _CALCULATOROPTIONS.fields_by_name['merge_fields']._options = None
+  _CALCULATOROPTIONS.fields_by_name['merge_fields']._serialized_options = b'\030\001'
+  _CALCULATOROPTIONS._serialized_start=59
+  _CALCULATOROPTIONS._serialized_end=129
+# @@protoc_insertion_point(module_scope)

gml/proto/mediapipe/framework/calculator_pb2.py

@@ -0,0 +1,56 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: mediapipe/framework/calculator.proto
+"""Generated protocol buffer code."""
+from google.protobuf.internal import builder as _builder
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import symbol_database as _symbol_database
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+from gml.proto.mediapipe.framework import calculator_options_pb2 as mediapipe_dot_framework_dot_calculator__options__pb2
+from google.protobuf import any_pb2 as google_dot_protobuf_dot_any__pb2
+from gml.proto.mediapipe.framework import mediapipe_options_pb2 as mediapipe_dot_framework_dot_mediapipe__options__pb2
+from gml.proto.mediapipe.framework import packet_factory_pb2 as mediapipe_dot_framework_dot_packet__factory__pb2
+from gml.proto.mediapipe.framework import packet_generator_pb2 as mediapipe_dot_framework_dot_packet__generator__pb2
+from gml.proto.mediapipe.framework import status_handler_pb2 as mediapipe_dot_framework_dot_status__handler__pb2
+from gml.proto.mediapipe.framework import stream_handler_pb2 as mediapipe_dot_framework_dot_stream__handler__pb2
+
+from gml.proto.mediapipe.framework.calculator_options_pb2 import *
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n$mediapipe/framework/calculator.proto\x12\tmediapipe\x1a,mediapipe/framework/calculator_options.proto\x1a\x19google/protobuf/any.proto\x1a+mediapipe/framework/mediapipe_options.proto\x1a(mediapipe/framework/packet_factory.proto\x1a*mediapipe/framework/packet_generator.proto\x1a(mediapipe/framework/status_handler.proto\x1a(mediapipe/framework/stream_handler.proto\"o\n\x0e\x45xecutorConfig\x12\x12\n\x04name\x18\x01 \x01(\tR\x04name\x12\x12\n\x04type\x18\x02 \x01(\tR\x04type\x12\x35\n\x07options\x18\x03 \x01(\x0b\x32\x1b.mediapipe.MediaPipeOptionsR\x07options\"\x87\x03\n\x0fInputCollection\x12\x12\n\x04name\x18\x01 \x01(\tR\x04name\x12(\n\x10side_packet_name\x18\x02 \x03(\tR\x0esidePacketName\x12/\n\x13\x65xternal_input_name\x18\xea\x07 \x03(\tR\x11\x65xternalInputName\x12\x43\n\ninput_type\x18\x03 \x01(\x0e\x32$.mediapipe.InputCollection.InputTypeR\tinputType\x12\x1b\n\tfile_name\x18\x04 \x01(\tR\x08\x66ileName\"\xa2\x01\n\tInputType\x12\x16\n\x12INPUT_TYPE_UNKNOWN\x10\x00\x12\x17\n\x13INPUT_TYPE_RECORDIO\x10\x01\x12\x1f\n\x1bINPUT_TYPE_FOREIGN_RECORDIO\x10\x02\x12\x1f\n\x1bINPUT_TYPE_FOREIGN_CSV_TEXT\x10\x03\x12\"\n\x1eINPUT_TYPE_INVALID_UPPER_BOUND\x10\x04\"[\n\x12InputCollectionSet\x12\x45\n\x10input_collection\x18\x01 \x03(\x0b\x32\x1a.mediapipe.InputCollectionR\x0finputCollection\"K\n\x0fInputStreamInfo\x12\x1b\n\ttag_index\x18\x01 \x01(\tR\x08tagIndex\x12\x1b\n\tback_edge\x18\x02 \x01(\x08R\x08\x62\x61\x63kEdge\"\xce\x07\n\x0eProfilerConfig\x12?\n\x1chistogram_interval_size_usec\x18\x01 \x01(\x03R\x19histogramIntervalSizeUsec\x12\x36\n\x17num_histogram_intervals\x18\x02 \x01(\x03R\x15numHistogramIntervals\x12\x41\n\x1b\x65nable_input_output_latency\x18\x03 \x01(\x08\x42\x02\x18\x01R\x18\x65nableInputOutputLatency\x12\'\n\x0f\x65nable_profiler\x18\x04 \x01(\x08R\x0e\x65nableProfiler\x12\x32\n\x15\x65nable_stream_latency\x18\x05 \x01(\x08R\x13\x65nableStreamLatency\x12O\n%use_packet_timestamp_for_added_packet\x18\x06 \x01(\x08R usePacketTimestampForAddedPacket\x12,\n\x12trace_log_capacity\x18\x07 \x01(\x03R\x10traceLogCapacity\x12;\n\x1atrace_event_types_disabled\x18\x08 \x03(\x05R\x17traceEventTypesDisabled\x12$\n\x0etrace_log_path\x18\t \x01(\tR\x0ctraceLogPath\x12&\n\x0ftrace_log_count\x18\n \x01(\x05R\rtraceLogCount\x12\x35\n\x17trace_log_interval_usec\x18\x0b \x01(\x03R\x14traceLogIntervalUsec\x12\x31\n\x15trace_log_margin_usec\x18\x0c \x01(\x03R\x12traceLogMarginUsec\x12=\n\x19trace_log_duration_events\x18\r \x01(\x08\x42\x02\x18\x01R\x16traceLogDurationEvents\x12\x37\n\x18trace_log_interval_count\x18\x0e \x01(\x05R\x15traceLogIntervalCount\x12,\n\x12trace_log_disabled\x18\x0f \x01(\x08R\x10traceLogDisabled\x12#\n\rtrace_enabled\x18\x10 \x01(\x08R\x0ctraceEnabled\x12\x37\n\x18trace_log_instant_events\x18\x11 \x01(\x08R\x15traceLogInstantEvents\x12+\n\x11\x63\x61lculator_filter\x18\x12 \x01(\tR\x10\x63\x61lculatorFilter\"\xfa\x0e\n\x15\x43\x61lculatorGraphConfig\x12\x39\n\x04node\x18\x01 \x03(\x0b\x32%.mediapipe.CalculatorGraphConfig.NodeR\x04node\x12\x45\n\x0epacket_factory\x18\x06 \x03(\x0b\x32\x1e.mediapipe.PacketFactoryConfigR\rpacketFactory\x12O\n\x10packet_generator\x18\x07 \x03(\x0b\x32 .mediapipe.PacketGeneratorConfigB\x02\x18\x01R\x0fpacketGenerator\x12\x1f\n\x0bnum_threads\x18\x08 \x01(\x05R\nnumThreads\x12\x45\n\x0estatus_handler\x18\t \x03(\x0b\x32\x1e.mediapipe.StatusHandlerConfigR\rstatusHandler\x12!\n\x0cinput_stream\x18\n \x03(\tR\x0binputStream\x12#\n\routput_stream\x18\x0f \x03(\tR\x0coutputStream\x12*\n\x11input_side_packet\x18\x10 \x03(\tR\x0finputSidePacket\x12,\n\x12output_side_packet\x18\x11 \x03(\tR\x10outputSidePacket\x12$\n\x0emax_queue_size\x18\x0b \x01(\x05R\x0cmaxQueueSize\x12\'\n\x0freport_deadlock\x18\x15 \x01(\x08R\x0ereportDeadlock\x12U\n\x14input_stream_handler\x18\x0c \x01(\x0b\x32#.mediapipe.InputStreamHandlerConfigR\x12inputStreamHandler\x12X\n\x15output_stream_handler\x18\r \x01(\x0b\x32$.mediapipe.OutputStreamHandlerConfigR\x13outputStreamHandler\x12\x35\n\x08\x65xecutor\x18\x0e \x03(\x0b\x32\x19.mediapipe.ExecutorConfigR\x08\x65xecutor\x12\x42\n\x0fprofiler_config\x18\x12 \x01(\x0b\x32\x19.mediapipe.ProfilerConfigR\x0eprofilerConfig\x12\x18\n\x07package\x18\x13 \x01(\tR\x07package\x12\x12\n\x04type\x18\x14 \x01(\tR\x04type\x12\x36\n\x07options\x18\xe9\x07 \x01(\x0b\x32\x1b.mediapipe.MediaPipeOptionsR\x07options\x12:\n\rgraph_options\x18\xea\x07 \x03(\x0b\x32\x14.google.protobuf.AnyR\x0cgraphOptions\x1a\xe6\x06\n\x04Node\x12\x12\n\x04name\x18\x01 \x01(\tR\x04name\x12\x1e\n\ncalculator\x18\x02 \x01(\tR\ncalculator\x12!\n\x0cinput_stream\x18\x03 \x03(\tR\x0binputStream\x12#\n\routput_stream\x18\x04 \x03(\tR\x0coutputStream\x12*\n\x11input_side_packet\x18\x05 \x03(\tR\x0finputSidePacket\x12,\n\x12output_side_packet\x18\x06 \x03(\tR\x10outputSidePacket\x12\x36\n\x07options\x18\x07 \x01(\x0b\x32\x1c.mediapipe.CalculatorOptionsR\x07options\x12\x37\n\x0cnode_options\x18\x08 \x03(\x0b\x32\x14.google.protobuf.AnyR\x0bnodeOptions\x12!\n\x0csource_layer\x18\t \x01(\x05R\x0bsourceLayer\x12(\n\x10\x62uffer_size_hint\x18\n \x01(\x05R\x0e\x62ufferSizeHint\x12U\n\x14input_stream_handler\x18\x0b \x01(\x0b\x32#.mediapipe.InputStreamHandlerConfigR\x12inputStreamHandler\x12X\n\x15output_stream_handler\x18\x0c \x01(\x0b\x32$.mediapipe.OutputStreamHandlerConfigR\x13outputStreamHandler\x12\x46\n\x11input_stream_info\x18\r \x03(\x0b\x32\x1a.mediapipe.InputStreamInfoR\x0finputStreamInfo\x12\x1a\n\x08\x65xecutor\x18\x0e \x01(\tR\x08\x65xecutor\x12\x46\n\x0fprofiler_config\x18\x0f \x01(\x0b\x32\x19.mediapipe.ProfilerConfigB\x02\x18\x01R\x0eprofilerConfig\x12\"\n\rmax_in_flight\x18\x10 \x01(\x05R\x0bmaxInFlight\x12!\n\x0coption_value\x18\x11 \x03(\tR\x0boptionValue\x12&\n\x0e\x65xternal_input\x18\xed\x07 \x03(\tR\rexternalInputB^\n\x1a\x63om.google.mediapipe.protoB\x0f\x43\x61lculatorProtoZ/github.com/google/mediapipe/mediapipe/frameworkP\x00\x62\x06proto3')
+
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, globals())
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'mediapipe.framework.calculator_pb2', globals())
+if _descriptor._USE_C_DESCRIPTORS == False:
+
+  DESCRIPTOR._options = None
+  DESCRIPTOR._serialized_options = b'\n\032com.google.mediapipe.protoB\017CalculatorProtoZ/github.com/google/mediapipe/mediapipe/framework'
+  _PROFILERCONFIG.fields_by_name['enable_input_output_latency']._options = None
+  _PROFILERCONFIG.fields_by_name['enable_input_output_latency']._serialized_options = b'\030\001'
+  _PROFILERCONFIG.fields_by_name['trace_log_duration_events']._options = None
+  _PROFILERCONFIG.fields_by_name['trace_log_duration_events']._serialized_options = b'\030\001'
+  _CALCULATORGRAPHCONFIG_NODE.fields_by_name['profiler_config']._options = None
+  _CALCULATORGRAPHCONFIG_NODE.fields_by_name['profiler_config']._serialized_options = b'\030\001'
+  _CALCULATORGRAPHCONFIG.fields_by_name['packet_generator']._options = None
+  _CALCULATORGRAPHCONFIG.fields_by_name['packet_generator']._serialized_options = b'\030\001'
+  _EXECUTORCONFIG._serialized_start=339
+  _EXECUTORCONFIG._serialized_end=450
+  _INPUTCOLLECTION._serialized_start=453
+  _INPUTCOLLECTION._serialized_end=844
+  _INPUTCOLLECTION_INPUTTYPE._serialized_start=682
+  _INPUTCOLLECTION_INPUTTYPE._serialized_end=844
+  _INPUTCOLLECTIONSET._serialized_start=846
+  _INPUTCOLLECTIONSET._serialized_end=937
+  _INPUTSTREAMINFO._serialized_start=939
+  _INPUTSTREAMINFO._serialized_end=1014
+  _PROFILERCONFIG._serialized_start=1017
+  _PROFILERCONFIG._serialized_end=1991
+  _CALCULATORGRAPHCONFIG._serialized_start=1994
+  _CALCULATORGRAPHCONFIG._serialized_end=3908
+  _CALCULATORGRAPHCONFIG_NODE._serialized_start=3038
+  _CALCULATORGRAPHCONFIG_NODE._serialized_end=3908
+# @@protoc_insertion_point(module_scope)

gml/proto/mediapipe/framework/calculator_profile_pb2.py

@@ -0,0 +1,47 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: mediapipe/framework/calculator_profile.proto
+"""Generated protocol buffer code."""
+from google.protobuf.internal import builder as _builder
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import symbol_database as _symbol_database
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+from gml.proto.mediapipe.framework import calculator_pb2 as mediapipe_dot_framework_dot_calculator__pb2
+try:
+  mediapipe_dot_framework_dot_calculator__options__pb2 = mediapipe_dot_framework_dot_calculator__pb2.mediapipe_dot_framework_dot_calculator__options__pb2
+except AttributeError:
+  mediapipe_dot_framework_dot_calculator__options__pb2 = mediapipe_dot_framework_dot_calculator__pb2.mediapipe.framework.calculator_options_pb2
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n,mediapipe/framework/calculator_profile.proto\x12\tmediapipe\x1a$mediapipe/framework/calculator.proto\"\x9d\x01\n\rTimeHistogram\x12\x17\n\x05total\x18\x01 \x01(\x03:\x01\x30R\x05total\x12\x35\n\x12interval_size_usec\x18\x02 \x01(\x03:\x07\x31\x30\x30\x30\x30\x30\x30R\x10intervalSizeUsec\x12&\n\rnum_intervals\x18\x03 \x01(\x03:\x01\x31R\x0cnumIntervals\x12\x14\n\x05\x63ount\x18\x04 \x03(\x03R\x05\x63ount\"{\n\rStreamProfile\x12\x12\n\x04name\x18\x01 \x01(\tR\x04name\x12\"\n\tback_edge\x18\x02 \x01(\x08:\x05\x66\x61lseR\x08\x62\x61\x63kEdge\x12\x32\n\x07latency\x18\x03 \x01(\x0b\x32\x18.mediapipe.TimeHistogramR\x07latency\"\xa4\x03\n\x11\x43\x61lculatorProfile\x12\x12\n\x04name\x18\x01 \x01(\tR\x04name\x12$\n\x0copen_runtime\x18\x02 \x01(\x03:\x01\x30R\x0bopenRuntime\x12&\n\rclose_runtime\x18\x03 \x01(\x03:\x01\x30R\x0c\x63loseRuntime\x12\x41\n\x0fprocess_runtime\x18\x04 \x01(\x0b\x32\x18.mediapipe.TimeHistogramR\x0eprocessRuntime\x12L\n\x15process_input_latency\x18\x05 \x01(\x0b\x32\x18.mediapipe.TimeHistogramR\x13processInputLatency\x12N\n\x16process_output_latency\x18\x06 \x01(\x0b\x32\x18.mediapipe.TimeHistogramR\x14processOutputLatency\x12L\n\x15input_stream_profiles\x18\x07 \x03(\x0b\x32\x18.mediapipe.StreamProfileR\x13inputStreamProfiles\"\xb2\x0b\n\nGraphTrace\x12\x1b\n\tbase_time\x18\x01 \x01(\x03R\x08\x62\x61seTime\x12%\n\x0e\x62\x61se_timestamp\x18\x02 \x01(\x03R\rbaseTimestamp\x12\'\n\x0f\x63\x61lculator_name\x18\x03 \x03(\tR\x0e\x63\x61lculatorName\x12\x1f\n\x0bstream_name\x18\x04 \x03(\tR\nstreamName\x12P\n\x10\x63\x61lculator_trace\x18\x05 \x03(\x0b\x32%.mediapipe.GraphTrace.CalculatorTraceR\x0f\x63\x61lculatorTrace\x1a\xd5\x01\n\x0bStreamTrace\x12\x1d\n\nstart_time\x18\x01 \x01(\x03R\tstartTime\x12\x1f\n\x0b\x66inish_time\x18\x02 \x01(\x03R\nfinishTime\x12)\n\x10packet_timestamp\x18\x03 \x01(\x03R\x0fpacketTimestamp\x12\x1b\n\tstream_id\x18\x04 \x01(\x05R\x08streamId\x12\x1f\n\tpacket_id\x18\x05 \x01(\x03\x42\x02\x18\x01R\x08packetId\x12\x1d\n\nevent_data\x18\x06 \x01(\x03R\teventData\x1a\xfa\x02\n\x0f\x43\x61lculatorTrace\x12\x17\n\x07node_id\x18\x01 \x01(\x05R\x06nodeId\x12\'\n\x0finput_timestamp\x18\x02 \x01(\x03R\x0einputTimestamp\x12>\n\nevent_type\x18\x03 \x01(\x0e\x32\x1f.mediapipe.GraphTrace.EventTypeR\teventType\x12\x1d\n\nstart_time\x18\x04 \x01(\x03R\tstartTime\x12\x1f\n\x0b\x66inish_time\x18\x05 \x01(\x03R\nfinishTime\x12\x42\n\x0binput_trace\x18\x06 \x03(\x0b\x32!.mediapipe.GraphTrace.StreamTraceR\ninputTrace\x12\x44\n\x0coutput_trace\x18\x07 \x03(\x0b\x32!.mediapipe.GraphTrace.StreamTraceR\x0boutputTrace\x12\x1b\n\tthread_id\x18\x08 \x01(\x05R\x08threadId\"\xee\x04\n\tEventType\x12\x16\n\x12\x45VENT_TYPE_UNKNOWN\x10\x00\x12\x13\n\x0f\x45VENT_TYPE_OPEN\x10\x01\x12\x16\n\x12\x45VENT_TYPE_PROCESS\x10\x02\x12\x14\n\x10\x45VENT_TYPE_CLOSE\x10\x03\x12\x18\n\x14\x45VENT_TYPE_NOT_READY\x10\x04\x12 \n\x1c\x45VENT_TYPE_READY_FOR_PROCESS\x10\x05\x12\x1e\n\x1a\x45VENT_TYPE_READY_FOR_CLOSE\x10\x06\x12\x18\n\x14\x45VENT_TYPE_THROTTLED\x10\x07\x12\x1a\n\x16\x45VENT_TYPE_UNTHROTTLED\x10\x08\x12\x1c\n\x18\x45VENT_TYPE_CPU_TASK_USER\x10\t\x12\x1e\n\x1a\x45VENT_TYPE_CPU_TASK_SYSTEM\x10\n\x12\x17\n\x13\x45VENT_TYPE_GPU_TASK\x10\x0b\x12\x17\n\x13\x45VENT_TYPE_DSP_TASK\x10\x0c\x12\x17\n\x13\x45VENT_TYPE_TPU_TASK\x10\r\x12\x1e\n\x1a\x45VENT_TYPE_GPU_CALIBRATION\x10\x0e\x12\x1c\n\x18\x45VENT_TYPE_PACKET_QUEUED\x10\x0f\x12\x1e\n\x1a\x45VENT_TYPE_GPU_TASK_INVOKE\x10\x10\x12\x1e\n\x1a\x45VENT_TYPE_TPU_TASK_INVOKE\x10\x11\x12\x1e\n\x1a\x45VENT_TYPE_CPU_TASK_INVOKE\x10\x12\x12\'\n#EVENT_TYPE_GPU_TASK_INVOKE_ADVANCED\x10\x13\x12$\n EVENT_TYPE_TPU_TASK_INVOKE_ASYNC\x10\x14\"\xcf\x01\n\x0cGraphProfile\x12\x36\n\x0bgraph_trace\x18\x01 \x03(\x0b\x32\x15.mediapipe.GraphTraceR\ngraphTrace\x12M\n\x13\x63\x61lculator_profiles\x18\x02 \x03(\x0b\x32\x1c.mediapipe.CalculatorProfileR\x12\x63\x61lculatorProfiles\x12\x38\n\x06\x63onfig\x18\x03 \x01(\x0b\x32 .mediapipe.CalculatorGraphConfigR\x06\x63onfigBe\n\x1a\x63om.google.mediapipe.protoB\x16\x43\x61lculatorProfileProtoZ/github.com/google/mediapipe/mediapipe/framework')
+
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, globals())
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'mediapipe.framework.calculator_profile_pb2', globals())
+if _descriptor._USE_C_DESCRIPTORS == False:
+
+  DESCRIPTOR._options = None
+  DESCRIPTOR._serialized_options = b'\n\032com.google.mediapipe.protoB\026CalculatorProfileProtoZ/github.com/google/mediapipe/mediapipe/framework'
+  _GRAPHTRACE_STREAMTRACE.fields_by_name['packet_id']._options = None
+  _GRAPHTRACE_STREAMTRACE.fields_by_name['packet_id']._serialized_options = b'\030\001'
+  _TIMEHISTOGRAM._serialized_start=98
+  _TIMEHISTOGRAM._serialized_end=255
+  _STREAMPROFILE._serialized_start=257
+  _STREAMPROFILE._serialized_end=380
+  _CALCULATORPROFILE._serialized_start=383
+  _CALCULATORPROFILE._serialized_end=803
+  _GRAPHTRACE._serialized_start=806
+  _GRAPHTRACE._serialized_end=2264
+  _GRAPHTRACE_STREAMTRACE._serialized_start=1045
+  _GRAPHTRACE_STREAMTRACE._serialized_end=1258
+  _GRAPHTRACE_CALCULATORTRACE._serialized_start=1261
+  _GRAPHTRACE_CALCULATORTRACE._serialized_end=1639
+  _GRAPHTRACE_EVENTTYPE._serialized_start=1642
+  _GRAPHTRACE_EVENTTYPE._serialized_end=2264
+  _GRAPHPROFILE._serialized_start=2267
+  _GRAPHPROFILE._serialized_end=2474
+# @@protoc_insertion_point(module_scope)

gml/proto/mediapipe/framework/mediapipe_options_pb2.py

@@ -0,0 +1,26 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: mediapipe/framework/mediapipe_options.proto
+"""Generated protocol buffer code."""
+from google.protobuf.internal import builder as _builder
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import symbol_database as _symbol_database
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n+mediapipe/framework/mediapipe_options.proto\x12\tmediapipe\"\x1e\n\x10MediaPipeOptions*\n\x08\xa0\x9c\x01\x10\x80\x80\x80\x80\x02\x42\x64\n\x1a\x63om.google.mediapipe.protoB\x15MediaPipeOptionsProtoZ/github.com/google/mediapipe/mediapipe/framework')
+
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, globals())
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'mediapipe.framework.mediapipe_options_pb2', globals())
+if _descriptor._USE_C_DESCRIPTORS == False:
+
+  DESCRIPTOR._options = None
+  DESCRIPTOR._serialized_options = b'\n\032com.google.mediapipe.protoB\025MediaPipeOptionsProtoZ/github.com/google/mediapipe/mediapipe/framework'
+  _MEDIAPIPEOPTIONS._serialized_start=58
+  _MEDIAPIPEOPTIONS._serialized_end=88
+# @@protoc_insertion_point(module_scope)

gml/proto/mediapipe/framework/packet_factory_pb2.py

@@ -0,0 +1,30 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: mediapipe/framework/packet_factory.proto
+"""Generated protocol buffer code."""
+from google.protobuf.internal import builder as _builder
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import symbol_database as _symbol_database
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n(mediapipe/framework/packet_factory.proto\x12\tmediapipe\"\"\n\x14PacketFactoryOptions*\n\x08\xa0\x9c\x01\x10\x80\x80\x80\x80\x02\"\xcf\x01\n\x13PacketFactoryConfig\x12%\n\x0epacket_factory\x18\x01 \x01(\tR\rpacketFactory\x12,\n\x12output_side_packet\x18\x02 \x01(\tR\x10outputSidePacket\x12(\n\x0f\x65xternal_output\x18\xea\x07 \x01(\tR\x0e\x65xternalOutput\x12\x39\n\x07options\x18\x03 \x01(\x0b\x32\x1f.mediapipe.PacketFactoryOptionsR\x07options\"M\n\x13PacketManagerConfig\x12\x36\n\x06packet\x18\x01 \x03(\x0b\x32\x1e.mediapipe.PacketFactoryConfigR\x06packetBa\n\x1a\x63om.google.mediapipe.protoB\x12PacketFactoryProtoZ/github.com/google/mediapipe/mediapipe/framework')
+
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, globals())
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'mediapipe.framework.packet_factory_pb2', globals())
+if _descriptor._USE_C_DESCRIPTORS == False:
+
+  DESCRIPTOR._options = None
+  DESCRIPTOR._serialized_options = b'\n\032com.google.mediapipe.protoB\022PacketFactoryProtoZ/github.com/google/mediapipe/mediapipe/framework'
+  _PACKETFACTORYOPTIONS._serialized_start=55
+  _PACKETFACTORYOPTIONS._serialized_end=89
+  _PACKETFACTORYCONFIG._serialized_start=92
+  _PACKETFACTORYCONFIG._serialized_end=299
+  _PACKETMANAGERCONFIG._serialized_start=301
+  _PACKETMANAGERCONFIG._serialized_end=378
+# @@protoc_insertion_point(module_scope)

gml/proto/mediapipe/framework/packet_generator_pb2.py

@@ -0,0 +1,32 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: mediapipe/framework/packet_generator.proto
+"""Generated protocol buffer code."""
+from google.protobuf.internal import builder as _builder
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import symbol_database as _symbol_database
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n*mediapipe/framework/packet_generator.proto\x12\tmediapipe\"Q\n\x16PacketGeneratorOptions\x12\'\n\x0cmerge_fields\x18\x01 \x01(\x08:\x04trueR\x0bmergeFields*\n\x08\xa0\x9c\x01\x10\x80\x80\x80\x80\x02:\x02\x18\x01\"\xaf\x02\n\x15PacketGeneratorConfig\x12)\n\x10packet_generator\x18\x01 \x01(\tR\x0fpacketGenerator\x12*\n\x11input_side_packet\x18\x02 \x03(\tR\x0finputSidePacket\x12&\n\x0e\x65xternal_input\x18\xea\x07 \x03(\tR\rexternalInput\x12,\n\x12output_side_packet\x18\x03 \x03(\tR\x10outputSidePacket\x12(\n\x0f\x65xternal_output\x18\xeb\x07 \x03(\tR\x0e\x65xternalOutput\x12;\n\x07options\x18\x04 \x01(\x0b\x32!.mediapipe.PacketGeneratorOptionsR\x07options:\x02\x18\x01\x42\x63\n\x1a\x63om.google.mediapipe.protoB\x14PacketGeneratorProtoZ/github.com/google/mediapipe/mediapipe/framework')
+
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, globals())
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'mediapipe.framework.packet_generator_pb2', globals())
+if _descriptor._USE_C_DESCRIPTORS == False:
+
+  DESCRIPTOR._options = None
+  DESCRIPTOR._serialized_options = b'\n\032com.google.mediapipe.protoB\024PacketGeneratorProtoZ/github.com/google/mediapipe/mediapipe/framework'
+  _PACKETGENERATOROPTIONS._options = None
+  _PACKETGENERATOROPTIONS._serialized_options = b'\030\001'
+  _PACKETGENERATORCONFIG._options = None
+  _PACKETGENERATORCONFIG._serialized_options = b'\030\001'
+  _PACKETGENERATOROPTIONS._serialized_start=57
+  _PACKETGENERATOROPTIONS._serialized_end=138
+  _PACKETGENERATORCONFIG._serialized_start=141
+  _PACKETGENERATORCONFIG._serialized_end=444
+# @@protoc_insertion_point(module_scope)