Initial commit

This commit is contained in:
Untriex Programming
2021-08-31 22:06:02 +02:00
commit 9b6723e11e
5142 changed files with 1455625 additions and 0 deletions

View File

@@ -0,0 +1,176 @@
# -*- coding: utf-8 -*-
# Generated by the protocol buffer compiler. DO NOT EDIT!
# source: mediapipe/calculators/tflite/ssd_anchors_calculator.proto
import sys
_b=sys.version_info[0]<3 and (lambda x:x) or (lambda x:x.encode('latin1'))
from google.protobuf import descriptor as _descriptor
from google.protobuf import message as _message
from google.protobuf import reflection as _reflection
from google.protobuf import symbol_database as _symbol_database
# @@protoc_insertion_point(imports)
_sym_db = _symbol_database.Default()
from 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.FileDescriptor(
name='mediapipe/calculators/tflite/ssd_anchors_calculator.proto',
package='mediapipe',
syntax='proto2',
serialized_options=None,
serialized_pb=_b('\n9mediapipe/calculators/tflite/ssd_anchors_calculator.proto\x12\tmediapipe\x1a$mediapipe/framework/calculator.proto\"\xf8\x03\n\x1bSsdAnchorsCalculatorOptions\x12\x18\n\x10input_size_width\x18\x01 \x02(\x05\x12\x19\n\x11input_size_height\x18\x02 \x02(\x05\x12\x11\n\tmin_scale\x18\x03 \x02(\x02\x12\x11\n\tmax_scale\x18\x04 \x02(\x02\x12\x1c\n\x0f\x61nchor_offset_x\x18\x05 \x02(\x02:\x03\x30.5\x12\x1c\n\x0f\x61nchor_offset_y\x18\x06 \x02(\x02:\x03\x30.5\x12\x12\n\nnum_layers\x18\x07 \x02(\x05\x12\x19\n\x11\x66\x65\x61ture_map_width\x18\x08 \x03(\x05\x12\x1a\n\x12\x66\x65\x61ture_map_height\x18\t \x03(\x05\x12\x0f\n\x07strides\x18\n \x03(\x05\x12\x15\n\raspect_ratios\x18\x0b \x03(\x02\x12+\n\x1creduce_boxes_in_lowest_layer\x18\x0c \x01(\x08:\x05\x66\x61lse\x12*\n\x1finterpolated_scale_aspect_ratio\x18\r \x01(\x02:\x01\x31\x12 \n\x11\x66ixed_anchor_size\x18\x0e \x01(\x08:\x05\x66\x61lse2T\n\x03\x65xt\x12\x1c.mediapipe.CalculatorOptions\x18\xff\xb8\xf3u \x01(\x0b\x32&.mediapipe.SsdAnchorsCalculatorOptions')
,
dependencies=[mediapipe_dot_framework_dot_calculator__pb2.DESCRIPTOR,])
_SSDANCHORSCALCULATOROPTIONS = _descriptor.Descriptor(
name='SsdAnchorsCalculatorOptions',
full_name='mediapipe.SsdAnchorsCalculatorOptions',
filename=None,
file=DESCRIPTOR,
containing_type=None,
fields=[
_descriptor.FieldDescriptor(
name='input_size_width', full_name='mediapipe.SsdAnchorsCalculatorOptions.input_size_width', index=0,
number=1, type=5, cpp_type=1, label=2,
has_default_value=False, default_value=0,
message_type=None, enum_type=None, containing_type=None,
is_extension=False, extension_scope=None,
serialized_options=None, file=DESCRIPTOR),
_descriptor.FieldDescriptor(
name='input_size_height', full_name='mediapipe.SsdAnchorsCalculatorOptions.input_size_height', index=1,
number=2, type=5, cpp_type=1, label=2,
has_default_value=False, default_value=0,
message_type=None, enum_type=None, containing_type=None,
is_extension=False, extension_scope=None,
serialized_options=None, file=DESCRIPTOR),
_descriptor.FieldDescriptor(
name='min_scale', full_name='mediapipe.SsdAnchorsCalculatorOptions.min_scale', index=2,
number=3, type=2, cpp_type=6, label=2,
has_default_value=False, default_value=float(0),
message_type=None, enum_type=None, containing_type=None,
is_extension=False, extension_scope=None,
serialized_options=None, file=DESCRIPTOR),
_descriptor.FieldDescriptor(
name='max_scale', full_name='mediapipe.SsdAnchorsCalculatorOptions.max_scale', index=3,
number=4, type=2, cpp_type=6, label=2,
has_default_value=False, default_value=float(0),
message_type=None, enum_type=None, containing_type=None,
is_extension=False, extension_scope=None,
serialized_options=None, file=DESCRIPTOR),
_descriptor.FieldDescriptor(
name='anchor_offset_x', full_name='mediapipe.SsdAnchorsCalculatorOptions.anchor_offset_x', index=4,
number=5, type=2, cpp_type=6, label=2,
has_default_value=True, default_value=float(0.5),
message_type=None, enum_type=None, containing_type=None,
is_extension=False, extension_scope=None,
serialized_options=None, file=DESCRIPTOR),
_descriptor.FieldDescriptor(
name='anchor_offset_y', full_name='mediapipe.SsdAnchorsCalculatorOptions.anchor_offset_y', index=5,
number=6, type=2, cpp_type=6, label=2,
has_default_value=True, default_value=float(0.5),
message_type=None, enum_type=None, containing_type=None,
is_extension=False, extension_scope=None,
serialized_options=None, file=DESCRIPTOR),
_descriptor.FieldDescriptor(
name='num_layers', full_name='mediapipe.SsdAnchorsCalculatorOptions.num_layers', index=6,
number=7, type=5, cpp_type=1, label=2,
has_default_value=False, default_value=0,
message_type=None, enum_type=None, containing_type=None,
is_extension=False, extension_scope=None,
serialized_options=None, file=DESCRIPTOR),
_descriptor.FieldDescriptor(
name='feature_map_width', full_name='mediapipe.SsdAnchorsCalculatorOptions.feature_map_width', index=7,
number=8, type=5, cpp_type=1, label=3,
has_default_value=False, default_value=[],
message_type=None, enum_type=None, containing_type=None,
is_extension=False, extension_scope=None,
serialized_options=None, file=DESCRIPTOR),
_descriptor.FieldDescriptor(
name='feature_map_height', full_name='mediapipe.SsdAnchorsCalculatorOptions.feature_map_height', index=8,
number=9, type=5, cpp_type=1, label=3,
has_default_value=False, default_value=[],
message_type=None, enum_type=None, containing_type=None,
is_extension=False, extension_scope=None,
serialized_options=None, file=DESCRIPTOR),
_descriptor.FieldDescriptor(
name='strides', full_name='mediapipe.SsdAnchorsCalculatorOptions.strides', index=9,
number=10, type=5, cpp_type=1, label=3,
has_default_value=False, default_value=[],
message_type=None, enum_type=None, containing_type=None,
is_extension=False, extension_scope=None,
serialized_options=None, file=DESCRIPTOR),
_descriptor.FieldDescriptor(
name='aspect_ratios', full_name='mediapipe.SsdAnchorsCalculatorOptions.aspect_ratios', index=10,
number=11, type=2, cpp_type=6, label=3,
has_default_value=False, default_value=[],
message_type=None, enum_type=None, containing_type=None,
is_extension=False, extension_scope=None,
serialized_options=None, file=DESCRIPTOR),
_descriptor.FieldDescriptor(
name='reduce_boxes_in_lowest_layer', full_name='mediapipe.SsdAnchorsCalculatorOptions.reduce_boxes_in_lowest_layer', index=11,
number=12, type=8, cpp_type=7, label=1,
has_default_value=True, default_value=False,
message_type=None, enum_type=None, containing_type=None,
is_extension=False, extension_scope=None,
serialized_options=None, file=DESCRIPTOR),
_descriptor.FieldDescriptor(
name='interpolated_scale_aspect_ratio', full_name='mediapipe.SsdAnchorsCalculatorOptions.interpolated_scale_aspect_ratio', index=12,
number=13, type=2, cpp_type=6, label=1,
has_default_value=True, default_value=float(1),
message_type=None, enum_type=None, containing_type=None,
is_extension=False, extension_scope=None,
serialized_options=None, file=DESCRIPTOR),
_descriptor.FieldDescriptor(
name='fixed_anchor_size', full_name='mediapipe.SsdAnchorsCalculatorOptions.fixed_anchor_size', index=13,
number=14, type=8, cpp_type=7, label=1,
has_default_value=True, default_value=False,
message_type=None, enum_type=None, containing_type=None,
is_extension=False, extension_scope=None,
serialized_options=None, file=DESCRIPTOR),
],
extensions=[
_descriptor.FieldDescriptor(
name='ext', full_name='mediapipe.SsdAnchorsCalculatorOptions.ext', index=0,
number=247258239, type=11, cpp_type=10, label=1,
has_default_value=False, default_value=None,
message_type=None, enum_type=None, containing_type=None,
is_extension=True, extension_scope=None,
serialized_options=None, file=DESCRIPTOR),
],
nested_types=[],
enum_types=[
],
serialized_options=None,
is_extendable=False,
syntax='proto2',
extension_ranges=[],
oneofs=[
],
serialized_start=111,
serialized_end=615,
)
DESCRIPTOR.message_types_by_name['SsdAnchorsCalculatorOptions'] = _SSDANCHORSCALCULATOROPTIONS
_sym_db.RegisterFileDescriptor(DESCRIPTOR)
SsdAnchorsCalculatorOptions = _reflection.GeneratedProtocolMessageType('SsdAnchorsCalculatorOptions', (_message.Message,), dict(
DESCRIPTOR = _SSDANCHORSCALCULATOROPTIONS,
__module__ = 'mediapipe.calculators.tflite.ssd_anchors_calculator_pb2'
# @@protoc_insertion_point(class_scope:mediapipe.SsdAnchorsCalculatorOptions)
))
_sym_db.RegisterMessage(SsdAnchorsCalculatorOptions)
_SSDANCHORSCALCULATOROPTIONS.extensions_by_name['ext'].message_type = _SSDANCHORSCALCULATOROPTIONS
mediapipe_dot_framework_dot_calculator__options__pb2.CalculatorOptions.RegisterExtension(_SSDANCHORSCALCULATOROPTIONS.extensions_by_name['ext'])
# @@protoc_insertion_point(module_scope)

View File

@@ -0,0 +1,188 @@
# -*- coding: utf-8 -*-
# Generated by the protocol buffer compiler. DO NOT EDIT!
# source: mediapipe/calculators/tflite/tflite_converter_calculator.proto
import sys
_b=sys.version_info[0]<3 and (lambda x:x) or (lambda x:x.encode('latin1'))
from google.protobuf import descriptor as _descriptor
from google.protobuf import message as _message
from google.protobuf import reflection as _reflection
from google.protobuf import symbol_database as _symbol_database
# @@protoc_insertion_point(imports)
_sym_db = _symbol_database.Default()
from 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.FileDescriptor(
name='mediapipe/calculators/tflite/tflite_converter_calculator.proto',
package='mediapipe',
syntax='proto2',
serialized_options=None,
serialized_pb=_b('\n>mediapipe/calculators/tflite/tflite_converter_calculator.proto\x12\tmediapipe\x1a$mediapipe/framework/calculator.proto\"\x84\x04\n TfLiteConverterCalculatorOptions\x12\x19\n\x0bzero_center\x18\x01 \x01(\x08:\x04true\x12\'\n\x18use_custom_normalization\x18\x06 \x01(\x08:\x05\x66\x61lse\x12\x16\n\ncustom_div\x18\x07 \x01(\x02:\x02-1\x12\x16\n\ncustom_sub\x18\x08 \x01(\x02:\x02-1\x12\x1e\n\x0f\x66lip_vertically\x18\x02 \x01(\x08:\x05\x66\x61lse\x12\x1b\n\x10max_num_channels\x18\x03 \x01(\x05:\x01\x33\x12\x1f\n\x10row_major_matrix\x18\x04 \x01(\x08:\x05\x66\x61lse\x12$\n\x15use_quantized_tensors\x18\x05 \x01(\x08:\x05\x66\x61lse\x12_\n\x19output_tensor_float_range\x18\t \x01(\x0b\x32<.mediapipe.TfLiteConverterCalculatorOptions.TensorFloatRange\x1a,\n\x10TensorFloatRange\x12\x0b\n\x03min\x18\x01 \x01(\x02\x12\x0b\n\x03max\x18\x02 \x01(\x02\x32Y\n\x03\x65xt\x12\x1c.mediapipe.CalculatorOptions\x18\xc5\xc3\x9bu \x01(\x0b\x32+.mediapipe.TfLiteConverterCalculatorOptions')
,
dependencies=[mediapipe_dot_framework_dot_calculator__pb2.DESCRIPTOR,])
_TFLITECONVERTERCALCULATOROPTIONS_TENSORFLOATRANGE = _descriptor.Descriptor(
name='TensorFloatRange',
full_name='mediapipe.TfLiteConverterCalculatorOptions.TensorFloatRange',
filename=None,
file=DESCRIPTOR,
containing_type=None,
fields=[
_descriptor.FieldDescriptor(
name='min', full_name='mediapipe.TfLiteConverterCalculatorOptions.TensorFloatRange.min', index=0,
number=1, type=2, cpp_type=6, label=1,
has_default_value=False, default_value=float(0),
message_type=None, enum_type=None, containing_type=None,
is_extension=False, extension_scope=None,
serialized_options=None, file=DESCRIPTOR),
_descriptor.FieldDescriptor(
name='max', full_name='mediapipe.TfLiteConverterCalculatorOptions.TensorFloatRange.max', index=1,
number=2, type=2, cpp_type=6, label=1,
has_default_value=False, default_value=float(0),
message_type=None, enum_type=None, containing_type=None,
is_extension=False, extension_scope=None,
serialized_options=None, file=DESCRIPTOR),
],
extensions=[
],
nested_types=[],
enum_types=[
],
serialized_options=None,
is_extendable=False,
syntax='proto2',
extension_ranges=[],
oneofs=[
],
serialized_start=497,
serialized_end=541,
)
_TFLITECONVERTERCALCULATOROPTIONS = _descriptor.Descriptor(
name='TfLiteConverterCalculatorOptions',
full_name='mediapipe.TfLiteConverterCalculatorOptions',
filename=None,
file=DESCRIPTOR,
containing_type=None,
fields=[
_descriptor.FieldDescriptor(
name='zero_center', full_name='mediapipe.TfLiteConverterCalculatorOptions.zero_center', index=0,
number=1, type=8, cpp_type=7, label=1,
has_default_value=True, default_value=True,
message_type=None, enum_type=None, containing_type=None,
is_extension=False, extension_scope=None,
serialized_options=None, file=DESCRIPTOR),
_descriptor.FieldDescriptor(
name='use_custom_normalization', full_name='mediapipe.TfLiteConverterCalculatorOptions.use_custom_normalization', index=1,
number=6, type=8, cpp_type=7, label=1,
has_default_value=True, default_value=False,
message_type=None, enum_type=None, containing_type=None,
is_extension=False, extension_scope=None,
serialized_options=None, file=DESCRIPTOR),
_descriptor.FieldDescriptor(
name='custom_div', full_name='mediapipe.TfLiteConverterCalculatorOptions.custom_div', index=2,
number=7, type=2, cpp_type=6, label=1,
has_default_value=True, default_value=float(-1),
message_type=None, enum_type=None, containing_type=None,
is_extension=False, extension_scope=None,
serialized_options=None, file=DESCRIPTOR),
_descriptor.FieldDescriptor(
name='custom_sub', full_name='mediapipe.TfLiteConverterCalculatorOptions.custom_sub', index=3,
number=8, type=2, cpp_type=6, label=1,
has_default_value=True, default_value=float(-1),
message_type=None, enum_type=None, containing_type=None,
is_extension=False, extension_scope=None,
serialized_options=None, file=DESCRIPTOR),
_descriptor.FieldDescriptor(
name='flip_vertically', full_name='mediapipe.TfLiteConverterCalculatorOptions.flip_vertically', index=4,
number=2, type=8, cpp_type=7, label=1,
has_default_value=True, default_value=False,
message_type=None, enum_type=None, containing_type=None,
is_extension=False, extension_scope=None,
serialized_options=None, file=DESCRIPTOR),
_descriptor.FieldDescriptor(
name='max_num_channels', full_name='mediapipe.TfLiteConverterCalculatorOptions.max_num_channels', index=5,
number=3, type=5, cpp_type=1, label=1,
has_default_value=True, default_value=3,
message_type=None, enum_type=None, containing_type=None,
is_extension=False, extension_scope=None,
serialized_options=None, file=DESCRIPTOR),
_descriptor.FieldDescriptor(
name='row_major_matrix', full_name='mediapipe.TfLiteConverterCalculatorOptions.row_major_matrix', index=6,
number=4, type=8, cpp_type=7, label=1,
has_default_value=True, default_value=False,
message_type=None, enum_type=None, containing_type=None,
is_extension=False, extension_scope=None,
serialized_options=None, file=DESCRIPTOR),
_descriptor.FieldDescriptor(
name='use_quantized_tensors', full_name='mediapipe.TfLiteConverterCalculatorOptions.use_quantized_tensors', index=7,
number=5, type=8, cpp_type=7, label=1,
has_default_value=True, default_value=False,
message_type=None, enum_type=None, containing_type=None,
is_extension=False, extension_scope=None,
serialized_options=None, file=DESCRIPTOR),
_descriptor.FieldDescriptor(
name='output_tensor_float_range', full_name='mediapipe.TfLiteConverterCalculatorOptions.output_tensor_float_range', index=8,
number=9, type=11, cpp_type=10, label=1,
has_default_value=False, default_value=None,
message_type=None, enum_type=None, containing_type=None,
is_extension=False, extension_scope=None,
serialized_options=None, file=DESCRIPTOR),
],
extensions=[
_descriptor.FieldDescriptor(
name='ext', full_name='mediapipe.TfLiteConverterCalculatorOptions.ext', index=0,
number=245817797, type=11, cpp_type=10, label=1,
has_default_value=False, default_value=None,
message_type=None, enum_type=None, containing_type=None,
is_extension=True, extension_scope=None,
serialized_options=None, file=DESCRIPTOR),
],
nested_types=[_TFLITECONVERTERCALCULATOROPTIONS_TENSORFLOATRANGE, ],
enum_types=[
],
serialized_options=None,
is_extendable=False,
syntax='proto2',
extension_ranges=[],
oneofs=[
],
serialized_start=116,
serialized_end=632,
)
_TFLITECONVERTERCALCULATOROPTIONS_TENSORFLOATRANGE.containing_type = _TFLITECONVERTERCALCULATOROPTIONS
_TFLITECONVERTERCALCULATOROPTIONS.fields_by_name['output_tensor_float_range'].message_type = _TFLITECONVERTERCALCULATOROPTIONS_TENSORFLOATRANGE
DESCRIPTOR.message_types_by_name['TfLiteConverterCalculatorOptions'] = _TFLITECONVERTERCALCULATOROPTIONS
_sym_db.RegisterFileDescriptor(DESCRIPTOR)
TfLiteConverterCalculatorOptions = _reflection.GeneratedProtocolMessageType('TfLiteConverterCalculatorOptions', (_message.Message,), dict(
TensorFloatRange = _reflection.GeneratedProtocolMessageType('TensorFloatRange', (_message.Message,), dict(
DESCRIPTOR = _TFLITECONVERTERCALCULATOROPTIONS_TENSORFLOATRANGE,
__module__ = 'mediapipe.calculators.tflite.tflite_converter_calculator_pb2'
# @@protoc_insertion_point(class_scope:mediapipe.TfLiteConverterCalculatorOptions.TensorFloatRange)
))
,
DESCRIPTOR = _TFLITECONVERTERCALCULATOROPTIONS,
__module__ = 'mediapipe.calculators.tflite.tflite_converter_calculator_pb2'
# @@protoc_insertion_point(class_scope:mediapipe.TfLiteConverterCalculatorOptions)
))
_sym_db.RegisterMessage(TfLiteConverterCalculatorOptions)
_sym_db.RegisterMessage(TfLiteConverterCalculatorOptions.TensorFloatRange)
_TFLITECONVERTERCALCULATOROPTIONS.extensions_by_name['ext'].message_type = _TFLITECONVERTERCALCULATOROPTIONS
mediapipe_dot_framework_dot_calculator__options__pb2.CalculatorOptions.RegisterExtension(_TFLITECONVERTERCALCULATOROPTIONS.extensions_by_name['ext'])
# @@protoc_insertion_point(module_scope)

View File

@@ -0,0 +1,85 @@
# -*- coding: utf-8 -*-
# Generated by the protocol buffer compiler. DO NOT EDIT!
# source: mediapipe/calculators/tflite/tflite_custom_op_resolver_calculator.proto
import sys
_b=sys.version_info[0]<3 and (lambda x:x) or (lambda x:x.encode('latin1'))
from google.protobuf import descriptor as _descriptor
from google.protobuf import message as _message
from google.protobuf import reflection as _reflection
from google.protobuf import symbol_database as _symbol_database
# @@protoc_insertion_point(imports)
_sym_db = _symbol_database.Default()
from 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.FileDescriptor(
name='mediapipe/calculators/tflite/tflite_custom_op_resolver_calculator.proto',
package='mediapipe',
syntax='proto2',
serialized_options=None,
serialized_pb=_b('\nGmediapipe/calculators/tflite/tflite_custom_op_resolver_calculator.proto\x12\tmediapipe\x1a$mediapipe/framework/calculator.proto\"\xa3\x01\n\'TfLiteCustomOpResolverCalculatorOptions\x12\x16\n\x07use_gpu\x18\x01 \x01(\x08:\x05\x66\x61lse2`\n\x03\x65xt\x12\x1c.mediapipe.CalculatorOptions\x18\x81\x9a\x9ax \x01(\x0b\x32\x32.mediapipe.TfLiteCustomOpResolverCalculatorOptions')
,
dependencies=[mediapipe_dot_framework_dot_calculator__pb2.DESCRIPTOR,])
_TFLITECUSTOMOPRESOLVERCALCULATOROPTIONS = _descriptor.Descriptor(
name='TfLiteCustomOpResolverCalculatorOptions',
full_name='mediapipe.TfLiteCustomOpResolverCalculatorOptions',
filename=None,
file=DESCRIPTOR,
containing_type=None,
fields=[
_descriptor.FieldDescriptor(
name='use_gpu', full_name='mediapipe.TfLiteCustomOpResolverCalculatorOptions.use_gpu', index=0,
number=1, type=8, cpp_type=7, label=1,
has_default_value=True, default_value=False,
message_type=None, enum_type=None, containing_type=None,
is_extension=False, extension_scope=None,
serialized_options=None, file=DESCRIPTOR),
],
extensions=[
_descriptor.FieldDescriptor(
name='ext', full_name='mediapipe.TfLiteCustomOpResolverCalculatorOptions.ext', index=0,
number=252087553, type=11, cpp_type=10, label=1,
has_default_value=False, default_value=None,
message_type=None, enum_type=None, containing_type=None,
is_extension=True, extension_scope=None,
serialized_options=None, file=DESCRIPTOR),
],
nested_types=[],
enum_types=[
],
serialized_options=None,
is_extendable=False,
syntax='proto2',
extension_ranges=[],
oneofs=[
],
serialized_start=125,
serialized_end=288,
)
DESCRIPTOR.message_types_by_name['TfLiteCustomOpResolverCalculatorOptions'] = _TFLITECUSTOMOPRESOLVERCALCULATOROPTIONS
_sym_db.RegisterFileDescriptor(DESCRIPTOR)
TfLiteCustomOpResolverCalculatorOptions = _reflection.GeneratedProtocolMessageType('TfLiteCustomOpResolverCalculatorOptions', (_message.Message,), dict(
DESCRIPTOR = _TFLITECUSTOMOPRESOLVERCALCULATOROPTIONS,
__module__ = 'mediapipe.calculators.tflite.tflite_custom_op_resolver_calculator_pb2'
# @@protoc_insertion_point(class_scope:mediapipe.TfLiteCustomOpResolverCalculatorOptions)
))
_sym_db.RegisterMessage(TfLiteCustomOpResolverCalculatorOptions)
_TFLITECUSTOMOPRESOLVERCALCULATOROPTIONS.extensions_by_name['ext'].message_type = _TFLITECUSTOMOPRESOLVERCALCULATOROPTIONS
mediapipe_dot_framework_dot_calculator__options__pb2.CalculatorOptions.RegisterExtension(_TFLITECUSTOMOPRESOLVERCALCULATOROPTIONS.extensions_by_name['ext'])
# @@protoc_insertion_point(module_scope)

View File

@@ -0,0 +1,437 @@
# -*- coding: utf-8 -*-
# Generated by the protocol buffer compiler. DO NOT EDIT!
# source: mediapipe/calculators/tflite/tflite_inference_calculator.proto
import sys
_b=sys.version_info[0]<3 and (lambda x:x) or (lambda x:x.encode('latin1'))
from google.protobuf import descriptor as _descriptor
from google.protobuf import message as _message
from google.protobuf import reflection as _reflection
from google.protobuf import symbol_database as _symbol_database
# @@protoc_insertion_point(imports)
_sym_db = _symbol_database.Default()
from 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.FileDescriptor(
name='mediapipe/calculators/tflite/tflite_inference_calculator.proto',
package='mediapipe',
syntax='proto2',
serialized_options=None,
serialized_pb=_b('\n>mediapipe/calculators/tflite/tflite_inference_calculator.proto\x12\tmediapipe\x1a$mediapipe/framework/calculator.proto\"\xfb\x08\n TfLiteInferenceCalculatorOptions\x12\x12\n\nmodel_path\x18\x01 \x01(\t\x12\x1a\n\x07use_gpu\x18\x02 \x01(\x08:\x05\x66\x61lseB\x02\x18\x01\x12\x1c\n\tuse_nnapi\x18\x03 \x01(\x08:\x05\x66\x61lseB\x02\x18\x01\x12\x1a\n\x0e\x63pu_num_thread\x18\x04 \x01(\x05:\x02-1\x12\x46\n\x08\x64\x65legate\x18\x05 \x01(\x0b\x32\x34.mediapipe.TfLiteInferenceCalculatorOptions.Delegate\x1a\xc9\x06\n\x08\x44\x65legate\x12M\n\x06tflite\x18\x01 \x01(\x0b\x32;.mediapipe.TfLiteInferenceCalculatorOptions.Delegate.TfLiteH\x00\x12G\n\x03gpu\x18\x02 \x01(\x0b\x32\x38.mediapipe.TfLiteInferenceCalculatorOptions.Delegate.GpuH\x00\x12K\n\x05nnapi\x18\x03 \x01(\x0b\x32:.mediapipe.TfLiteInferenceCalculatorOptions.Delegate.NnapiH\x00\x12O\n\x07xnnpack\x18\x04 \x01(\x0b\x32<.mediapipe.TfLiteInferenceCalculatorOptions.Delegate.XnnpackH\x00\x1a\x08\n\x06TfLite\x1a\x9b\x03\n\x03Gpu\x12#\n\x14use_advanced_gpu_api\x18\x01 \x01(\x08:\x05\x66\x61lse\x12N\n\x03\x61pi\x18\x04 \x01(\x0e\x32<.mediapipe.TfLiteInferenceCalculatorOptions.Delegate.Gpu.Api:\x03\x41NY\x12\"\n\x14\x61llow_precision_loss\x18\x03 \x01(\x08:\x04true\x12\x1a\n\x12\x63\x61\x63hed_kernel_path\x18\x02 \x01(\t\x12g\n\x05usage\x18\x05 \x01(\x0e\x32G.mediapipe.TfLiteInferenceCalculatorOptions.Delegate.Gpu.InferenceUsage:\x0fSUSTAINED_SPEED\"&\n\x03\x41pi\x12\x07\n\x03\x41NY\x10\x00\x12\n\n\x06OPENGL\x10\x01\x12\n\n\x06OPENCL\x10\x02\"N\n\x0eInferenceUsage\x12\x0f\n\x0bUNSPECIFIED\x10\x00\x12\x16\n\x12\x46\x41ST_SINGLE_ANSWER\x10\x01\x12\x13\n\x0fSUSTAINED_SPEED\x10\x02\x1a/\n\x05Nnapi\x12\x11\n\tcache_dir\x18\x01 \x01(\t\x12\x13\n\x0bmodel_token\x18\x02 \x01(\t\x1a\"\n\x07Xnnpack\x12\x17\n\x0bnum_threads\x18\x01 \x01(\x05:\x02-1B\n\n\x08\x64\x65legate2Y\n\x03\x65xt\x12\x1c.mediapipe.CalculatorOptions\x18\xcd\x8f\xc2o \x01(\x0b\x32+.mediapipe.TfLiteInferenceCalculatorOptions')
,
dependencies=[mediapipe_dot_framework_dot_calculator__pb2.DESCRIPTOR,])
_TFLITEINFERENCECALCULATOROPTIONS_DELEGATE_GPU_API = _descriptor.EnumDescriptor(
name='Api',
full_name='mediapipe.TfLiteInferenceCalculatorOptions.Delegate.Gpu.Api',
filename=None,
file=DESCRIPTOR,
values=[
_descriptor.EnumValueDescriptor(
name='ANY', index=0, number=0,
serialized_options=None,
type=None),
_descriptor.EnumValueDescriptor(
name='OPENGL', index=1, number=1,
serialized_options=None,
type=None),
_descriptor.EnumValueDescriptor(
name='OPENCL', index=2, number=2,
serialized_options=None,
type=None),
],
containing_type=None,
serialized_options=None,
serialized_start=957,
serialized_end=995,
)
_sym_db.RegisterEnumDescriptor(_TFLITEINFERENCECALCULATOROPTIONS_DELEGATE_GPU_API)
_TFLITEINFERENCECALCULATOROPTIONS_DELEGATE_GPU_INFERENCEUSAGE = _descriptor.EnumDescriptor(
name='InferenceUsage',
full_name='mediapipe.TfLiteInferenceCalculatorOptions.Delegate.Gpu.InferenceUsage',
filename=None,
file=DESCRIPTOR,
values=[
_descriptor.EnumValueDescriptor(
name='UNSPECIFIED', index=0, number=0,
serialized_options=None,
type=None),
_descriptor.EnumValueDescriptor(
name='FAST_SINGLE_ANSWER', index=1, number=1,
serialized_options=None,
type=None),
_descriptor.EnumValueDescriptor(
name='SUSTAINED_SPEED', index=2, number=2,
serialized_options=None,
type=None),
],
containing_type=None,
serialized_options=None,
serialized_start=997,
serialized_end=1075,
)
_sym_db.RegisterEnumDescriptor(_TFLITEINFERENCECALCULATOROPTIONS_DELEGATE_GPU_INFERENCEUSAGE)
_TFLITEINFERENCECALCULATOROPTIONS_DELEGATE_TFLITE = _descriptor.Descriptor(
name='TfLite',
full_name='mediapipe.TfLiteInferenceCalculatorOptions.Delegate.TfLite',
filename=None,
file=DESCRIPTOR,
containing_type=None,
fields=[
],
extensions=[
],
nested_types=[],
enum_types=[
],
serialized_options=None,
is_extendable=False,
syntax='proto2',
extension_ranges=[],
oneofs=[
],
serialized_start=653,
serialized_end=661,
)
_TFLITEINFERENCECALCULATOROPTIONS_DELEGATE_GPU = _descriptor.Descriptor(
name='Gpu',
full_name='mediapipe.TfLiteInferenceCalculatorOptions.Delegate.Gpu',
filename=None,
file=DESCRIPTOR,
containing_type=None,
fields=[
_descriptor.FieldDescriptor(
name='use_advanced_gpu_api', full_name='mediapipe.TfLiteInferenceCalculatorOptions.Delegate.Gpu.use_advanced_gpu_api', index=0,
number=1, type=8, cpp_type=7, label=1,
has_default_value=True, default_value=False,
message_type=None, enum_type=None, containing_type=None,
is_extension=False, extension_scope=None,
serialized_options=None, file=DESCRIPTOR),
_descriptor.FieldDescriptor(
name='api', full_name='mediapipe.TfLiteInferenceCalculatorOptions.Delegate.Gpu.api', index=1,
number=4, type=14, cpp_type=8, label=1,
has_default_value=True, default_value=0,
message_type=None, enum_type=None, containing_type=None,
is_extension=False, extension_scope=None,
serialized_options=None, file=DESCRIPTOR),
_descriptor.FieldDescriptor(
name='allow_precision_loss', full_name='mediapipe.TfLiteInferenceCalculatorOptions.Delegate.Gpu.allow_precision_loss', index=2,
number=3, type=8, cpp_type=7, label=1,
has_default_value=True, default_value=True,
message_type=None, enum_type=None, containing_type=None,
is_extension=False, extension_scope=None,
serialized_options=None, file=DESCRIPTOR),
_descriptor.FieldDescriptor(
name='cached_kernel_path', full_name='mediapipe.TfLiteInferenceCalculatorOptions.Delegate.Gpu.cached_kernel_path', index=3,
number=2, type=9, cpp_type=9, label=1,
has_default_value=False, default_value=_b("").decode('utf-8'),
message_type=None, enum_type=None, containing_type=None,
is_extension=False, extension_scope=None,
serialized_options=None, file=DESCRIPTOR),
_descriptor.FieldDescriptor(
name='usage', full_name='mediapipe.TfLiteInferenceCalculatorOptions.Delegate.Gpu.usage', index=4,
number=5, type=14, cpp_type=8, label=1,
has_default_value=True, default_value=2,
message_type=None, enum_type=None, containing_type=None,
is_extension=False, extension_scope=None,
serialized_options=None, file=DESCRIPTOR),
],
extensions=[
],
nested_types=[],
enum_types=[
_TFLITEINFERENCECALCULATOROPTIONS_DELEGATE_GPU_API,
_TFLITEINFERENCECALCULATOROPTIONS_DELEGATE_GPU_INFERENCEUSAGE,
],
serialized_options=None,
is_extendable=False,
syntax='proto2',
extension_ranges=[],
oneofs=[
],
serialized_start=664,
serialized_end=1075,
)
_TFLITEINFERENCECALCULATOROPTIONS_DELEGATE_NNAPI = _descriptor.Descriptor(
name='Nnapi',
full_name='mediapipe.TfLiteInferenceCalculatorOptions.Delegate.Nnapi',
filename=None,
file=DESCRIPTOR,
containing_type=None,
fields=[
_descriptor.FieldDescriptor(
name='cache_dir', full_name='mediapipe.TfLiteInferenceCalculatorOptions.Delegate.Nnapi.cache_dir', index=0,
number=1, type=9, cpp_type=9, label=1,
has_default_value=False, default_value=_b("").decode('utf-8'),
message_type=None, enum_type=None, containing_type=None,
is_extension=False, extension_scope=None,
serialized_options=None, file=DESCRIPTOR),
_descriptor.FieldDescriptor(
name='model_token', full_name='mediapipe.TfLiteInferenceCalculatorOptions.Delegate.Nnapi.model_token', index=1,
number=2, type=9, cpp_type=9, label=1,
has_default_value=False, default_value=_b("").decode('utf-8'),
message_type=None, enum_type=None, containing_type=None,
is_extension=False, extension_scope=None,
serialized_options=None, file=DESCRIPTOR),
],
extensions=[
],
nested_types=[],
enum_types=[
],
serialized_options=None,
is_extendable=False,
syntax='proto2',
extension_ranges=[],
oneofs=[
],
serialized_start=1077,
serialized_end=1124,
)
_TFLITEINFERENCECALCULATOROPTIONS_DELEGATE_XNNPACK = _descriptor.Descriptor(
name='Xnnpack',
full_name='mediapipe.TfLiteInferenceCalculatorOptions.Delegate.Xnnpack',
filename=None,
file=DESCRIPTOR,
containing_type=None,
fields=[
_descriptor.FieldDescriptor(
name='num_threads', full_name='mediapipe.TfLiteInferenceCalculatorOptions.Delegate.Xnnpack.num_threads', index=0,
number=1, type=5, cpp_type=1, label=1,
has_default_value=True, default_value=-1,
message_type=None, enum_type=None, containing_type=None,
is_extension=False, extension_scope=None,
serialized_options=None, file=DESCRIPTOR),
],
extensions=[
],
nested_types=[],
enum_types=[
],
serialized_options=None,
is_extendable=False,
syntax='proto2',
extension_ranges=[],
oneofs=[
],
serialized_start=1126,
serialized_end=1160,
)
_TFLITEINFERENCECALCULATOROPTIONS_DELEGATE = _descriptor.Descriptor(
name='Delegate',
full_name='mediapipe.TfLiteInferenceCalculatorOptions.Delegate',
filename=None,
file=DESCRIPTOR,
containing_type=None,
fields=[
_descriptor.FieldDescriptor(
name='tflite', full_name='mediapipe.TfLiteInferenceCalculatorOptions.Delegate.tflite', index=0,
number=1, type=11, cpp_type=10, label=1,
has_default_value=False, default_value=None,
message_type=None, enum_type=None, containing_type=None,
is_extension=False, extension_scope=None,
serialized_options=None, file=DESCRIPTOR),
_descriptor.FieldDescriptor(
name='gpu', full_name='mediapipe.TfLiteInferenceCalculatorOptions.Delegate.gpu', index=1,
number=2, type=11, cpp_type=10, label=1,
has_default_value=False, default_value=None,
message_type=None, enum_type=None, containing_type=None,
is_extension=False, extension_scope=None,
serialized_options=None, file=DESCRIPTOR),
_descriptor.FieldDescriptor(
name='nnapi', full_name='mediapipe.TfLiteInferenceCalculatorOptions.Delegate.nnapi', index=2,
number=3, type=11, cpp_type=10, label=1,
has_default_value=False, default_value=None,
message_type=None, enum_type=None, containing_type=None,
is_extension=False, extension_scope=None,
serialized_options=None, file=DESCRIPTOR),
_descriptor.FieldDescriptor(
name='xnnpack', full_name='mediapipe.TfLiteInferenceCalculatorOptions.Delegate.xnnpack', index=3,
number=4, type=11, cpp_type=10, label=1,
has_default_value=False, default_value=None,
message_type=None, enum_type=None, containing_type=None,
is_extension=False, extension_scope=None,
serialized_options=None, file=DESCRIPTOR),
],
extensions=[
],
nested_types=[_TFLITEINFERENCECALCULATOROPTIONS_DELEGATE_TFLITE, _TFLITEINFERENCECALCULATOROPTIONS_DELEGATE_GPU, _TFLITEINFERENCECALCULATOROPTIONS_DELEGATE_NNAPI, _TFLITEINFERENCECALCULATOROPTIONS_DELEGATE_XNNPACK, ],
enum_types=[
],
serialized_options=None,
is_extendable=False,
syntax='proto2',
extension_ranges=[],
oneofs=[
_descriptor.OneofDescriptor(
name='delegate', full_name='mediapipe.TfLiteInferenceCalculatorOptions.Delegate.delegate',
index=0, containing_type=None, fields=[]),
],
serialized_start=331,
serialized_end=1172,
)
_TFLITEINFERENCECALCULATOROPTIONS = _descriptor.Descriptor(
name='TfLiteInferenceCalculatorOptions',
full_name='mediapipe.TfLiteInferenceCalculatorOptions',
filename=None,
file=DESCRIPTOR,
containing_type=None,
fields=[
_descriptor.FieldDescriptor(
name='model_path', full_name='mediapipe.TfLiteInferenceCalculatorOptions.model_path', index=0,
number=1, type=9, cpp_type=9, label=1,
has_default_value=False, default_value=_b("").decode('utf-8'),
message_type=None, enum_type=None, containing_type=None,
is_extension=False, extension_scope=None,
serialized_options=None, file=DESCRIPTOR),
_descriptor.FieldDescriptor(
name='use_gpu', full_name='mediapipe.TfLiteInferenceCalculatorOptions.use_gpu', index=1,
number=2, type=8, cpp_type=7, label=1,
has_default_value=True, default_value=False,
message_type=None, enum_type=None, containing_type=None,
is_extension=False, extension_scope=None,
serialized_options=_b('\030\001'), file=DESCRIPTOR),
_descriptor.FieldDescriptor(
name='use_nnapi', full_name='mediapipe.TfLiteInferenceCalculatorOptions.use_nnapi', index=2,
number=3, type=8, cpp_type=7, label=1,
has_default_value=True, default_value=False,
message_type=None, enum_type=None, containing_type=None,
is_extension=False, extension_scope=None,
serialized_options=_b('\030\001'), file=DESCRIPTOR),
_descriptor.FieldDescriptor(
name='cpu_num_thread', full_name='mediapipe.TfLiteInferenceCalculatorOptions.cpu_num_thread', index=3,
number=4, type=5, cpp_type=1, label=1,
has_default_value=True, default_value=-1,
message_type=None, enum_type=None, containing_type=None,
is_extension=False, extension_scope=None,
serialized_options=None, file=DESCRIPTOR),
_descriptor.FieldDescriptor(
name='delegate', full_name='mediapipe.TfLiteInferenceCalculatorOptions.delegate', index=4,
number=5, type=11, cpp_type=10, label=1,
has_default_value=False, default_value=None,
message_type=None, enum_type=None, containing_type=None,
is_extension=False, extension_scope=None,
serialized_options=None, file=DESCRIPTOR),
],
extensions=[
_descriptor.FieldDescriptor(
name='ext', full_name='mediapipe.TfLiteInferenceCalculatorOptions.ext', index=0,
number=233867213, type=11, cpp_type=10, label=1,
has_default_value=False, default_value=None,
message_type=None, enum_type=None, containing_type=None,
is_extension=True, extension_scope=None,
serialized_options=None, file=DESCRIPTOR),
],
nested_types=[_TFLITEINFERENCECALCULATOROPTIONS_DELEGATE, ],
enum_types=[
],
serialized_options=None,
is_extendable=False,
syntax='proto2',
extension_ranges=[],
oneofs=[
],
serialized_start=116,
serialized_end=1263,
)
_TFLITEINFERENCECALCULATOROPTIONS_DELEGATE_TFLITE.containing_type = _TFLITEINFERENCECALCULATOROPTIONS_DELEGATE
_TFLITEINFERENCECALCULATOROPTIONS_DELEGATE_GPU.fields_by_name['api'].enum_type = _TFLITEINFERENCECALCULATOROPTIONS_DELEGATE_GPU_API
_TFLITEINFERENCECALCULATOROPTIONS_DELEGATE_GPU.fields_by_name['usage'].enum_type = _TFLITEINFERENCECALCULATOROPTIONS_DELEGATE_GPU_INFERENCEUSAGE
_TFLITEINFERENCECALCULATOROPTIONS_DELEGATE_GPU.containing_type = _TFLITEINFERENCECALCULATOROPTIONS_DELEGATE
_TFLITEINFERENCECALCULATOROPTIONS_DELEGATE_GPU_API.containing_type = _TFLITEINFERENCECALCULATOROPTIONS_DELEGATE_GPU
_TFLITEINFERENCECALCULATOROPTIONS_DELEGATE_GPU_INFERENCEUSAGE.containing_type = _TFLITEINFERENCECALCULATOROPTIONS_DELEGATE_GPU
_TFLITEINFERENCECALCULATOROPTIONS_DELEGATE_NNAPI.containing_type = _TFLITEINFERENCECALCULATOROPTIONS_DELEGATE
_TFLITEINFERENCECALCULATOROPTIONS_DELEGATE_XNNPACK.containing_type = _TFLITEINFERENCECALCULATOROPTIONS_DELEGATE
_TFLITEINFERENCECALCULATOROPTIONS_DELEGATE.fields_by_name['tflite'].message_type = _TFLITEINFERENCECALCULATOROPTIONS_DELEGATE_TFLITE
_TFLITEINFERENCECALCULATOROPTIONS_DELEGATE.fields_by_name['gpu'].message_type = _TFLITEINFERENCECALCULATOROPTIONS_DELEGATE_GPU
_TFLITEINFERENCECALCULATOROPTIONS_DELEGATE.fields_by_name['nnapi'].message_type = _TFLITEINFERENCECALCULATOROPTIONS_DELEGATE_NNAPI
_TFLITEINFERENCECALCULATOROPTIONS_DELEGATE.fields_by_name['xnnpack'].message_type = _TFLITEINFERENCECALCULATOROPTIONS_DELEGATE_XNNPACK
_TFLITEINFERENCECALCULATOROPTIONS_DELEGATE.containing_type = _TFLITEINFERENCECALCULATOROPTIONS
_TFLITEINFERENCECALCULATOROPTIONS_DELEGATE.oneofs_by_name['delegate'].fields.append(
_TFLITEINFERENCECALCULATOROPTIONS_DELEGATE.fields_by_name['tflite'])
_TFLITEINFERENCECALCULATOROPTIONS_DELEGATE.fields_by_name['tflite'].containing_oneof = _TFLITEINFERENCECALCULATOROPTIONS_DELEGATE.oneofs_by_name['delegate']
_TFLITEINFERENCECALCULATOROPTIONS_DELEGATE.oneofs_by_name['delegate'].fields.append(
_TFLITEINFERENCECALCULATOROPTIONS_DELEGATE.fields_by_name['gpu'])
_TFLITEINFERENCECALCULATOROPTIONS_DELEGATE.fields_by_name['gpu'].containing_oneof = _TFLITEINFERENCECALCULATOROPTIONS_DELEGATE.oneofs_by_name['delegate']
_TFLITEINFERENCECALCULATOROPTIONS_DELEGATE.oneofs_by_name['delegate'].fields.append(
_TFLITEINFERENCECALCULATOROPTIONS_DELEGATE.fields_by_name['nnapi'])
_TFLITEINFERENCECALCULATOROPTIONS_DELEGATE.fields_by_name['nnapi'].containing_oneof = _TFLITEINFERENCECALCULATOROPTIONS_DELEGATE.oneofs_by_name['delegate']
_TFLITEINFERENCECALCULATOROPTIONS_DELEGATE.oneofs_by_name['delegate'].fields.append(
_TFLITEINFERENCECALCULATOROPTIONS_DELEGATE.fields_by_name['xnnpack'])
_TFLITEINFERENCECALCULATOROPTIONS_DELEGATE.fields_by_name['xnnpack'].containing_oneof = _TFLITEINFERENCECALCULATOROPTIONS_DELEGATE.oneofs_by_name['delegate']
_TFLITEINFERENCECALCULATOROPTIONS.fields_by_name['delegate'].message_type = _TFLITEINFERENCECALCULATOROPTIONS_DELEGATE
DESCRIPTOR.message_types_by_name['TfLiteInferenceCalculatorOptions'] = _TFLITEINFERENCECALCULATOROPTIONS
_sym_db.RegisterFileDescriptor(DESCRIPTOR)
TfLiteInferenceCalculatorOptions = _reflection.GeneratedProtocolMessageType('TfLiteInferenceCalculatorOptions', (_message.Message,), dict(
Delegate = _reflection.GeneratedProtocolMessageType('Delegate', (_message.Message,), dict(
TfLite = _reflection.GeneratedProtocolMessageType('TfLite', (_message.Message,), dict(
DESCRIPTOR = _TFLITEINFERENCECALCULATOROPTIONS_DELEGATE_TFLITE,
__module__ = 'mediapipe.calculators.tflite.tflite_inference_calculator_pb2'
# @@protoc_insertion_point(class_scope:mediapipe.TfLiteInferenceCalculatorOptions.Delegate.TfLite)
))
,
Gpu = _reflection.GeneratedProtocolMessageType('Gpu', (_message.Message,), dict(
DESCRIPTOR = _TFLITEINFERENCECALCULATOROPTIONS_DELEGATE_GPU,
__module__ = 'mediapipe.calculators.tflite.tflite_inference_calculator_pb2'
# @@protoc_insertion_point(class_scope:mediapipe.TfLiteInferenceCalculatorOptions.Delegate.Gpu)
))
,
Nnapi = _reflection.GeneratedProtocolMessageType('Nnapi', (_message.Message,), dict(
DESCRIPTOR = _TFLITEINFERENCECALCULATOROPTIONS_DELEGATE_NNAPI,
__module__ = 'mediapipe.calculators.tflite.tflite_inference_calculator_pb2'
# @@protoc_insertion_point(class_scope:mediapipe.TfLiteInferenceCalculatorOptions.Delegate.Nnapi)
))
,
Xnnpack = _reflection.GeneratedProtocolMessageType('Xnnpack', (_message.Message,), dict(
DESCRIPTOR = _TFLITEINFERENCECALCULATOROPTIONS_DELEGATE_XNNPACK,
__module__ = 'mediapipe.calculators.tflite.tflite_inference_calculator_pb2'
# @@protoc_insertion_point(class_scope:mediapipe.TfLiteInferenceCalculatorOptions.Delegate.Xnnpack)
))
,
DESCRIPTOR = _TFLITEINFERENCECALCULATOROPTIONS_DELEGATE,
__module__ = 'mediapipe.calculators.tflite.tflite_inference_calculator_pb2'
# @@protoc_insertion_point(class_scope:mediapipe.TfLiteInferenceCalculatorOptions.Delegate)
))
,
DESCRIPTOR = _TFLITEINFERENCECALCULATOROPTIONS,
__module__ = 'mediapipe.calculators.tflite.tflite_inference_calculator_pb2'
# @@protoc_insertion_point(class_scope:mediapipe.TfLiteInferenceCalculatorOptions)
))
_sym_db.RegisterMessage(TfLiteInferenceCalculatorOptions)
_sym_db.RegisterMessage(TfLiteInferenceCalculatorOptions.Delegate)
_sym_db.RegisterMessage(TfLiteInferenceCalculatorOptions.Delegate.TfLite)
_sym_db.RegisterMessage(TfLiteInferenceCalculatorOptions.Delegate.Gpu)
_sym_db.RegisterMessage(TfLiteInferenceCalculatorOptions.Delegate.Nnapi)
_sym_db.RegisterMessage(TfLiteInferenceCalculatorOptions.Delegate.Xnnpack)
_TFLITEINFERENCECALCULATOROPTIONS.extensions_by_name['ext'].message_type = _TFLITEINFERENCECALCULATOROPTIONS
mediapipe_dot_framework_dot_calculator__options__pb2.CalculatorOptions.RegisterExtension(_TFLITEINFERENCECALCULATOROPTIONS.extensions_by_name['ext'])
_TFLITEINFERENCECALCULATOROPTIONS.fields_by_name['use_gpu']._options = None
_TFLITEINFERENCECALCULATOROPTIONS.fields_by_name['use_nnapi']._options = None
# @@protoc_insertion_point(module_scope)

View File

@@ -0,0 +1,106 @@
# -*- coding: utf-8 -*-
# Generated by the protocol buffer compiler. DO NOT EDIT!
# source: mediapipe/calculators/tflite/tflite_tensors_to_classification_calculator.proto
import sys
_b=sys.version_info[0]<3 and (lambda x:x) or (lambda x:x.encode('latin1'))
from google.protobuf import descriptor as _descriptor
from google.protobuf import message as _message
from google.protobuf import reflection as _reflection
from google.protobuf import symbol_database as _symbol_database
# @@protoc_insertion_point(imports)
_sym_db = _symbol_database.Default()
from 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.FileDescriptor(
name='mediapipe/calculators/tflite/tflite_tensors_to_classification_calculator.proto',
package='mediapipe',
syntax='proto2',
serialized_options=None,
serialized_pb=_b('\nNmediapipe/calculators/tflite/tflite_tensors_to_classification_calculator.proto\x12\tmediapipe\x1a$mediapipe/framework/calculator.proto\"\xfc\x01\n.TfLiteTensorsToClassificationCalculatorOptions\x12\x1b\n\x13min_score_threshold\x18\x01 \x01(\x02\x12\r\n\x05top_k\x18\x02 \x01(\x05\x12\x16\n\x0elabel_map_path\x18\x03 \x01(\t\x12\x1d\n\x15\x62inary_classification\x18\x04 \x01(\x08\x32g\n\x03\x65xt\x12\x1c.mediapipe.CalculatorOptions\x18\xe7\xdd\x83\x7f \x01(\x0b\x32\x39.mediapipe.TfLiteTensorsToClassificationCalculatorOptions')
,
dependencies=[mediapipe_dot_framework_dot_calculator__pb2.DESCRIPTOR,])
_TFLITETENSORSTOCLASSIFICATIONCALCULATOROPTIONS = _descriptor.Descriptor(
name='TfLiteTensorsToClassificationCalculatorOptions',
full_name='mediapipe.TfLiteTensorsToClassificationCalculatorOptions',
filename=None,
file=DESCRIPTOR,
containing_type=None,
fields=[
_descriptor.FieldDescriptor(
name='min_score_threshold', full_name='mediapipe.TfLiteTensorsToClassificationCalculatorOptions.min_score_threshold', index=0,
number=1, type=2, cpp_type=6, label=1,
has_default_value=False, default_value=float(0),
message_type=None, enum_type=None, containing_type=None,
is_extension=False, extension_scope=None,
serialized_options=None, file=DESCRIPTOR),
_descriptor.FieldDescriptor(
name='top_k', full_name='mediapipe.TfLiteTensorsToClassificationCalculatorOptions.top_k', index=1,
number=2, type=5, cpp_type=1, label=1,
has_default_value=False, default_value=0,
message_type=None, enum_type=None, containing_type=None,
is_extension=False, extension_scope=None,
serialized_options=None, file=DESCRIPTOR),
_descriptor.FieldDescriptor(
name='label_map_path', full_name='mediapipe.TfLiteTensorsToClassificationCalculatorOptions.label_map_path', index=2,
number=3, type=9, cpp_type=9, label=1,
has_default_value=False, default_value=_b("").decode('utf-8'),
message_type=None, enum_type=None, containing_type=None,
is_extension=False, extension_scope=None,
serialized_options=None, file=DESCRIPTOR),
_descriptor.FieldDescriptor(
name='binary_classification', full_name='mediapipe.TfLiteTensorsToClassificationCalculatorOptions.binary_classification', index=3,
number=4, type=8, cpp_type=7, label=1,
has_default_value=False, default_value=False,
message_type=None, enum_type=None, containing_type=None,
is_extension=False, extension_scope=None,
serialized_options=None, file=DESCRIPTOR),
],
extensions=[
_descriptor.FieldDescriptor(
name='ext', full_name='mediapipe.TfLiteTensorsToClassificationCalculatorOptions.ext', index=0,
number=266399463, type=11, cpp_type=10, label=1,
has_default_value=False, default_value=None,
message_type=None, enum_type=None, containing_type=None,
is_extension=True, extension_scope=None,
serialized_options=None, file=DESCRIPTOR),
],
nested_types=[],
enum_types=[
],
serialized_options=None,
is_extendable=False,
syntax='proto2',
extension_ranges=[],
oneofs=[
],
serialized_start=132,
serialized_end=384,
)
DESCRIPTOR.message_types_by_name['TfLiteTensorsToClassificationCalculatorOptions'] = _TFLITETENSORSTOCLASSIFICATIONCALCULATOROPTIONS
_sym_db.RegisterFileDescriptor(DESCRIPTOR)
TfLiteTensorsToClassificationCalculatorOptions = _reflection.GeneratedProtocolMessageType('TfLiteTensorsToClassificationCalculatorOptions', (_message.Message,), dict(
DESCRIPTOR = _TFLITETENSORSTOCLASSIFICATIONCALCULATOROPTIONS,
__module__ = 'mediapipe.calculators.tflite.tflite_tensors_to_classification_calculator_pb2'
# @@protoc_insertion_point(class_scope:mediapipe.TfLiteTensorsToClassificationCalculatorOptions)
))
_sym_db.RegisterMessage(TfLiteTensorsToClassificationCalculatorOptions)
_TFLITETENSORSTOCLASSIFICATIONCALCULATOROPTIONS.extensions_by_name['ext'].message_type = _TFLITETENSORSTOCLASSIFICATIONCALCULATOROPTIONS
mediapipe_dot_framework_dot_calculator__options__pb2.CalculatorOptions.RegisterExtension(_TFLITETENSORSTOCLASSIFICATIONCALCULATOROPTIONS.extensions_by_name['ext'])
# @@protoc_insertion_point(module_scope)

View File

@@ -0,0 +1,204 @@
# -*- coding: utf-8 -*-
# Generated by the protocol buffer compiler. DO NOT EDIT!
# source: mediapipe/calculators/tflite/tflite_tensors_to_detections_calculator.proto
import sys
_b=sys.version_info[0]<3 and (lambda x:x) or (lambda x:x.encode('latin1'))
from google.protobuf import descriptor as _descriptor
from google.protobuf import message as _message
from google.protobuf import reflection as _reflection
from google.protobuf import symbol_database as _symbol_database
# @@protoc_insertion_point(imports)
_sym_db = _symbol_database.Default()
from 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.FileDescriptor(
name='mediapipe/calculators/tflite/tflite_tensors_to_detections_calculator.proto',
package='mediapipe',
syntax='proto2',
serialized_options=None,
serialized_pb=_b('\nJmediapipe/calculators/tflite/tflite_tensors_to_detections_calculator.proto\x12\tmediapipe\x1a$mediapipe/framework/calculator.proto\"\xf9\x04\n*TfLiteTensorsToDetectionsCalculatorOptions\x12\x13\n\x0bnum_classes\x18\x01 \x02(\x05\x12\x11\n\tnum_boxes\x18\x02 \x02(\x05\x12\x12\n\nnum_coords\x18\x03 \x02(\x05\x12\x1d\n\x15keypoint_coord_offset\x18\t \x01(\x05\x12\x18\n\rnum_keypoints\x18\n \x01(\x05:\x01\x30\x12\"\n\x17num_values_per_keypoint\x18\x0b \x01(\x05:\x01\x32\x12\x1b\n\x10\x62ox_coord_offset\x18\x0c \x01(\x05:\x01\x30\x12\x12\n\x07x_scale\x18\x04 \x01(\x02:\x01\x30\x12\x12\n\x07y_scale\x18\x05 \x01(\x02:\x01\x30\x12\x12\n\x07w_scale\x18\x06 \x01(\x02:\x01\x30\x12\x12\n\x07h_scale\x18\x07 \x01(\x02:\x01\x30\x12,\n\x1d\x61pply_exponential_on_box_size\x18\r \x01(\x08:\x05\x66\x61lse\x12#\n\x14reverse_output_order\x18\x0e \x01(\x08:\x05\x66\x61lse\x12\x16\n\x0eignore_classes\x18\x08 \x03(\x05\x12\x1c\n\rsigmoid_score\x18\x0f \x01(\x08:\x05\x66\x61lse\x12\x1d\n\x15score_clipping_thresh\x18\x10 \x01(\x02\x12\x1e\n\x0f\x66lip_vertically\x18\x12 \x01(\x08:\x05\x66\x61lse\x12\x18\n\x10min_score_thresh\x18\x13 \x01(\x02\x32\x63\n\x03\x65xt\x12\x1c.mediapipe.CalculatorOptions\x18\x98\x8a\xc6u \x01(\x0b\x32\x35.mediapipe.TfLiteTensorsToDetectionsCalculatorOptions')
,
dependencies=[mediapipe_dot_framework_dot_calculator__pb2.DESCRIPTOR,])
_TFLITETENSORSTODETECTIONSCALCULATOROPTIONS = _descriptor.Descriptor(
name='TfLiteTensorsToDetectionsCalculatorOptions',
full_name='mediapipe.TfLiteTensorsToDetectionsCalculatorOptions',
filename=None,
file=DESCRIPTOR,
containing_type=None,
fields=[
_descriptor.FieldDescriptor(
name='num_classes', full_name='mediapipe.TfLiteTensorsToDetectionsCalculatorOptions.num_classes', index=0,
number=1, type=5, cpp_type=1, label=2,
has_default_value=False, default_value=0,
message_type=None, enum_type=None, containing_type=None,
is_extension=False, extension_scope=None,
serialized_options=None, file=DESCRIPTOR),
_descriptor.FieldDescriptor(
name='num_boxes', full_name='mediapipe.TfLiteTensorsToDetectionsCalculatorOptions.num_boxes', index=1,
number=2, type=5, cpp_type=1, label=2,
has_default_value=False, default_value=0,
message_type=None, enum_type=None, containing_type=None,
is_extension=False, extension_scope=None,
serialized_options=None, file=DESCRIPTOR),
_descriptor.FieldDescriptor(
name='num_coords', full_name='mediapipe.TfLiteTensorsToDetectionsCalculatorOptions.num_coords', index=2,
number=3, type=5, cpp_type=1, label=2,
has_default_value=False, default_value=0,
message_type=None, enum_type=None, containing_type=None,
is_extension=False, extension_scope=None,
serialized_options=None, file=DESCRIPTOR),
_descriptor.FieldDescriptor(
name='keypoint_coord_offset', full_name='mediapipe.TfLiteTensorsToDetectionsCalculatorOptions.keypoint_coord_offset', index=3,
number=9, type=5, cpp_type=1, label=1,
has_default_value=False, default_value=0,
message_type=None, enum_type=None, containing_type=None,
is_extension=False, extension_scope=None,
serialized_options=None, file=DESCRIPTOR),
_descriptor.FieldDescriptor(
name='num_keypoints', full_name='mediapipe.TfLiteTensorsToDetectionsCalculatorOptions.num_keypoints', index=4,
number=10, type=5, cpp_type=1, label=1,
has_default_value=True, default_value=0,
message_type=None, enum_type=None, containing_type=None,
is_extension=False, extension_scope=None,
serialized_options=None, file=DESCRIPTOR),
_descriptor.FieldDescriptor(
name='num_values_per_keypoint', full_name='mediapipe.TfLiteTensorsToDetectionsCalculatorOptions.num_values_per_keypoint', index=5,
number=11, type=5, cpp_type=1, label=1,
has_default_value=True, default_value=2,
message_type=None, enum_type=None, containing_type=None,
is_extension=False, extension_scope=None,
serialized_options=None, file=DESCRIPTOR),
_descriptor.FieldDescriptor(
name='box_coord_offset', full_name='mediapipe.TfLiteTensorsToDetectionsCalculatorOptions.box_coord_offset', index=6,
number=12, type=5, cpp_type=1, label=1,
has_default_value=True, default_value=0,
message_type=None, enum_type=None, containing_type=None,
is_extension=False, extension_scope=None,
serialized_options=None, file=DESCRIPTOR),
_descriptor.FieldDescriptor(
name='x_scale', full_name='mediapipe.TfLiteTensorsToDetectionsCalculatorOptions.x_scale', index=7,
number=4, type=2, cpp_type=6, label=1,
has_default_value=True, default_value=float(0),
message_type=None, enum_type=None, containing_type=None,
is_extension=False, extension_scope=None,
serialized_options=None, file=DESCRIPTOR),
_descriptor.FieldDescriptor(
name='y_scale', full_name='mediapipe.TfLiteTensorsToDetectionsCalculatorOptions.y_scale', index=8,
number=5, type=2, cpp_type=6, label=1,
has_default_value=True, default_value=float(0),
message_type=None, enum_type=None, containing_type=None,
is_extension=False, extension_scope=None,
serialized_options=None, file=DESCRIPTOR),
_descriptor.FieldDescriptor(
name='w_scale', full_name='mediapipe.TfLiteTensorsToDetectionsCalculatorOptions.w_scale', index=9,
number=6, type=2, cpp_type=6, label=1,
has_default_value=True, default_value=float(0),
message_type=None, enum_type=None, containing_type=None,
is_extension=False, extension_scope=None,
serialized_options=None, file=DESCRIPTOR),
_descriptor.FieldDescriptor(
name='h_scale', full_name='mediapipe.TfLiteTensorsToDetectionsCalculatorOptions.h_scale', index=10,
number=7, type=2, cpp_type=6, label=1,
has_default_value=True, default_value=float(0),
message_type=None, enum_type=None, containing_type=None,
is_extension=False, extension_scope=None,
serialized_options=None, file=DESCRIPTOR),
_descriptor.FieldDescriptor(
name='apply_exponential_on_box_size', full_name='mediapipe.TfLiteTensorsToDetectionsCalculatorOptions.apply_exponential_on_box_size', index=11,
number=13, type=8, cpp_type=7, label=1,
has_default_value=True, default_value=False,
message_type=None, enum_type=None, containing_type=None,
is_extension=False, extension_scope=None,
serialized_options=None, file=DESCRIPTOR),
_descriptor.FieldDescriptor(
name='reverse_output_order', full_name='mediapipe.TfLiteTensorsToDetectionsCalculatorOptions.reverse_output_order', index=12,
number=14, type=8, cpp_type=7, label=1,
has_default_value=True, default_value=False,
message_type=None, enum_type=None, containing_type=None,
is_extension=False, extension_scope=None,
serialized_options=None, file=DESCRIPTOR),
_descriptor.FieldDescriptor(
name='ignore_classes', full_name='mediapipe.TfLiteTensorsToDetectionsCalculatorOptions.ignore_classes', index=13,
number=8, type=5, cpp_type=1, label=3,
has_default_value=False, default_value=[],
message_type=None, enum_type=None, containing_type=None,
is_extension=False, extension_scope=None,
serialized_options=None, file=DESCRIPTOR),
_descriptor.FieldDescriptor(
name='sigmoid_score', full_name='mediapipe.TfLiteTensorsToDetectionsCalculatorOptions.sigmoid_score', index=14,
number=15, type=8, cpp_type=7, label=1,
has_default_value=True, default_value=False,
message_type=None, enum_type=None, containing_type=None,
is_extension=False, extension_scope=None,
serialized_options=None, file=DESCRIPTOR),
_descriptor.FieldDescriptor(
name='score_clipping_thresh', full_name='mediapipe.TfLiteTensorsToDetectionsCalculatorOptions.score_clipping_thresh', index=15,
number=16, type=2, cpp_type=6, label=1,
has_default_value=False, default_value=float(0),
message_type=None, enum_type=None, containing_type=None,
is_extension=False, extension_scope=None,
serialized_options=None, file=DESCRIPTOR),
_descriptor.FieldDescriptor(
name='flip_vertically', full_name='mediapipe.TfLiteTensorsToDetectionsCalculatorOptions.flip_vertically', index=16,
number=18, type=8, cpp_type=7, label=1,
has_default_value=True, default_value=False,
message_type=None, enum_type=None, containing_type=None,
is_extension=False, extension_scope=None,
serialized_options=None, file=DESCRIPTOR),
_descriptor.FieldDescriptor(
name='min_score_thresh', full_name='mediapipe.TfLiteTensorsToDetectionsCalculatorOptions.min_score_thresh', index=17,
number=19, type=2, cpp_type=6, label=1,
has_default_value=False, default_value=float(0),
message_type=None, enum_type=None, containing_type=None,
is_extension=False, extension_scope=None,
serialized_options=None, file=DESCRIPTOR),
],
extensions=[
_descriptor.FieldDescriptor(
name='ext', full_name='mediapipe.TfLiteTensorsToDetectionsCalculatorOptions.ext', index=0,
number=246514968, type=11, cpp_type=10, label=1,
has_default_value=False, default_value=None,
message_type=None, enum_type=None, containing_type=None,
is_extension=True, extension_scope=None,
serialized_options=None, file=DESCRIPTOR),
],
nested_types=[],
enum_types=[
],
serialized_options=None,
is_extendable=False,
syntax='proto2',
extension_ranges=[],
oneofs=[
],
serialized_start=128,
serialized_end=761,
)
DESCRIPTOR.message_types_by_name['TfLiteTensorsToDetectionsCalculatorOptions'] = _TFLITETENSORSTODETECTIONSCALCULATOROPTIONS
_sym_db.RegisterFileDescriptor(DESCRIPTOR)
TfLiteTensorsToDetectionsCalculatorOptions = _reflection.GeneratedProtocolMessageType('TfLiteTensorsToDetectionsCalculatorOptions', (_message.Message,), dict(
DESCRIPTOR = _TFLITETENSORSTODETECTIONSCALCULATOROPTIONS,
__module__ = 'mediapipe.calculators.tflite.tflite_tensors_to_detections_calculator_pb2'
# @@protoc_insertion_point(class_scope:mediapipe.TfLiteTensorsToDetectionsCalculatorOptions)
))
_sym_db.RegisterMessage(TfLiteTensorsToDetectionsCalculatorOptions)
_TFLITETENSORSTODETECTIONSCALCULATOROPTIONS.extensions_by_name['ext'].message_type = _TFLITETENSORSTODETECTIONSCALCULATOROPTIONS
mediapipe_dot_framework_dot_calculator__options__pb2.CalculatorOptions.RegisterExtension(_TFLITETENSORSTODETECTIONSCALCULATOROPTIONS.extensions_by_name['ext'])
# @@protoc_insertion_point(module_scope)

View File

@@ -0,0 +1,160 @@
# -*- coding: utf-8 -*-
# Generated by the protocol buffer compiler. DO NOT EDIT!
# source: mediapipe/calculators/tflite/tflite_tensors_to_landmarks_calculator.proto
import sys
_b=sys.version_info[0]<3 and (lambda x:x) or (lambda x:x.encode('latin1'))
from google.protobuf import descriptor as _descriptor
from google.protobuf import message as _message
from google.protobuf import reflection as _reflection
from google.protobuf import symbol_database as _symbol_database
# @@protoc_insertion_point(imports)
_sym_db = _symbol_database.Default()
from 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.FileDescriptor(
name='mediapipe/calculators/tflite/tflite_tensors_to_landmarks_calculator.proto',
package='mediapipe',
syntax='proto2',
serialized_options=None,
serialized_pb=_b('\nImediapipe/calculators/tflite/tflite_tensors_to_landmarks_calculator.proto\x12\tmediapipe\x1a$mediapipe/framework/calculator.proto\"\xa6\x04\n)TfLiteTensorsToLandmarksCalculatorOptions\x12\x15\n\rnum_landmarks\x18\x01 \x02(\x05\x12\x19\n\x11input_image_width\x18\x02 \x01(\x05\x12\x1a\n\x12input_image_height\x18\x03 \x01(\x05\x12\x1e\n\x0f\x66lip_vertically\x18\x04 \x01(\x08:\x05\x66\x61lse\x12 \n\x11\x66lip_horizontally\x18\x06 \x01(\x08:\x05\x66\x61lse\x12\x16\n\x0bnormalize_z\x18\x05 \x01(\x02:\x01\x31\x12\x64\n\x15visibility_activation\x18\x07 \x01(\x0e\x32?.mediapipe.TfLiteTensorsToLandmarksCalculatorOptions.Activation:\x04NONE\x12\x62\n\x13presence_activation\x18\x08 \x01(\x0e\x32?.mediapipe.TfLiteTensorsToLandmarksCalculatorOptions.Activation:\x04NONE\"#\n\nActivation\x12\x08\n\x04NONE\x10\x00\x12\x0b\n\x07SIGMOID\x10\x01\x32\x62\n\x03\x65xt\x12\x1c.mediapipe.CalculatorOptions\x18\xca\xe0\xdez \x01(\x0b\x32\x34.mediapipe.TfLiteTensorsToLandmarksCalculatorOptions')
,
dependencies=[mediapipe_dot_framework_dot_calculator__pb2.DESCRIPTOR,])
_TFLITETENSORSTOLANDMARKSCALCULATOROPTIONS_ACTIVATION = _descriptor.EnumDescriptor(
name='Activation',
full_name='mediapipe.TfLiteTensorsToLandmarksCalculatorOptions.Activation',
filename=None,
file=DESCRIPTOR,
values=[
_descriptor.EnumValueDescriptor(
name='NONE', index=0, number=0,
serialized_options=None,
type=None),
_descriptor.EnumValueDescriptor(
name='SIGMOID', index=1, number=1,
serialized_options=None,
type=None),
],
containing_type=None,
serialized_options=None,
serialized_start=542,
serialized_end=577,
)
_sym_db.RegisterEnumDescriptor(_TFLITETENSORSTOLANDMARKSCALCULATOROPTIONS_ACTIVATION)
_TFLITETENSORSTOLANDMARKSCALCULATOROPTIONS = _descriptor.Descriptor(
name='TfLiteTensorsToLandmarksCalculatorOptions',
full_name='mediapipe.TfLiteTensorsToLandmarksCalculatorOptions',
filename=None,
file=DESCRIPTOR,
containing_type=None,
fields=[
_descriptor.FieldDescriptor(
name='num_landmarks', full_name='mediapipe.TfLiteTensorsToLandmarksCalculatorOptions.num_landmarks', index=0,
number=1, type=5, cpp_type=1, label=2,
has_default_value=False, default_value=0,
message_type=None, enum_type=None, containing_type=None,
is_extension=False, extension_scope=None,
serialized_options=None, file=DESCRIPTOR),
_descriptor.FieldDescriptor(
name='input_image_width', full_name='mediapipe.TfLiteTensorsToLandmarksCalculatorOptions.input_image_width', index=1,
number=2, type=5, cpp_type=1, label=1,
has_default_value=False, default_value=0,
message_type=None, enum_type=None, containing_type=None,
is_extension=False, extension_scope=None,
serialized_options=None, file=DESCRIPTOR),
_descriptor.FieldDescriptor(
name='input_image_height', full_name='mediapipe.TfLiteTensorsToLandmarksCalculatorOptions.input_image_height', index=2,
number=3, type=5, cpp_type=1, label=1,
has_default_value=False, default_value=0,
message_type=None, enum_type=None, containing_type=None,
is_extension=False, extension_scope=None,
serialized_options=None, file=DESCRIPTOR),
_descriptor.FieldDescriptor(
name='flip_vertically', full_name='mediapipe.TfLiteTensorsToLandmarksCalculatorOptions.flip_vertically', index=3,
number=4, type=8, cpp_type=7, label=1,
has_default_value=True, default_value=False,
message_type=None, enum_type=None, containing_type=None,
is_extension=False, extension_scope=None,
serialized_options=None, file=DESCRIPTOR),
_descriptor.FieldDescriptor(
name='flip_horizontally', full_name='mediapipe.TfLiteTensorsToLandmarksCalculatorOptions.flip_horizontally', index=4,
number=6, type=8, cpp_type=7, label=1,
has_default_value=True, default_value=False,
message_type=None, enum_type=None, containing_type=None,
is_extension=False, extension_scope=None,
serialized_options=None, file=DESCRIPTOR),
_descriptor.FieldDescriptor(
name='normalize_z', full_name='mediapipe.TfLiteTensorsToLandmarksCalculatorOptions.normalize_z', index=5,
number=5, type=2, cpp_type=6, label=1,
has_default_value=True, default_value=float(1),
message_type=None, enum_type=None, containing_type=None,
is_extension=False, extension_scope=None,
serialized_options=None, file=DESCRIPTOR),
_descriptor.FieldDescriptor(
name='visibility_activation', full_name='mediapipe.TfLiteTensorsToLandmarksCalculatorOptions.visibility_activation', index=6,
number=7, type=14, cpp_type=8, label=1,
has_default_value=True, default_value=0,
message_type=None, enum_type=None, containing_type=None,
is_extension=False, extension_scope=None,
serialized_options=None, file=DESCRIPTOR),
_descriptor.FieldDescriptor(
name='presence_activation', full_name='mediapipe.TfLiteTensorsToLandmarksCalculatorOptions.presence_activation', index=7,
number=8, type=14, cpp_type=8, label=1,
has_default_value=True, default_value=0,
message_type=None, enum_type=None, containing_type=None,
is_extension=False, extension_scope=None,
serialized_options=None, file=DESCRIPTOR),
],
extensions=[
_descriptor.FieldDescriptor(
name='ext', full_name='mediapipe.TfLiteTensorsToLandmarksCalculatorOptions.ext', index=0,
number=257405002, type=11, cpp_type=10, label=1,
has_default_value=False, default_value=None,
message_type=None, enum_type=None, containing_type=None,
is_extension=True, extension_scope=None,
serialized_options=None, file=DESCRIPTOR),
],
nested_types=[],
enum_types=[
_TFLITETENSORSTOLANDMARKSCALCULATOROPTIONS_ACTIVATION,
],
serialized_options=None,
is_extendable=False,
syntax='proto2',
extension_ranges=[],
oneofs=[
],
serialized_start=127,
serialized_end=677,
)
_TFLITETENSORSTOLANDMARKSCALCULATOROPTIONS.fields_by_name['visibility_activation'].enum_type = _TFLITETENSORSTOLANDMARKSCALCULATOROPTIONS_ACTIVATION
_TFLITETENSORSTOLANDMARKSCALCULATOROPTIONS.fields_by_name['presence_activation'].enum_type = _TFLITETENSORSTOLANDMARKSCALCULATOROPTIONS_ACTIVATION
_TFLITETENSORSTOLANDMARKSCALCULATOROPTIONS_ACTIVATION.containing_type = _TFLITETENSORSTOLANDMARKSCALCULATOROPTIONS
DESCRIPTOR.message_types_by_name['TfLiteTensorsToLandmarksCalculatorOptions'] = _TFLITETENSORSTOLANDMARKSCALCULATOROPTIONS
_sym_db.RegisterFileDescriptor(DESCRIPTOR)
TfLiteTensorsToLandmarksCalculatorOptions = _reflection.GeneratedProtocolMessageType('TfLiteTensorsToLandmarksCalculatorOptions', (_message.Message,), dict(
DESCRIPTOR = _TFLITETENSORSTOLANDMARKSCALCULATOROPTIONS,
__module__ = 'mediapipe.calculators.tflite.tflite_tensors_to_landmarks_calculator_pb2'
# @@protoc_insertion_point(class_scope:mediapipe.TfLiteTensorsToLandmarksCalculatorOptions)
))
_sym_db.RegisterMessage(TfLiteTensorsToLandmarksCalculatorOptions)
_TFLITETENSORSTOLANDMARKSCALCULATOROPTIONS.extensions_by_name['ext'].message_type = _TFLITETENSORSTOLANDMARKSCALCULATOROPTIONS
mediapipe_dot_framework_dot_calculator__options__pb2.CalculatorOptions.RegisterExtension(_TFLITETENSORSTOLANDMARKSCALCULATOROPTIONS.extensions_by_name['ext'])
# @@protoc_insertion_point(module_scope)

View File

@@ -0,0 +1,120 @@
# -*- coding: utf-8 -*-
# Generated by the protocol buffer compiler. DO NOT EDIT!
# source: mediapipe/calculators/tflite/tflite_tensors_to_segmentation_calculator.proto
import sys
_b=sys.version_info[0]<3 and (lambda x:x) or (lambda x:x.encode('latin1'))
from google.protobuf import descriptor as _descriptor
from google.protobuf import message as _message
from google.protobuf import reflection as _reflection
from google.protobuf import symbol_database as _symbol_database
# @@protoc_insertion_point(imports)
_sym_db = _symbol_database.Default()
from 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.FileDescriptor(
name='mediapipe/calculators/tflite/tflite_tensors_to_segmentation_calculator.proto',
package='mediapipe',
syntax='proto2',
serialized_options=None,
serialized_pb=_b('\nLmediapipe/calculators/tflite/tflite_tensors_to_segmentation_calculator.proto\x12\tmediapipe\x1a$mediapipe/framework/calculator.proto\"\xbb\x02\n,TfLiteTensorsToSegmentationCalculatorOptions\x12\x14\n\x0ctensor_width\x18\x01 \x02(\x05\x12\x15\n\rtensor_height\x18\x02 \x02(\x05\x12\x17\n\x0ftensor_channels\x18\x03 \x02(\x05\x12&\n\x1b\x63ombine_with_previous_ratio\x18\x04 \x01(\x02:\x01\x31\x12\x1d\n\x12output_layer_index\x18\x05 \x01(\x05:\x01\x31\x12\x17\n\x0f\x66lip_vertically\x18\x06 \x01(\x08\x32\x65\n\x03\x65xt\x12\x1c.mediapipe.CalculatorOptions\x18\xca\xfb\xb4x \x01(\x0b\x32\x37.mediapipe.TfLiteTensorsToSegmentationCalculatorOptions')
,
dependencies=[mediapipe_dot_framework_dot_calculator__pb2.DESCRIPTOR,])
_TFLITETENSORSTOSEGMENTATIONCALCULATOROPTIONS = _descriptor.Descriptor(
name='TfLiteTensorsToSegmentationCalculatorOptions',
full_name='mediapipe.TfLiteTensorsToSegmentationCalculatorOptions',
filename=None,
file=DESCRIPTOR,
containing_type=None,
fields=[
_descriptor.FieldDescriptor(
name='tensor_width', full_name='mediapipe.TfLiteTensorsToSegmentationCalculatorOptions.tensor_width', index=0,
number=1, type=5, cpp_type=1, label=2,
has_default_value=False, default_value=0,
message_type=None, enum_type=None, containing_type=None,
is_extension=False, extension_scope=None,
serialized_options=None, file=DESCRIPTOR),
_descriptor.FieldDescriptor(
name='tensor_height', full_name='mediapipe.TfLiteTensorsToSegmentationCalculatorOptions.tensor_height', index=1,
number=2, type=5, cpp_type=1, label=2,
has_default_value=False, default_value=0,
message_type=None, enum_type=None, containing_type=None,
is_extension=False, extension_scope=None,
serialized_options=None, file=DESCRIPTOR),
_descriptor.FieldDescriptor(
name='tensor_channels', full_name='mediapipe.TfLiteTensorsToSegmentationCalculatorOptions.tensor_channels', index=2,
number=3, type=5, cpp_type=1, label=2,
has_default_value=False, default_value=0,
message_type=None, enum_type=None, containing_type=None,
is_extension=False, extension_scope=None,
serialized_options=None, file=DESCRIPTOR),
_descriptor.FieldDescriptor(
name='combine_with_previous_ratio', full_name='mediapipe.TfLiteTensorsToSegmentationCalculatorOptions.combine_with_previous_ratio', index=3,
number=4, type=2, cpp_type=6, label=1,
has_default_value=True, default_value=float(1),
message_type=None, enum_type=None, containing_type=None,
is_extension=False, extension_scope=None,
serialized_options=None, file=DESCRIPTOR),
_descriptor.FieldDescriptor(
name='output_layer_index', full_name='mediapipe.TfLiteTensorsToSegmentationCalculatorOptions.output_layer_index', index=4,
number=5, type=5, cpp_type=1, label=1,
has_default_value=True, default_value=1,
message_type=None, enum_type=None, containing_type=None,
is_extension=False, extension_scope=None,
serialized_options=None, file=DESCRIPTOR),
_descriptor.FieldDescriptor(
name='flip_vertically', full_name='mediapipe.TfLiteTensorsToSegmentationCalculatorOptions.flip_vertically', index=5,
number=6, type=8, cpp_type=7, label=1,
has_default_value=False, default_value=False,
message_type=None, enum_type=None, containing_type=None,
is_extension=False, extension_scope=None,
serialized_options=None, file=DESCRIPTOR),
],
extensions=[
_descriptor.FieldDescriptor(
name='ext', full_name='mediapipe.TfLiteTensorsToSegmentationCalculatorOptions.ext', index=0,
number=252526026, type=11, cpp_type=10, label=1,
has_default_value=False, default_value=None,
message_type=None, enum_type=None, containing_type=None,
is_extension=True, extension_scope=None,
serialized_options=None, file=DESCRIPTOR),
],
nested_types=[],
enum_types=[
],
serialized_options=None,
is_extendable=False,
syntax='proto2',
extension_ranges=[],
oneofs=[
],
serialized_start=130,
serialized_end=445,
)
DESCRIPTOR.message_types_by_name['TfLiteTensorsToSegmentationCalculatorOptions'] = _TFLITETENSORSTOSEGMENTATIONCALCULATOROPTIONS
_sym_db.RegisterFileDescriptor(DESCRIPTOR)
TfLiteTensorsToSegmentationCalculatorOptions = _reflection.GeneratedProtocolMessageType('TfLiteTensorsToSegmentationCalculatorOptions', (_message.Message,), dict(
DESCRIPTOR = _TFLITETENSORSTOSEGMENTATIONCALCULATOROPTIONS,
__module__ = 'mediapipe.calculators.tflite.tflite_tensors_to_segmentation_calculator_pb2'
# @@protoc_insertion_point(class_scope:mediapipe.TfLiteTensorsToSegmentationCalculatorOptions)
))
_sym_db.RegisterMessage(TfLiteTensorsToSegmentationCalculatorOptions)
_TFLITETENSORSTOSEGMENTATIONCALCULATOROPTIONS.extensions_by_name['ext'].message_type = _TFLITETENSORSTOSEGMENTATIONCALCULATOROPTIONS
mediapipe_dot_framework_dot_calculator__options__pb2.CalculatorOptions.RegisterExtension(_TFLITETENSORSTOSEGMENTATIONCALCULATOROPTIONS.extensions_by_name['ext'])
# @@protoc_insertion_point(module_scope)