Initial commit
This commit is contained in:
Untriex Programming
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
@@ -0,0 +1,167 @@
|
||||
# -*- coding: utf-8 -*-
|
||||
# Generated by the protocol buffer compiler. DO NOT EDIT!
|
||||
# source: mediapipe/util/audio_decoder.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/util/audio_decoder.proto',
|
||||
package='mediapipe',
|
||||
syntax='proto2',
|
||||
serialized_options=None,
|
||||
serialized_pb=_b('\n\"mediapipe/util/audio_decoder.proto\x12\tmediapipe\x1a$mediapipe/framework/calculator.proto\"\xc1\x01\n\x12\x41udioStreamOptions\x12\x17\n\x0cstream_index\x18\x01 \x01(\x03:\x01\x30\x12\x1c\n\rallow_missing\x18\x02 \x01(\x08:\x05\x66\x61lse\x12%\n\x16ignore_decode_failures\x18\x03 \x01(\x08:\x05\x66\x61lse\x12+\n\x1coutput_regressing_timestamps\x18\x04 \x01(\x08:\x05\x66\x61lse\x12 \n\x18\x63orrect_pts_for_rollover\x18\x05 \x01(\x08\"\xbe\x01\n\x13\x41udioDecoderOptions\x12\x33\n\x0c\x61udio_stream\x18\x01 \x03(\x0b\x32\x1d.mediapipe.AudioStreamOptions\x12\x12\n\nstart_time\x18\x02 \x01(\x01\x12\x10\n\x08\x65nd_time\x18\x03 \x01(\x01\x32L\n\x03\x65xt\x12\x1c.mediapipe.CalculatorOptions\x18\xb2\xef\xca} \x01(\x0b\x32\x1e.mediapipe.AudioDecoderOptions')
|
||||
,
|
||||
dependencies=[mediapipe_dot_framework_dot_calculator__pb2.DESCRIPTOR,])
|
||||
|
||||
|
||||
|
||||
|
||||
_AUDIOSTREAMOPTIONS = _descriptor.Descriptor(
|
||||
name='AudioStreamOptions',
|
||||
full_name='mediapipe.AudioStreamOptions',
|
||||
filename=None,
|
||||
file=DESCRIPTOR,
|
||||
containing_type=None,
|
||||
fields=[
|
||||
_descriptor.FieldDescriptor(
|
||||
name='stream_index', full_name='mediapipe.AudioStreamOptions.stream_index', index=0,
|
||||
number=1, type=3, cpp_type=2, 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_missing', full_name='mediapipe.AudioStreamOptions.allow_missing', 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=None, file=DESCRIPTOR),
|
||||
_descriptor.FieldDescriptor(
|
||||
name='ignore_decode_failures', full_name='mediapipe.AudioStreamOptions.ignore_decode_failures', 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=None, file=DESCRIPTOR),
|
||||
_descriptor.FieldDescriptor(
|
||||
name='output_regressing_timestamps', full_name='mediapipe.AudioStreamOptions.output_regressing_timestamps', 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='correct_pts_for_rollover', full_name='mediapipe.AudioStreamOptions.correct_pts_for_rollover', index=4,
|
||||
number=5, 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=[
|
||||
],
|
||||
nested_types=[],
|
||||
enum_types=[
|
||||
],
|
||||
serialized_options=None,
|
||||
is_extendable=False,
|
||||
syntax='proto2',
|
||||
extension_ranges=[],
|
||||
oneofs=[
|
||||
],
|
||||
serialized_start=88,
|
||||
serialized_end=281,
|
||||
)
|
||||
|
||||
|
||||
_AUDIODECODEROPTIONS = _descriptor.Descriptor(
|
||||
name='AudioDecoderOptions',
|
||||
full_name='mediapipe.AudioDecoderOptions',
|
||||
filename=None,
|
||||
file=DESCRIPTOR,
|
||||
containing_type=None,
|
||||
fields=[
|
||||
_descriptor.FieldDescriptor(
|
||||
name='audio_stream', full_name='mediapipe.AudioDecoderOptions.audio_stream', index=0,
|
||||
number=1, type=11, cpp_type=10, 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='start_time', full_name='mediapipe.AudioDecoderOptions.start_time', index=1,
|
||||
number=2, type=1, cpp_type=5, 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='end_time', full_name='mediapipe.AudioDecoderOptions.end_time', index=2,
|
||||
number=3, type=1, cpp_type=5, 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.AudioDecoderOptions.ext', index=0,
|
||||
number=263370674, 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=284,
|
||||
serialized_end=474,
|
||||
)
|
||||
|
||||
_AUDIODECODEROPTIONS.fields_by_name['audio_stream'].message_type = _AUDIOSTREAMOPTIONS
|
||||
DESCRIPTOR.message_types_by_name['AudioStreamOptions'] = _AUDIOSTREAMOPTIONS
|
||||
DESCRIPTOR.message_types_by_name['AudioDecoderOptions'] = _AUDIODECODEROPTIONS
|
||||
_sym_db.RegisterFileDescriptor(DESCRIPTOR)
|
||||
|
||||
AudioStreamOptions = _reflection.GeneratedProtocolMessageType('AudioStreamOptions', (_message.Message,), dict(
|
||||
DESCRIPTOR = _AUDIOSTREAMOPTIONS,
|
||||
__module__ = 'mediapipe.util.audio_decoder_pb2'
|
||||
# @@protoc_insertion_point(class_scope:mediapipe.AudioStreamOptions)
|
||||
))
|
||||
_sym_db.RegisterMessage(AudioStreamOptions)
|
||||
|
||||
AudioDecoderOptions = _reflection.GeneratedProtocolMessageType('AudioDecoderOptions', (_message.Message,), dict(
|
||||
DESCRIPTOR = _AUDIODECODEROPTIONS,
|
||||
__module__ = 'mediapipe.util.audio_decoder_pb2'
|
||||
# @@protoc_insertion_point(class_scope:mediapipe.AudioDecoderOptions)
|
||||
))
|
||||
_sym_db.RegisterMessage(AudioDecoderOptions)
|
||||
|
||||
_AUDIODECODEROPTIONS.extensions_by_name['ext'].message_type = _AUDIODECODEROPTIONS
|
||||
mediapipe_dot_framework_dot_calculator__options__pb2.CalculatorOptions.RegisterExtension(_AUDIODECODEROPTIONS.extensions_by_name['ext'])
|
||||
|
||||
# @@protoc_insertion_point(module_scope)
|
||||
172
.venv/lib/python3.9/site-packages/mediapipe/util/color_pb2.py
Normal file
172
.venv/lib/python3.9/site-packages/mediapipe/util/color_pb2.py
Normal file
@@ -0,0 +1,172 @@
|
||||
# -*- coding: utf-8 -*-
|
||||
# Generated by the protocol buffer compiler. DO NOT EDIT!
|
||||
# source: mediapipe/util/color.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()
|
||||
|
||||
|
||||
|
||||
|
||||
DESCRIPTOR = _descriptor.FileDescriptor(
|
||||
name='mediapipe/util/color.proto',
|
||||
package='mediapipe',
|
||||
syntax='proto2',
|
||||
serialized_options=None,
|
||||
serialized_pb=_b('\n\x1amediapipe/util/color.proto\x12\tmediapipe\"(\n\x05\x43olor\x12\t\n\x01r\x18\x01 \x01(\x05\x12\t\n\x01g\x18\x02 \x01(\x05\x12\t\n\x01\x62\x18\x03 \x01(\x05\"\x90\x01\n\x08\x43olorMap\x12=\n\x0elabel_to_color\x18\x01 \x03(\x0b\x32%.mediapipe.ColorMap.LabelToColorEntry\x1a\x45\n\x11LabelToColorEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\x1f\n\x05value\x18\x02 \x01(\x0b\x32\x10.mediapipe.Color:\x02\x38\x01')
|
||||
)
|
||||
|
||||
|
||||
|
||||
|
||||
_COLOR = _descriptor.Descriptor(
|
||||
name='Color',
|
||||
full_name='mediapipe.Color',
|
||||
filename=None,
|
||||
file=DESCRIPTOR,
|
||||
containing_type=None,
|
||||
fields=[
|
||||
_descriptor.FieldDescriptor(
|
||||
name='r', full_name='mediapipe.Color.r', index=0,
|
||||
number=1, 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='g', full_name='mediapipe.Color.g', 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='b', full_name='mediapipe.Color.b', 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),
|
||||
],
|
||||
extensions=[
|
||||
],
|
||||
nested_types=[],
|
||||
enum_types=[
|
||||
],
|
||||
serialized_options=None,
|
||||
is_extendable=False,
|
||||
syntax='proto2',
|
||||
extension_ranges=[],
|
||||
oneofs=[
|
||||
],
|
||||
serialized_start=41,
|
||||
serialized_end=81,
|
||||
)
|
||||
|
||||
|
||||
_COLORMAP_LABELTOCOLORENTRY = _descriptor.Descriptor(
|
||||
name='LabelToColorEntry',
|
||||
full_name='mediapipe.ColorMap.LabelToColorEntry',
|
||||
filename=None,
|
||||
file=DESCRIPTOR,
|
||||
containing_type=None,
|
||||
fields=[
|
||||
_descriptor.FieldDescriptor(
|
||||
name='key', full_name='mediapipe.ColorMap.LabelToColorEntry.key', 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='value', full_name='mediapipe.ColorMap.LabelToColorEntry.value', 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),
|
||||
],
|
||||
extensions=[
|
||||
],
|
||||
nested_types=[],
|
||||
enum_types=[
|
||||
],
|
||||
serialized_options=_b('8\001'),
|
||||
is_extendable=False,
|
||||
syntax='proto2',
|
||||
extension_ranges=[],
|
||||
oneofs=[
|
||||
],
|
||||
serialized_start=159,
|
||||
serialized_end=228,
|
||||
)
|
||||
|
||||
_COLORMAP = _descriptor.Descriptor(
|
||||
name='ColorMap',
|
||||
full_name='mediapipe.ColorMap',
|
||||
filename=None,
|
||||
file=DESCRIPTOR,
|
||||
containing_type=None,
|
||||
fields=[
|
||||
_descriptor.FieldDescriptor(
|
||||
name='label_to_color', full_name='mediapipe.ColorMap.label_to_color', index=0,
|
||||
number=1, type=11, cpp_type=10, 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),
|
||||
],
|
||||
extensions=[
|
||||
],
|
||||
nested_types=[_COLORMAP_LABELTOCOLORENTRY, ],
|
||||
enum_types=[
|
||||
],
|
||||
serialized_options=None,
|
||||
is_extendable=False,
|
||||
syntax='proto2',
|
||||
extension_ranges=[],
|
||||
oneofs=[
|
||||
],
|
||||
serialized_start=84,
|
||||
serialized_end=228,
|
||||
)
|
||||
|
||||
_COLORMAP_LABELTOCOLORENTRY.fields_by_name['value'].message_type = _COLOR
|
||||
_COLORMAP_LABELTOCOLORENTRY.containing_type = _COLORMAP
|
||||
_COLORMAP.fields_by_name['label_to_color'].message_type = _COLORMAP_LABELTOCOLORENTRY
|
||||
DESCRIPTOR.message_types_by_name['Color'] = _COLOR
|
||||
DESCRIPTOR.message_types_by_name['ColorMap'] = _COLORMAP
|
||||
_sym_db.RegisterFileDescriptor(DESCRIPTOR)
|
||||
|
||||
Color = _reflection.GeneratedProtocolMessageType('Color', (_message.Message,), dict(
|
||||
DESCRIPTOR = _COLOR,
|
||||
__module__ = 'mediapipe.util.color_pb2'
|
||||
# @@protoc_insertion_point(class_scope:mediapipe.Color)
|
||||
))
|
||||
_sym_db.RegisterMessage(Color)
|
||||
|
||||
ColorMap = _reflection.GeneratedProtocolMessageType('ColorMap', (_message.Message,), dict(
|
||||
|
||||
LabelToColorEntry = _reflection.GeneratedProtocolMessageType('LabelToColorEntry', (_message.Message,), dict(
|
||||
DESCRIPTOR = _COLORMAP_LABELTOCOLORENTRY,
|
||||
__module__ = 'mediapipe.util.color_pb2'
|
||||
# @@protoc_insertion_point(class_scope:mediapipe.ColorMap.LabelToColorEntry)
|
||||
))
|
||||
,
|
||||
DESCRIPTOR = _COLORMAP,
|
||||
__module__ = 'mediapipe.util.color_pb2'
|
||||
# @@protoc_insertion_point(class_scope:mediapipe.ColorMap)
|
||||
))
|
||||
_sym_db.RegisterMessage(ColorMap)
|
||||
_sym_db.RegisterMessage(ColorMap.LabelToColorEntry)
|
||||
|
||||
|
||||
_COLORMAP_LABELTOCOLORENTRY._options = None
|
||||
# @@protoc_insertion_point(module_scope)
|
||||
1032
.venv/lib/python3.9/site-packages/mediapipe/util/render_data_pb2.py
Normal file
1032
.venv/lib/python3.9/site-packages/mediapipe/util/render_data_pb2.py
Normal file
File diff suppressed because it is too large
Load Diff
@@ -0,0 +1,14 @@
|
||||
"""Copyright 2019 The MediaPipe Authors.
|
||||
|
||||
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.
|
||||
"""
|
||||
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
@@ -0,0 +1,679 @@
|
||||
"""Copyright 2019 The MediaPipe Authors.
|
||||
|
||||
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.
|
||||
|
||||
|
||||
This script defines a large number of getters and setters for storing
|
||||
multimedia, such as video or audio, and related machine learning data in
|
||||
tf.train.SequenceExamples. These getters and setters simplify sharing
|
||||
data by enforcing common patterns for storing data in SequenceExample
|
||||
key-value pairs.
|
||||
|
||||
The constants, macros, and functions are organized into 6 groups: clip
|
||||
metadata, clip label related, segment related, bounding-box related, image
|
||||
related, feature list related, and keyframe related. The following examples
|
||||
will walk through common task structures, but the relevant data to store can
|
||||
vary by task.
|
||||
|
||||
The clip metadata group is generally data about the media and stored in the
|
||||
SequenceExample.context. Specifying the metadata enables media pipelines,
|
||||
such as MediaPipe, to retrieve that data. Typically, set_clip_data_path,
|
||||
set_clip_start_timestamp, and set_clip_end_timestamp define which data to use
|
||||
without storing the data itself. Example:
|
||||
tensorflow.train.SequenceExample sequence
|
||||
set_clip_data_path("/relative/path/to/data.mp4", sequence)
|
||||
set_clip_start_timestamp(0, sequence)
|
||||
set_clip_end_timestamp(10000000, sequence) # 10 seconds in microseconds.
|
||||
|
||||
The clip label group adds labels that apply to the entire media clip. To
|
||||
annotate that a video clip has a particular label, set the clip metadata
|
||||
above and also set the set_clip_label_index and set_clip_label_string. Most
|
||||
training pipelines will only use the label index or string, but we recommend
|
||||
storing both to improve readability while maintaining ease of use.
|
||||
Example:
|
||||
set_clip_label_string(("run", "jump"), sequence)
|
||||
set_Clip_label_index((35, 47), sequence)
|
||||
|
||||
The segment group is generally data about time spans within the media clip
|
||||
and stored in the SequenceExample.context. In this code, continuous lengths
|
||||
of media are called clips, and each clip may have subregions of interest that
|
||||
are called segments. To annotate that a video clip has time spans with labels
|
||||
set the clip metadata above and use the functions set_segment_start_timestamp,
|
||||
set_segment_end_timestamp, set_segment_label_index, and
|
||||
set_segment_label_string. Most training pipelines will only use the label index
|
||||
or string, but we recommend storing both to improve readability while
|
||||
maintaining ease of use. By listing segments as times, the frame rate or other
|
||||
properties can change without affecting the labels.
|
||||
Example:
|
||||
set_segment_start_timestamp((500000, 1000000), sequence) # in microseconds
|
||||
set_segment_end_timestamp((2000000, 6000000), sequence)
|
||||
set_segment_label_index((35, 47), sequence)
|
||||
set_segment_label_string(("run", "jump"), sequence)
|
||||
|
||||
The bounding box group is useful for identifying spatio-temporal annotations
|
||||
for detection, tracking, or action recognition. The exact keys that are
|
||||
needed can vary by task, but to annotate a video clip for detection set the
|
||||
clip metadata above and use repeatedly call add_bbox, add_bbox_timestamp,
|
||||
add_bbox_label_index, and add_bbox_label_string. Most training pipelines will
|
||||
only use the label index or string, but we recommend storing both to improve
|
||||
readability while maintaining ease of use. Because bounding boxes are
|
||||
assigned to timepoints in a video, changing the image frame rate can can
|
||||
change the alignment. The media_sequence.h's ReconcileMetadata function can
|
||||
align bounding boxes to the nearest image.
|
||||
|
||||
The image group is useful for storing data as sequential 2D arrays, typically
|
||||
encoded as bytes. Images can be RGB images stored as JPEG, discrete masks
|
||||
stored as PNG, or some other format. Parameters that are static over time are
|
||||
set in the context using set_image_width, set_image_height, set_image_format,
|
||||
etc. The series of frames and timestamps are then added with add_image_encoded
|
||||
and
|
||||
add_image_timestamp. For discrete masks, the class or instance indices can be
|
||||
mapped to labels or classes using
|
||||
set_class_segmentation_class_label_{index,string} and
|
||||
set_instance_segmentation_object_class_index.
|
||||
|
||||
The feature list group is useful for storing audio and extracted features,
|
||||
such as per-frame embeddings. SequenceExamples only store lists of floats per
|
||||
timestep, so the dimensions are stored in the context to enable reshaping.
|
||||
For example, set_feature_dimensions and repeatedly calling add_feature_floats
|
||||
and add_feature_timestamp adds per-frame embeddings. The feature methods also
|
||||
support audio features.
|
||||
|
||||
Macros for common patterns are created in media_sequence_util.py and are used
|
||||
here extensively. Because these macros are formulaic, I will only include a
|
||||
usage example here in the code rather than repeating documentation for every
|
||||
instance. This header defines additional functions to simplify working with
|
||||
MediaPipe types.
|
||||
|
||||
Each msu.create_{TYPE}_context_feature takes a NAME and a KEY. It provides
|
||||
setters and getters for SequenceExamples and stores a single value under KEY
|
||||
in the context field. The provided functions are has_${NAME}, get_${NAME},
|
||||
set_${Name}, and clear_${NAME}.
|
||||
Eg.
|
||||
tf.train.SequenceExample example
|
||||
set_data_path("data_path", example)
|
||||
if has_data_path(example):
|
||||
data_path = get_data_path(example)
|
||||
clear_data_path(example)
|
||||
|
||||
Each msu.create_{TYPE}_list_context_feature takes a NAME and a KEY. It provides
|
||||
setters and getters for SequenceExamples and stores a sequence of values
|
||||
under KEY in the context field. The provided functions are has_${NAME},
|
||||
get_${NAME}, set_${Name}, clear_${NAME}, get_${NAME}_at, and add_${NAME}.
|
||||
Eg.
|
||||
tf.train.SequenceExample example
|
||||
set_clip_label_string(("run", "jump"), example)
|
||||
if has_clip_label_string(example):
|
||||
values = get_clip_label_string(example)
|
||||
clear_clip_label_string(example)
|
||||
|
||||
Each msu.create_{TYPE}_feature_list takes a NAME and a KEY. It provides setters
|
||||
and getters for SequenceExamples and stores a single value in each feature field
|
||||
under KEY of the feature_lists field. The provided functions are has_${NAME},
|
||||
get_${NAME}, clear_${NAME}, get_${NAME}_size, get_${NAME}_at, and add_${NAME}.
|
||||
tf.train.SequenceExample example
|
||||
add_image_timestamp(1000000, example)
|
||||
add_image_timestamp(2000000, example)
|
||||
if has_image_timestamp(example):
|
||||
for i in range(get_image_timestamp_size()):
|
||||
timestamp = get_image_timestamp_at(example, i)
|
||||
clear_image_timestamp(example)
|
||||
|
||||
Each VECTOR_{TYPE}_FEATURE_LIST takes a NAME and a KEY. It provides setters
|
||||
and getters for SequenceExamples and stores a sequence of values in each
|
||||
feature field under KEY of the feature_lists field. The provided functions
|
||||
are Has${NAME}, Get${NAME}, Clear${NAME}, Get${NAME}Size, Get${NAME}At, and
|
||||
Add${NAME}.
|
||||
tf.train.SequenceExample example
|
||||
add_bbox_label_string(("run", "jump"), example)
|
||||
add_bbox_label_string(("run", "fall"), example)
|
||||
if has_bbox_label_string(example):
|
||||
for i in range(get_bbox_label_string_size(example)):
|
||||
labels = get_bbox_label_string_at(example, i)
|
||||
clear_bbox_label_string(example)
|
||||
|
||||
As described in media_sequence_util.h, each of these functions can take an
|
||||
additional string prefix argument as their first argument. The prefix can
|
||||
be fixed with a new NAME by using functools.partial. Prefixes are used to
|
||||
identify common storage patterns (e.g. storing an image along with the height
|
||||
and width) under different names (e.g. storing a left and right image in a
|
||||
stereo pair.) An example creating functions such as
|
||||
add_left_image_encoded that adds a string under the key "LEFT/image/encoded"
|
||||
add_left_image_encoded = msu.function_with_default(add_image_encoded, "LEFT")
|
||||
"""
|
||||
|
||||
from __future__ import absolute_import
|
||||
from __future__ import division
|
||||
from __future__ import print_function
|
||||
import numpy as np
|
||||
from mediapipe.util.sequence import media_sequence_util
|
||||
msu = media_sequence_util
|
||||
|
||||
_HAS_DYNAMIC_ATTRIBUTES = True
|
||||
|
||||
################################## METADATA #################################
|
||||
# A unique identifier for each example.
|
||||
EXAMPLE_ID_KEY = "example/id"
|
||||
# The name o fthe data set, including the version.
|
||||
EXAMPLE_DATASET_NAME_KEY = "example/dataset_name"
|
||||
# The relative path to the data on disk from some root directory.
|
||||
CLIP_DATA_PATH_KEY = "clip/data_path"
|
||||
# Any identifier for the media beyond the data path.
|
||||
CLIP_MEDIA_ID_KEY = "clip/media_id"
|
||||
# Yet another alternative identifier.
|
||||
ALTERNATIVE_CLIP_MEDIA_ID_KEY = "clip/alternative_media_id"
|
||||
# The encoded bytes for storing media directly in the SequenceExample.
|
||||
CLIP_ENCODED_MEDIA_BYTES_KEY = "clip/encoded_media_bytes"
|
||||
# The start time for the encoded media if not preserved during encoding.
|
||||
CLIP_ENCODED_MEDIA_START_TIMESTAMP_KEY = "clip/encoded_media_start_timestamp"
|
||||
# The start time, in microseconds, for the start of the clip in the media.
|
||||
CLIP_START_TIMESTAMP_KEY = "clip/start/timestamp"
|
||||
# The end time, in microseconds, for the end of the clip in the media.
|
||||
CLIP_END_TIMESTAMP_KEY = "clip/end/timestamp"
|
||||
# A list of label indices for this clip.
|
||||
CLIP_LABEL_INDEX_KEY = "clip/label/index"
|
||||
# A list of label strings for this clip.
|
||||
CLIP_LABEL_STRING_KEY = "clip/label/string"
|
||||
# A list of label confidences for this clip.
|
||||
CLIP_LABEL_CONFIDENCE_KEY = "clip/label/confidence"
|
||||
msu.create_bytes_context_feature(
|
||||
"example_id", EXAMPLE_ID_KEY, module_dict=globals())
|
||||
msu.create_bytes_context_feature(
|
||||
"example_dataset_name", EXAMPLE_DATASET_NAME_KEY, module_dict=globals())
|
||||
msu.create_bytes_context_feature(
|
||||
"clip_media_id", CLIP_MEDIA_ID_KEY, module_dict=globals())
|
||||
msu.create_bytes_context_feature(
|
||||
"clip_alternative_media_id", ALTERNATIVE_CLIP_MEDIA_ID_KEY,
|
||||
module_dict=globals())
|
||||
msu.create_bytes_context_feature(
|
||||
"clip_encoded_media_bytes", CLIP_ENCODED_MEDIA_BYTES_KEY,
|
||||
module_dict=globals())
|
||||
msu.create_bytes_context_feature(
|
||||
"clip_data_path", CLIP_DATA_PATH_KEY, module_dict=globals())
|
||||
msu.create_int_context_feature(
|
||||
"clip_encoded_media_start_timestamp",
|
||||
CLIP_ENCODED_MEDIA_START_TIMESTAMP_KEY, module_dict=globals())
|
||||
msu.create_int_context_feature(
|
||||
"clip_start_timestamp", CLIP_START_TIMESTAMP_KEY, module_dict=globals())
|
||||
msu.create_int_context_feature(
|
||||
"clip_end_timestamp", CLIP_END_TIMESTAMP_KEY, module_dict=globals())
|
||||
msu.create_bytes_list_context_feature(
|
||||
"clip_label_string", CLIP_LABEL_STRING_KEY, module_dict=globals())
|
||||
msu.create_int_list_context_feature(
|
||||
"clip_label_index", CLIP_LABEL_INDEX_KEY, module_dict=globals())
|
||||
msu.create_float_list_context_feature(
|
||||
"clip_label_confidence", CLIP_LABEL_CONFIDENCE_KEY, module_dict=globals())
|
||||
|
||||
################################## SEGMENTS #################################
|
||||
# A list of segment start times in microseconds.
|
||||
SEGMENT_START_TIMESTAMP_KEY = "segment/start/timestamp"
|
||||
# A list of indices marking the first frame index >= the start timestamp.
|
||||
SEGMENT_START_INDEX_KEY = "segment/start/index"
|
||||
# A list of segment end times in microseconds.
|
||||
SEGMENT_END_TIMESTAMP_KEY = "segment/end/timestamp"
|
||||
# A list of indices marking the last frame index <= the end timestamp.
|
||||
SEGMENT_END_INDEX_KEY = "segment/end/index"
|
||||
# A list with the label index for each segment.
|
||||
# Multiple labels for the same segment are encoded as repeated segments.
|
||||
SEGMENT_LABEL_INDEX_KEY = "segment/label/index"
|
||||
# A list with the label string for each segment.
|
||||
# Multiple labels for the same segment are encoded as repeated segments.
|
||||
SEGMENT_LABEL_STRING_KEY = "segment/label/string"
|
||||
# A list with the label confidence for each segment.
|
||||
# Multiple labels for the same segment are encoded as repeated segments.
|
||||
SEGMENT_LABEL_CONFIDENCE_KEY = "segment/label/confidence"
|
||||
msu.create_bytes_list_context_feature(
|
||||
"segment_label_string", SEGMENT_LABEL_STRING_KEY, module_dict=globals())
|
||||
msu.create_int_list_context_feature(
|
||||
"segment_start_timestamp",
|
||||
SEGMENT_START_TIMESTAMP_KEY, module_dict=globals())
|
||||
msu.create_int_list_context_feature(
|
||||
"segment_start_index", SEGMENT_START_INDEX_KEY, module_dict=globals())
|
||||
msu.create_int_list_context_feature(
|
||||
"segment_end_timestamp", SEGMENT_END_TIMESTAMP_KEY, module_dict=globals())
|
||||
msu.create_int_list_context_feature(
|
||||
"segment_end_index", SEGMENT_END_INDEX_KEY, module_dict=globals())
|
||||
msu.create_int_list_context_feature(
|
||||
"segment_label_index", SEGMENT_LABEL_INDEX_KEY, module_dict=globals())
|
||||
msu.create_float_list_context_feature(
|
||||
"segment_label_confidence",
|
||||
SEGMENT_LABEL_CONFIDENCE_KEY, module_dict=globals())
|
||||
|
||||
########################## REGIONS / BOUNDING BOXES #########################
|
||||
|
||||
# Normalized coordinates of bounding boxes are provided in four lists to avoid
|
||||
# order ambiguity. We provide additional accessors for complete bounding boxes
|
||||
# below.
|
||||
REGION_BBOX_YMIN_KEY = "region/bbox/ymin"
|
||||
REGION_BBOX_XMIN_KEY = "region/bbox/xmin"
|
||||
REGION_BBOX_YMAX_KEY = "region/bbox/ymax"
|
||||
REGION_BBOX_XMAX_KEY = "region/bbox/xmax"
|
||||
# The point and radius can denote keypoints.
|
||||
REGION_POINT_X_KEY = "region/point/x"
|
||||
REGION_POINT_Y_KEY = "region/point/y"
|
||||
REGION_RADIUS_KEY = "region/radius"
|
||||
# The 3D point can denote keypoints.
|
||||
REGION_3D_POINT_X_KEY = "region/3d_point/x"
|
||||
REGION_3D_POINT_Y_KEY = "region/3d_point/y"
|
||||
REGION_3D_POINT_Z_KEY = "region/3d_point/z"
|
||||
# The number of regions at that timestep.
|
||||
REGION_NUM_REGIONS_KEY = "region/num_regions"
|
||||
# Whether that timestep is annotated for regions.
|
||||
# (Disambiguates between multiple meanings of num_regions = 0.)
|
||||
REGION_IS_ANNOTATED_KEY = "region/is_annotated"
|
||||
# A list indicating if each region is generated (1) or manually annotated (0)
|
||||
REGION_IS_GENERATED_KEY = "region/is_generated"
|
||||
# A list indicating if each region is occluded (1) or visible (0)
|
||||
REGION_IS_OCCLUDED_KEY = "region/is_occluded"
|
||||
# Lists with a label for each region.
|
||||
# Multiple labels for the same region require duplicating the region.
|
||||
REGION_LABEL_INDEX_KEY = "region/label/index"
|
||||
REGION_LABEL_STRING_KEY = "region/label/string"
|
||||
REGION_LABEL_CONFIDENCE_KEY = "region/label/confidence"
|
||||
# Lists with a track identifier for each region.
|
||||
# Multiple track identifier for the same region require duplicating the region.
|
||||
REGION_TRACK_INDEX_KEY = "region/track/index"
|
||||
REGION_TRACK_STRING_KEY = "region/track/string"
|
||||
REGION_TRACK_CONFIDENCE_KEY = "region/track/confidence"
|
||||
# Lists with a class for each region. In general, prefer to use the label
|
||||
# fields. These class fields exist to distinguish tracks when different classes
|
||||
# have overlapping track ids.
|
||||
REGION_CLASS_INDEX_KEY = "region/class/index"
|
||||
REGION_CLASS_STRING_KEY = "region/class/string"
|
||||
REGION_CLASS_CONFIDENCE_KEY = "region/class/confidence"
|
||||
# The timestamp of the region annotation in microseconds.
|
||||
REGION_TIMESTAMP_KEY = "region/timestamp"
|
||||
# The original timestamp in microseconds for region annotations.
|
||||
# If regions are aligned to image frames, this field preserves the original
|
||||
# timestamps.
|
||||
REGION_UNMODIFIED_TIMESTAMP_KEY = "region/unmodified_timestamp"
|
||||
# The list of region parts expected in this example.
|
||||
REGION_PARTS_KEY = "region/parts"
|
||||
# The dimensions of each embedding per region / bounding box.
|
||||
REGION_EMBEDDING_DIMENSIONS_PER_REGION_KEY = (
|
||||
"region/embedding/dimensions_per_region")
|
||||
# The format encoding embeddings as strings.
|
||||
REGION_EMBEDDING_FORMAT_KEY = "region/embedding/format"
|
||||
# An embedding for each region. The length of each list must be the product of
|
||||
# the number of regions and the product of the embedding dimensions.
|
||||
REGION_EMBEDDING_FLOAT_KEY = "region/embedding/float"
|
||||
# A string encoded embedding for each regions.
|
||||
REGION_EMBEDDING_ENCODED_KEY = "region/embedding/encoded"
|
||||
# The confidence of the embedding.
|
||||
REGION_EMBEDDING_CONFIDENCE_KEY = "region/embedding/confidence"
|
||||
|
||||
|
||||
def _create_region_with_prefix(name, prefix):
|
||||
"""Create multiple accessors for region based data."""
|
||||
msu.create_int_feature_list(name + "_num_regions", REGION_NUM_REGIONS_KEY,
|
||||
prefix=prefix, module_dict=globals())
|
||||
msu.create_int_feature_list(name + "_is_annotated", REGION_IS_ANNOTATED_KEY,
|
||||
prefix=prefix, module_dict=globals())
|
||||
msu.create_int_list_feature_list(
|
||||
name + "_is_occluded", REGION_IS_OCCLUDED_KEY,
|
||||
prefix=prefix, module_dict=globals())
|
||||
msu.create_int_list_feature_list(
|
||||
name + "_is_generated", REGION_IS_GENERATED_KEY,
|
||||
prefix=prefix, module_dict=globals())
|
||||
msu.create_int_feature_list(name + "_timestamp", REGION_TIMESTAMP_KEY,
|
||||
prefix=prefix, module_dict=globals())
|
||||
msu.create_int_feature_list(
|
||||
name + "_unmodified_timestamp", REGION_UNMODIFIED_TIMESTAMP_KEY,
|
||||
prefix=prefix, module_dict=globals())
|
||||
msu.create_bytes_list_feature_list(
|
||||
name + "_label_string", REGION_LABEL_STRING_KEY,
|
||||
prefix=prefix, module_dict=globals())
|
||||
msu.create_int_list_feature_list(
|
||||
name + "_label_index", REGION_LABEL_INDEX_KEY,
|
||||
prefix=prefix, module_dict=globals())
|
||||
msu.create_float_list_feature_list(
|
||||
name + "_label_confidence", REGION_LABEL_CONFIDENCE_KEY,
|
||||
prefix=prefix, module_dict=globals())
|
||||
msu.create_bytes_list_feature_list(
|
||||
name + "_class_string", REGION_CLASS_STRING_KEY,
|
||||
prefix=prefix, module_dict=globals())
|
||||
msu.create_int_list_feature_list(
|
||||
name + "_class_index", REGION_CLASS_INDEX_KEY,
|
||||
prefix=prefix, module_dict=globals())
|
||||
msu.create_float_list_feature_list(
|
||||
name + "_class_confidence", REGION_CLASS_CONFIDENCE_KEY,
|
||||
prefix=prefix, module_dict=globals())
|
||||
msu.create_bytes_list_feature_list(
|
||||
name + "_track_string", REGION_TRACK_STRING_KEY,
|
||||
prefix=prefix, module_dict=globals())
|
||||
msu.create_int_list_feature_list(
|
||||
name + "_track_index", REGION_TRACK_INDEX_KEY,
|
||||
prefix=prefix, module_dict=globals())
|
||||
msu.create_float_list_feature_list(
|
||||
name + "_track_confidence", REGION_TRACK_CONFIDENCE_KEY,
|
||||
prefix=prefix, module_dict=globals())
|
||||
msu.create_float_list_feature_list(name + "_ymin", REGION_BBOX_YMIN_KEY,
|
||||
prefix=prefix, module_dict=globals())
|
||||
msu.create_float_list_feature_list(name + "_xmin", REGION_BBOX_XMIN_KEY,
|
||||
prefix=prefix, module_dict=globals())
|
||||
msu.create_float_list_feature_list(name + "_ymax", REGION_BBOX_YMAX_KEY,
|
||||
prefix=prefix, module_dict=globals())
|
||||
msu.create_float_list_feature_list(name + "_xmax", REGION_BBOX_XMAX_KEY,
|
||||
prefix=prefix, module_dict=globals())
|
||||
msu.create_float_list_feature_list(name + "_point_x", REGION_POINT_X_KEY,
|
||||
prefix=prefix, module_dict=globals())
|
||||
msu.create_float_list_feature_list(name + "_point_y", REGION_POINT_Y_KEY,
|
||||
prefix=prefix, module_dict=globals())
|
||||
msu.create_float_list_feature_list(
|
||||
name + "_3d_point_x", REGION_3D_POINT_X_KEY,
|
||||
prefix=prefix, module_dict=globals())
|
||||
msu.create_float_list_feature_list(
|
||||
name + "_3d_point_y", REGION_3D_POINT_Y_KEY,
|
||||
prefix=prefix, module_dict=globals())
|
||||
msu.create_float_list_feature_list(
|
||||
name + "_3d_point_z", REGION_3D_POINT_Z_KEY,
|
||||
prefix=prefix, module_dict=globals())
|
||||
msu.create_bytes_list_context_feature(name + "_parts",
|
||||
REGION_PARTS_KEY,
|
||||
prefix=prefix, module_dict=globals())
|
||||
msu.create_float_list_context_feature(
|
||||
name + "_embedding_dimensions_per_region",
|
||||
REGION_EMBEDDING_DIMENSIONS_PER_REGION_KEY,
|
||||
prefix=prefix, module_dict=globals())
|
||||
msu.create_bytes_context_feature(name + "_embedding_format",
|
||||
REGION_EMBEDDING_FORMAT_KEY,
|
||||
prefix=prefix, module_dict=globals())
|
||||
msu.create_float_list_feature_list(name + "_embedding_floats",
|
||||
REGION_EMBEDDING_FLOAT_KEY,
|
||||
prefix=prefix, module_dict=globals())
|
||||
msu.create_bytes_list_feature_list(name + "_embedding_encoded",
|
||||
REGION_EMBEDDING_ENCODED_KEY,
|
||||
prefix=prefix, module_dict=globals())
|
||||
msu.create_float_list_feature_list(name + "_embedding_confidence",
|
||||
REGION_EMBEDDING_CONFIDENCE_KEY,
|
||||
prefix=prefix, module_dict=globals())
|
||||
# pylint: disable=undefined-variable
|
||||
def get_prefixed_bbox_at(index, sequence_example, prefix):
|
||||
return np.stack((
|
||||
get_bbox_ymin_at(index, sequence_example, prefix=prefix),
|
||||
get_bbox_xmin_at(index, sequence_example, prefix=prefix),
|
||||
get_bbox_ymax_at(index, sequence_example, prefix=prefix),
|
||||
get_bbox_xmax_at(index, sequence_example, prefix=prefix)),
|
||||
1)
|
||||
def add_prefixed_bbox(values, sequence_example, prefix):
|
||||
values = np.array(values)
|
||||
if values.size == 0:
|
||||
add_bbox_ymin([], sequence_example, prefix=prefix)
|
||||
add_bbox_xmin([], sequence_example, prefix=prefix)
|
||||
add_bbox_ymax([], sequence_example, prefix=prefix)
|
||||
add_bbox_xmax([], sequence_example, prefix=prefix)
|
||||
else:
|
||||
add_bbox_ymin(values[:, 0], sequence_example, prefix=prefix)
|
||||
add_bbox_xmin(values[:, 1], sequence_example, prefix=prefix)
|
||||
add_bbox_ymax(values[:, 2], sequence_example, prefix=prefix)
|
||||
add_bbox_xmax(values[:, 3], sequence_example, prefix=prefix)
|
||||
def get_prefixed_bbox_size(sequence_example, prefix):
|
||||
return get_bbox_ymin_size(sequence_example, prefix=prefix)
|
||||
def has_prefixed_bbox(sequence_example, prefix):
|
||||
return has_bbox_ymin(sequence_example, prefix=prefix)
|
||||
def clear_prefixed_bbox(sequence_example, prefix):
|
||||
clear_bbox_ymin(sequence_example, prefix=prefix)
|
||||
clear_bbox_xmin(sequence_example, prefix=prefix)
|
||||
clear_bbox_ymax(sequence_example, prefix=prefix)
|
||||
clear_bbox_xmax(sequence_example, prefix=prefix)
|
||||
def get_prefixed_point_at(index, sequence_example, prefix):
|
||||
return np.stack((
|
||||
get_bbox_point_y_at(index, sequence_example, prefix=prefix),
|
||||
get_bbox_point_x_at(index, sequence_example, prefix=prefix)),
|
||||
1)
|
||||
def add_prefixed_point(values, sequence_example, prefix):
|
||||
add_bbox_point_y(values[:, 0], sequence_example, prefix=prefix)
|
||||
add_bbox_point_x(values[:, 1], sequence_example, prefix=prefix)
|
||||
def get_prefixed_point_size(sequence_example, prefix):
|
||||
return get_bbox_point_y_size(sequence_example, prefix=prefix)
|
||||
def has_prefixed_point(sequence_example, prefix):
|
||||
return has_bbox_point_y(sequence_example, prefix=prefix)
|
||||
def clear_prefixed_point(sequence_example, prefix):
|
||||
clear_bbox_point_y(sequence_example, prefix=prefix)
|
||||
clear_bbox_point_x(sequence_example, prefix=prefix)
|
||||
def get_prefixed_3d_point_at(index, sequence_example, prefix):
|
||||
return np.stack((
|
||||
get_bbox_3d_point_x_at(index, sequence_example, prefix=prefix),
|
||||
get_bbox_3d_point_y_at(index, sequence_example, prefix=prefix),
|
||||
get_bbox_3d_point_z_at(index, sequence_example, prefix=prefix)),
|
||||
1)
|
||||
def add_prefixed_3d_point(values, sequence_example, prefix):
|
||||
add_bbox_3d_point_x(values[:, 0], sequence_example, prefix=prefix)
|
||||
add_bbox_3d_point_y(values[:, 1], sequence_example, prefix=prefix)
|
||||
add_bbox_3d_point_z(values[:, 2], sequence_example, prefix=prefix)
|
||||
def get_prefixed_3d_point_size(sequence_example, prefix):
|
||||
return get_bbox_3d_point_x_size(sequence_example, prefix=prefix)
|
||||
def has_prefixed_3d_point(sequence_example, prefix):
|
||||
return has_bbox_3d_point_x(sequence_example, prefix=prefix)
|
||||
def clear_prefixed_3d_point(sequence_example, prefix):
|
||||
clear_bbox_3d_point_x(sequence_example, prefix=prefix)
|
||||
clear_bbox_3d_point_y(sequence_example, prefix=prefix)
|
||||
clear_bbox_3d_point_z(sequence_example, prefix=prefix)
|
||||
# pylint: enable=undefined-variable
|
||||
msu.add_functions_to_module({
|
||||
"get_" + name + "_at":
|
||||
msu.function_with_default(get_prefixed_bbox_at, prefix),
|
||||
"add_" + name:
|
||||
msu.function_with_default(add_prefixed_bbox, prefix),
|
||||
"get_" + name + "_size":
|
||||
msu.function_with_default(get_prefixed_bbox_size, prefix),
|
||||
"has_" + name:
|
||||
msu.function_with_default(has_prefixed_bbox, prefix),
|
||||
"clear_" + name:
|
||||
msu.function_with_default(clear_prefixed_bbox, prefix),
|
||||
}, module_dict=globals())
|
||||
msu.add_functions_to_module({
|
||||
"get_" + name + "_point_at":
|
||||
msu.function_with_default(get_prefixed_point_at, prefix),
|
||||
"add_" + name + "_point":
|
||||
msu.function_with_default(add_prefixed_point, prefix),
|
||||
"get_" + name + "_point_size":
|
||||
msu.function_with_default(get_prefixed_point_size, prefix),
|
||||
"has_" + name + "_point":
|
||||
msu.function_with_default(has_prefixed_point, prefix),
|
||||
"clear_" + name + "_point":
|
||||
msu.function_with_default(clear_prefixed_point, prefix),
|
||||
}, module_dict=globals())
|
||||
msu.add_functions_to_module({
|
||||
"get_" + name + "_3d_point_at":
|
||||
msu.function_with_default(get_prefixed_3d_point_at, prefix),
|
||||
"add_" + name + "_3d_point":
|
||||
msu.function_with_default(add_prefixed_3d_point, prefix),
|
||||
"get_" + name + "_3d_point_size":
|
||||
msu.function_with_default(get_prefixed_3d_point_size, prefix),
|
||||
"has_" + name + "_3d_point":
|
||||
msu.function_with_default(has_prefixed_3d_point, prefix),
|
||||
"clear_" + name + "_3d_point":
|
||||
msu.function_with_default(clear_prefixed_3d_point, prefix),
|
||||
}, module_dict=globals())
|
||||
|
||||
|
||||
PREDICTED_PREFIX = "PREDICTED"
|
||||
_create_region_with_prefix("bbox", "")
|
||||
_create_region_with_prefix("predicted_bbox", PREDICTED_PREFIX)
|
||||
|
||||
|
||||
################################### IMAGES #################################
|
||||
# The format the images are encoded as (e.g. "JPEG", "PNG")
|
||||
IMAGE_FORMAT_KEY = "image/format"
|
||||
# The number of channels in the image.
|
||||
IMAGE_CHANNELS_KEY = "image/channels"
|
||||
# The colorspace of the iamge.
|
||||
IMAGE_COLORSPACE_KEY = "image/colorspace"
|
||||
# The height of the image in pixels.
|
||||
IMAGE_HEIGHT_KEY = "image/height"
|
||||
# The width of the image in pixels.
|
||||
IMAGE_WIDTH_KEY = "image/width"
|
||||
# frame rate in images/second of media.
|
||||
IMAGE_FRAME_RATE_KEY = "image/frame_rate"
|
||||
# The maximum values if the images were saturated and normalized for encoding.
|
||||
IMAGE_SATURATION_KEY = "image/saturation"
|
||||
# The listing from discrete image values (as indices) to class indices.
|
||||
IMAGE_CLASS_LABEL_INDEX_KEY = "image/class/label/index"
|
||||
# The listing from discrete image values (as indices) to class strings.
|
||||
IMAGE_CLASS_LABEL_STRING_KEY = "image/class/label/string"
|
||||
# The listing from discrete instance indices to class indices they embody.
|
||||
IMAGE_OBJECT_CLASS_INDEX_KEY = "image/object/class/index"
|
||||
# The encoded image frame.
|
||||
IMAGE_ENCODED_KEY = "image/encoded"
|
||||
# Multiple images from the same timestep (e.g. multiview video).
|
||||
IMAGE_MULTI_ENCODED_KEY = "image/multi_encoded"
|
||||
# The timestamp of the frame in microseconds.
|
||||
IMAGE_TIMESTAMP_KEY = "image/timestamp"
|
||||
# A per image label if specific frames have labels.
|
||||
# If time spans have labels, segments are preferred to allow changing rates.
|
||||
IMAGE_LABEL_INDEX_KEY = "image/label/index"
|
||||
IMAGE_LABEL_STRING_KEY = "image/label/string"
|
||||
IMAGE_LABEL_CONFIDENCE_KEY = "image/label/confidence"
|
||||
# The path of the image file if it did not come from a media clip.
|
||||
IMAGE_DATA_PATH_KEY = "image/data_path"
|
||||
|
||||
|
||||
def _create_image_with_prefix(name, prefix):
|
||||
"""Create multiple accessors for image based data."""
|
||||
msu.create_bytes_context_feature(name + "_format", IMAGE_FORMAT_KEY,
|
||||
prefix=prefix, module_dict=globals())
|
||||
msu.create_bytes_context_feature(name + "_colorspace", IMAGE_COLORSPACE_KEY,
|
||||
prefix=prefix, module_dict=globals())
|
||||
msu.create_int_context_feature(name + "_channels", IMAGE_CHANNELS_KEY,
|
||||
prefix=prefix, module_dict=globals())
|
||||
msu.create_int_context_feature(name + "_height", IMAGE_HEIGHT_KEY,
|
||||
prefix=prefix, module_dict=globals())
|
||||
msu.create_int_context_feature(name + "_width", IMAGE_WIDTH_KEY,
|
||||
prefix=prefix, module_dict=globals())
|
||||
msu.create_bytes_feature_list(name + "_encoded", IMAGE_ENCODED_KEY,
|
||||
prefix=prefix, module_dict=globals())
|
||||
msu.create_float_context_feature(name + "_frame_rate", IMAGE_FRAME_RATE_KEY,
|
||||
prefix=prefix, module_dict=globals())
|
||||
msu.create_bytes_list_context_feature(
|
||||
name + "_class_label_string", IMAGE_CLASS_LABEL_STRING_KEY,
|
||||
prefix=prefix, module_dict=globals())
|
||||
msu.create_int_list_context_feature(
|
||||
name + "_class_label_index", IMAGE_CLASS_LABEL_INDEX_KEY,
|
||||
prefix=prefix, module_dict=globals())
|
||||
msu.create_int_list_context_feature(
|
||||
name + "_object_class_index", IMAGE_OBJECT_CLASS_INDEX_KEY,
|
||||
prefix=prefix, module_dict=globals())
|
||||
msu.create_bytes_context_feature(name + "_data_path", IMAGE_DATA_PATH_KEY,
|
||||
prefix=prefix, module_dict=globals())
|
||||
msu.create_int_feature_list(name + "_timestamp", IMAGE_TIMESTAMP_KEY,
|
||||
prefix=prefix, module_dict=globals())
|
||||
msu.create_bytes_list_feature_list(name + "_multi_encoded",
|
||||
IMAGE_MULTI_ENCODED_KEY, prefix=prefix,
|
||||
module_dict=globals())
|
||||
FORWARD_FLOW_PREFIX = "FORWARD_FLOW"
|
||||
CLASS_SEGMENTATION_PREFIX = "CLASS_SEGMENTATION"
|
||||
INSTANCE_SEGMENTATION_PREFIX = "INSTANCE_SEGMENTATION"
|
||||
_create_image_with_prefix("image", "")
|
||||
_create_image_with_prefix("forward_flow", FORWARD_FLOW_PREFIX)
|
||||
_create_image_with_prefix("class_segmentation", CLASS_SEGMENTATION_PREFIX)
|
||||
_create_image_with_prefix("instance_segmentation", INSTANCE_SEGMENTATION_PREFIX)
|
||||
|
||||
################################## TEXT #################################
|
||||
# Which language text tokens are likely to be in.
|
||||
TEXT_LANGUAGE_KEY = "text/language"
|
||||
# A large block of text that applies to the media.
|
||||
TEXT_CONTEXT_CONTENT_KEY = "text/context/content"
|
||||
|
||||
# The text contents for a given time.
|
||||
TEXT_CONTENT_KEY = "text/content"
|
||||
# The start time for the text becoming relevant.
|
||||
TEXT_TIMESTAMP_KEY = "text/timestamp"
|
||||
# The duration where the text is relevant.
|
||||
TEXT_DURATION_KEY = "text/duration"
|
||||
# The confidence that this is the correct text.
|
||||
TEXT_CONFIDENCE_KEY = "text/confidence"
|
||||
# A floating point embedding corresponding to the text.
|
||||
TEXT_EMBEDDING_KEY = "text/embedding"
|
||||
# An integer id corresponding to the text.
|
||||
TEXT_TOKEN_ID_KEY = "text/token/id"
|
||||
|
||||
msu.create_bytes_context_feature(
|
||||
"text_language", TEXT_LANGUAGE_KEY, module_dict=globals())
|
||||
msu.create_bytes_context_feature(
|
||||
"text_context_content", TEXT_CONTEXT_CONTENT_KEY, module_dict=globals())
|
||||
msu.create_bytes_feature_list(
|
||||
"text_content", TEXT_CONTENT_KEY, module_dict=globals())
|
||||
msu.create_int_feature_list(
|
||||
"text_timestamp", TEXT_TIMESTAMP_KEY, module_dict=globals())
|
||||
msu.create_int_feature_list(
|
||||
"text_duration", TEXT_DURATION_KEY, module_dict=globals())
|
||||
msu.create_float_feature_list(
|
||||
"text_confidence", TEXT_CONFIDENCE_KEY, module_dict=globals())
|
||||
msu.create_float_list_feature_list(
|
||||
"text_embedding", TEXT_EMBEDDING_KEY, module_dict=globals())
|
||||
msu.create_int_feature_list(
|
||||
"text_token_id", TEXT_TOKEN_ID_KEY, module_dict=globals())
|
||||
|
||||
################################## FEATURES #################################
|
||||
# The dimensions of the feature.
|
||||
FEATURE_DIMENSIONS_KEY = "feature/dimensions"
|
||||
# The rate the features are extracted per second of media.
|
||||
FEATURE_RATE_KEY = "feature/rate"
|
||||
# The encoding format if any for the feature.
|
||||
FEATURE_BYTES_FORMAT_KEY = "feature/bytes/format"
|
||||
# For audio, the rate the samples are extracted per second of media.
|
||||
FEATURE_SAMPLE_RATE_KEY = "feature/sample_rate"
|
||||
# For audio, the number of channels per extracted feature.
|
||||
FEATURE_NUM_CHANNELS_KEY = "feature/num_channels"
|
||||
# For audio, th enumber of samples per extracted feature.
|
||||
FEATURE_NUM_SAMPLES_KEY = "feature/num_samples"
|
||||
# For audio, the rate the features are extracted per second of media.
|
||||
FEATURE_PACKET_RATE_KEY = "feature/packet_rate"
|
||||
# For audio, the original audio sampling rate the feature is derived from.
|
||||
FEATURE_AUDIO_SAMPLE_RATE_KEY = "feature/audio_sample_rate"
|
||||
# The feature as a list of floats.
|
||||
FEATURE_FLOATS_KEY = "feature/floats"
|
||||
# The feature as a list of bytes. May be encoded.
|
||||
FEATURE_BYTES_KEY = "feature/bytes"
|
||||
# The feature as a list of ints.
|
||||
FEATURE_INTS_KEY = "feature/ints"
|
||||
# The timestamp, in microseconds, of the feature.
|
||||
FEATURE_TIMESTAMP_KEY = "feature/timestamp"
|
||||
# It is occasionally useful to indicate that a feature applies to a given range.
|
||||
# This should be used for features only and annotations should be provided as
|
||||
# segments.
|
||||
FEATURE_DURATION_KEY = "feature/duration"
|
||||
# Encodes an optional confidence score for the generated features.
|
||||
FEATURE_CONFIDENCE_KEY = "feature/confidence"
|
||||
|
||||
msu.create_int_list_context_feature(
|
||||
"feature_dimensions", FEATURE_DIMENSIONS_KEY, module_dict=globals())
|
||||
msu.create_float_context_feature(
|
||||
"feature_rate", FEATURE_RATE_KEY, module_dict=globals())
|
||||
msu.create_bytes_context_feature(
|
||||
"feature_bytes_format", FEATURE_BYTES_FORMAT_KEY, module_dict=globals())
|
||||
msu.create_float_context_feature(
|
||||
"feature_sample_rate", FEATURE_SAMPLE_RATE_KEY, module_dict=globals())
|
||||
msu.create_int_context_feature(
|
||||
"feature_num_channels", FEATURE_NUM_CHANNELS_KEY, module_dict=globals())
|
||||
msu.create_int_context_feature(
|
||||
"feature_num_samples", FEATURE_NUM_SAMPLES_KEY, module_dict=globals())
|
||||
msu.create_float_context_feature(
|
||||
"feature_packet_rate", FEATURE_PACKET_RATE_KEY, module_dict=globals())
|
||||
msu.create_float_context_feature(
|
||||
"feature_audio_sample_rate", FEATURE_AUDIO_SAMPLE_RATE_KEY,
|
||||
module_dict=globals())
|
||||
msu.create_float_list_feature_list(
|
||||
"feature_floats", FEATURE_FLOATS_KEY, module_dict=globals())
|
||||
msu.create_bytes_list_feature_list(
|
||||
"feature_bytes", FEATURE_BYTES_KEY, module_dict=globals())
|
||||
msu.create_int_list_feature_list(
|
||||
"feature_ints", FEATURE_INTS_KEY, module_dict=globals())
|
||||
msu.create_int_feature_list(
|
||||
"feature_timestamp", FEATURE_TIMESTAMP_KEY, module_dict=globals())
|
||||
msu.create_int_list_feature_list(
|
||||
"feature_duration", FEATURE_DURATION_KEY, module_dict=globals())
|
||||
msu.create_float_list_feature_list(
|
||||
"feature_confidence", FEATURE_CONFIDENCE_KEY, module_dict=globals())
|
||||
|
||||
@@ -0,0 +1,261 @@
|
||||
"""Copyright 2019 The MediaPipe Authors.
|
||||
|
||||
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.
|
||||
|
||||
Tests for media_sequence.py.
|
||||
"""
|
||||
|
||||
from __future__ import absolute_import
|
||||
from __future__ import division
|
||||
from __future__ import print_function
|
||||
|
||||
import numpy as np
|
||||
import tensorflow.compat.v1 as tf
|
||||
from mediapipe.util.sequence import media_sequence as ms
|
||||
|
||||
|
||||
class MediaSequenceTest(tf.test.TestCase):
|
||||
|
||||
def test_expected_functions_are_defined(self):
|
||||
# The code from media_sequence_util is already tested, but this test ensures
|
||||
# that we actually generate the expected methods. We only test one per
|
||||
# feature and the only test is to not crash with undefined attributes. By
|
||||
# passing in a value, we also ensure that the types are correct because the
|
||||
# underlying code crashes with a type mismatch.
|
||||
example = tf.train.SequenceExample()
|
||||
# context
|
||||
ms.set_example_id(b"string", example)
|
||||
ms.set_example_dataset_name(b"string", example)
|
||||
ms.set_clip_media_id(b"string", example)
|
||||
ms.set_clip_alternative_media_id(b"string", example)
|
||||
ms.set_clip_encoded_media_bytes(b"string", example)
|
||||
ms.set_clip_encoded_media_start_timestamp(47, example)
|
||||
ms.set_clip_data_path(b"string", example)
|
||||
ms.set_clip_start_timestamp(47, example)
|
||||
ms.set_clip_end_timestamp(47, example)
|
||||
ms.set_clip_label_string((b"string", b"test"), example)
|
||||
ms.set_clip_label_index((47, 49), example)
|
||||
ms.set_clip_label_confidence((0.47, 0.49), example)
|
||||
ms.set_segment_start_timestamp((47, 49), example)
|
||||
ms.set_segment_start_index((47, 49), example)
|
||||
ms.set_segment_end_timestamp((47, 49), example)
|
||||
ms.set_segment_end_index((47, 49), example)
|
||||
ms.set_segment_label_index((47, 49), example)
|
||||
ms.set_segment_label_string((b"test", b"strings"), example)
|
||||
ms.set_segment_label_confidence((0.47, 0.49), example)
|
||||
ms.set_image_format(b"test", example)
|
||||
ms.set_image_channels(47, example)
|
||||
ms.set_image_colorspace(b"test", example)
|
||||
ms.set_image_height(47, example)
|
||||
ms.set_image_width(47, example)
|
||||
ms.set_image_frame_rate(0.47, example)
|
||||
ms.set_image_data_path(b"test", example)
|
||||
ms.set_forward_flow_format(b"test", example)
|
||||
ms.set_forward_flow_channels(47, example)
|
||||
ms.set_forward_flow_colorspace(b"test", example)
|
||||
ms.set_forward_flow_height(47, example)
|
||||
ms.set_forward_flow_width(47, example)
|
||||
ms.set_forward_flow_frame_rate(0.47, example)
|
||||
ms.set_class_segmentation_format(b"test", example)
|
||||
ms.set_class_segmentation_height(47, example)
|
||||
ms.set_class_segmentation_width(47, example)
|
||||
ms.set_class_segmentation_class_label_string((b"test", b"strings"), example)
|
||||
ms.set_class_segmentation_class_label_index((47, 49), example)
|
||||
ms.set_instance_segmentation_format(b"test", example)
|
||||
ms.set_instance_segmentation_height(47, example)
|
||||
ms.set_instance_segmentation_width(47, example)
|
||||
ms.set_instance_segmentation_object_class_index((47, 49), example)
|
||||
ms.set_bbox_parts((b"HEAD", b"TOE"), example)
|
||||
# feature lists
|
||||
ms.add_image_encoded(b"test", example)
|
||||
ms.add_image_multi_encoded([b"test", b"test"], example)
|
||||
ms.add_image_timestamp(47, example)
|
||||
ms.add_forward_flow_encoded(b"test", example)
|
||||
ms.add_forward_flow_multi_encoded([b"test", b"test"], example)
|
||||
ms.add_forward_flow_timestamp(47, example)
|
||||
ms.add_bbox_ymin((0.47, 0.49), example)
|
||||
ms.add_bbox_xmin((0.47, 0.49), example)
|
||||
ms.add_bbox_ymax((0.47, 0.49), example)
|
||||
ms.add_bbox_xmax((0.47, 0.49), example)
|
||||
ms.add_bbox_point_x((0.47, 0.49), example)
|
||||
ms.add_bbox_point_y((0.47, 0.49), example)
|
||||
ms.add_bbox_3d_point_x((0.47, 0.49), example)
|
||||
ms.add_bbox_3d_point_y((0.47, 0.49), example)
|
||||
ms.add_bbox_3d_point_z((0.47, 0.49), example)
|
||||
ms.add_predicted_bbox_ymin((0.47, 0.49), example)
|
||||
ms.add_predicted_bbox_xmin((0.47, 0.49), example)
|
||||
ms.add_predicted_bbox_ymax((0.47, 0.49), example)
|
||||
ms.add_predicted_bbox_xmax((0.47, 0.49), example)
|
||||
ms.add_bbox_num_regions(47, example)
|
||||
ms.add_bbox_is_annotated(47, example)
|
||||
ms.add_bbox_is_generated((47, 49), example)
|
||||
ms.add_bbox_is_occluded((47, 49), example)
|
||||
ms.add_bbox_label_index((47, 49), example)
|
||||
ms.add_bbox_label_string((b"test", b"strings"), example)
|
||||
ms.add_bbox_label_confidence((0.47, 0.49), example)
|
||||
ms.add_bbox_class_index((47, 49), example)
|
||||
ms.add_bbox_class_string((b"test", b"strings"), example)
|
||||
ms.add_bbox_class_confidence((0.47, 0.49), example)
|
||||
ms.add_bbox_track_index((47, 49), example)
|
||||
ms.add_bbox_track_string((b"test", b"strings"), example)
|
||||
ms.add_bbox_track_confidence((0.47, 0.49), example)
|
||||
ms.add_bbox_timestamp(47, example)
|
||||
ms.add_predicted_bbox_class_index((47, 49), example)
|
||||
ms.add_predicted_bbox_class_string((b"test", b"strings"), example)
|
||||
ms.add_predicted_bbox_timestamp(47, example)
|
||||
ms.add_class_segmentation_encoded(b"test", example)
|
||||
ms.add_class_segmentation_multi_encoded([b"test", b"test"], example)
|
||||
ms.add_instance_segmentation_encoded(b"test", example)
|
||||
ms.add_instance_segmentation_multi_encoded([b"test", b"test"], example)
|
||||
ms.add_class_segmentation_timestamp(47, example)
|
||||
ms.set_bbox_embedding_dimensions_per_region((47, 49), example)
|
||||
ms.set_bbox_embedding_format(b"test", example)
|
||||
ms.add_bbox_embedding_floats((0.47, 0.49), example)
|
||||
ms.add_bbox_embedding_encoded((b"text", b"stings"), example)
|
||||
ms.add_bbox_embedding_confidence((0.47, 0.49), example)
|
||||
ms.set_text_language(b"test", example)
|
||||
ms.set_text_context_content(b"text", example)
|
||||
ms.add_text_content(b"one", example)
|
||||
ms.add_text_timestamp(47, example)
|
||||
ms.add_text_confidence(0.47, example)
|
||||
ms.add_text_duration(47, example)
|
||||
ms.add_text_token_id(47, example)
|
||||
ms.add_text_embedding((0.47, 0.49), example)
|
||||
|
||||
def test_bbox_round_trip(self):
|
||||
example = tf.train.SequenceExample()
|
||||
boxes = np.array([[0.1, 0.2, 0.3, 0.4],
|
||||
[0.5, 0.6, 0.7, 0.8]])
|
||||
empty_boxes = np.array([])
|
||||
ms.add_bbox(boxes, example)
|
||||
ms.add_bbox(empty_boxes, example)
|
||||
self.assertEqual(2, ms.get_bbox_size(example))
|
||||
self.assertAllClose(boxes, ms.get_bbox_at(0, example))
|
||||
self.assertTrue(ms.has_bbox(example))
|
||||
ms.clear_bbox(example)
|
||||
self.assertEqual(0, ms.get_bbox_size(example))
|
||||
|
||||
def test_point_round_trip(self):
|
||||
example = tf.train.SequenceExample()
|
||||
points = np.array([[0.1, 0.2],
|
||||
[0.5, 0.6]])
|
||||
ms.add_bbox_point(points, example)
|
||||
ms.add_bbox_point(points, example)
|
||||
self.assertEqual(2, ms.get_bbox_point_size(example))
|
||||
self.assertAllClose(points, ms.get_bbox_point_at(0, example))
|
||||
self.assertTrue(ms.has_bbox_point(example))
|
||||
ms.clear_bbox_point(example)
|
||||
self.assertEqual(0, ms.get_bbox_point_size(example))
|
||||
|
||||
def test_prefixed_point_round_trip(self):
|
||||
example = tf.train.SequenceExample()
|
||||
points = np.array([[0.1, 0.2],
|
||||
[0.5, 0.6]])
|
||||
ms.add_bbox_point(points, example, "test")
|
||||
ms.add_bbox_point(points, example, "test")
|
||||
self.assertEqual(2, ms.get_bbox_point_size(example, "test"))
|
||||
self.assertAllClose(points, ms.get_bbox_point_at(0, example, "test"))
|
||||
self.assertTrue(ms.has_bbox_point(example, "test"))
|
||||
ms.clear_bbox_point(example, "test")
|
||||
self.assertEqual(0, ms.get_bbox_point_size(example, "test"))
|
||||
|
||||
def test_3d_point_round_trip(self):
|
||||
example = tf.train.SequenceExample()
|
||||
points = np.array([[0.1, 0.2, 0.3],
|
||||
[0.5, 0.6, 0.7]])
|
||||
ms.add_bbox_3d_point(points, example)
|
||||
ms.add_bbox_3d_point(points, example)
|
||||
self.assertEqual(2, ms.get_bbox_3d_point_size(example))
|
||||
self.assertAllClose(points, ms.get_bbox_3d_point_at(0, example))
|
||||
self.assertTrue(ms.has_bbox_3d_point(example))
|
||||
ms.clear_bbox_3d_point(example)
|
||||
self.assertEqual(0, ms.get_bbox_3d_point_size(example))
|
||||
|
||||
def test_predicted_bbox_round_trip(self):
|
||||
example = tf.train.SequenceExample()
|
||||
boxes = np.array([[0.1, 0.2, 0.3, 0.4],
|
||||
[0.5, 0.6, 0.7, 0.8]])
|
||||
ms.add_predicted_bbox(boxes, example)
|
||||
ms.add_predicted_bbox(boxes, example)
|
||||
self.assertEqual(2, ms.get_predicted_bbox_size(example))
|
||||
self.assertAllClose(boxes, ms.get_predicted_bbox_at(0, example))
|
||||
self.assertTrue(ms.has_predicted_bbox(example))
|
||||
ms.clear_predicted_bbox(example)
|
||||
self.assertEqual(0, ms.get_predicted_bbox_size(example))
|
||||
|
||||
def test_float_list_round_trip(self):
|
||||
example = tf.train.SequenceExample()
|
||||
values_1 = [0.1, 0.2, 0.3]
|
||||
values_2 = [0.2, 0.3, 0.4]
|
||||
ms.add_feature_floats(values_1, example, "1")
|
||||
ms.add_feature_floats(values_1, example, "1")
|
||||
ms.add_feature_floats(values_2, example, "2")
|
||||
self.assertEqual(2, ms.get_feature_floats_size(example, "1"))
|
||||
self.assertEqual(1, ms.get_feature_floats_size(example, "2"))
|
||||
self.assertTrue(ms.has_feature_floats(example, "1"))
|
||||
self.assertTrue(ms.has_feature_floats(example, "2"))
|
||||
self.assertAllClose(values_1, ms.get_feature_floats_at(0, example, "1"))
|
||||
self.assertAllClose(values_2, ms.get_feature_floats_at(0, example, "2"))
|
||||
ms.clear_feature_floats(example, "1")
|
||||
self.assertEqual(0, ms.get_feature_floats_size(example, "1"))
|
||||
self.assertFalse(ms.has_feature_floats(example, "1"))
|
||||
self.assertEqual(1, ms.get_feature_floats_size(example, "2"))
|
||||
self.assertTrue(ms.has_feature_floats(example, "2"))
|
||||
ms.clear_feature_floats(example, "2")
|
||||
self.assertEqual(0, ms.get_feature_floats_size(example, "2"))
|
||||
self.assertFalse(ms.has_feature_floats(example, "2"))
|
||||
|
||||
def test_feature_timestamp_round_trip(self):
|
||||
example = tf.train.SequenceExample()
|
||||
values_1 = 47
|
||||
values_2 = 49
|
||||
ms.add_feature_timestamp(values_1, example, "1")
|
||||
ms.add_feature_timestamp(values_1, example, "1")
|
||||
ms.add_feature_timestamp(values_2, example, "2")
|
||||
self.assertEqual(2, ms.get_feature_timestamp_size(example, "1"))
|
||||
self.assertEqual(1, ms.get_feature_timestamp_size(example, "2"))
|
||||
self.assertTrue(ms.has_feature_timestamp(example, "1"))
|
||||
self.assertTrue(ms.has_feature_timestamp(example, "2"))
|
||||
self.assertAllClose(values_1,
|
||||
ms.get_feature_timestamp_at(0, example, "1"))
|
||||
self.assertAllClose(values_2,
|
||||
ms.get_feature_timestamp_at(0, example, "2"))
|
||||
ms.clear_feature_timestamp(example, "1")
|
||||
self.assertEqual(0, ms.get_feature_timestamp_size(example, "1"))
|
||||
self.assertFalse(ms.has_feature_timestamp(example, "1"))
|
||||
self.assertEqual(1, ms.get_feature_timestamp_size(example, "2"))
|
||||
self.assertTrue(ms.has_feature_timestamp(example, "2"))
|
||||
ms.clear_feature_timestamp(example, "2")
|
||||
self.assertEqual(0, ms.get_feature_timestamp_size(example, "2"))
|
||||
self.assertFalse(ms.has_feature_timestamp(example, "2"))
|
||||
|
||||
def test_feature_dimensions_round_trip(self):
|
||||
example = tf.train.SequenceExample()
|
||||
ms.set_feature_dimensions([47, 49], example, "1")
|
||||
ms.set_feature_dimensions([49, 50], example, "2")
|
||||
self.assertSequenceEqual([47, 49],
|
||||
ms.get_feature_dimensions(example, "1"))
|
||||
self.assertSequenceEqual([49, 50],
|
||||
ms.get_feature_dimensions(example, "2"))
|
||||
self.assertTrue(ms.has_feature_dimensions(example, "1"))
|
||||
self.assertTrue(ms.has_feature_dimensions(example, "2"))
|
||||
ms.clear_feature_dimensions(example, "1")
|
||||
self.assertFalse(ms.has_feature_dimensions(example, "1"))
|
||||
self.assertTrue(ms.has_feature_dimensions(example, "2"))
|
||||
ms.clear_feature_dimensions(example, "2")
|
||||
self.assertFalse(ms.has_feature_dimensions(example, "1"))
|
||||
self.assertFalse(ms.has_feature_dimensions(example, "2"))
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
tf.test.main()
|
||||
@@ -0,0 +1,800 @@
|
||||
"""Copyright 2019 The MediaPipe Authors.
|
||||
|
||||
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.
|
||||
|
||||
Utilities for creating MediaSequence functions.
|
||||
|
||||
A set of lightweight functions to simplify access to tensorflow SequenceExample
|
||||
features and functions to create getters and setters for common types.
|
||||
"""
|
||||
|
||||
from __future__ import absolute_import
|
||||
from __future__ import division
|
||||
from __future__ import print_function
|
||||
|
||||
import types
|
||||
import tensorflow.compat.v1 as tf
|
||||
|
||||
|
||||
def function_with_default(f, default):
|
||||
"""Creates a new function with a default last parameter."""
|
||||
return types.FunctionType(f.__code__, f.__globals__, f.__name__,
|
||||
(default,), f.__closure__)
|
||||
|
||||
|
||||
def add_functions_to_module(function_dict, module_dict=None):
|
||||
"""Adds functions to another module.
|
||||
|
||||
Args:
|
||||
function_dict: a list of names and function references. The functions will
|
||||
be accessible in the new module by the name.
|
||||
module_dict: The results of globals() for the current module or
|
||||
a_module.__dict__() call. Adding values to these dicts makes the functions
|
||||
accessible from the module. If not specified, globals() is used.
|
||||
"""
|
||||
if module_dict is None:
|
||||
module_dict = globals()
|
||||
for name in function_dict:
|
||||
module_dict[name] = function_dict[name]
|
||||
|
||||
|
||||
def merge_prefix(prefix, key):
|
||||
if prefix:
|
||||
return "/".join((prefix, key))
|
||||
else:
|
||||
return key
|
||||
|
||||
|
||||
def has_context(key, sequence, prefix=""):
|
||||
return merge_prefix(prefix, key) in sequence.context.feature
|
||||
|
||||
|
||||
def clear_context(key, sequence, prefix=""):
|
||||
del sequence.context.feature[merge_prefix(prefix, key)]
|
||||
|
||||
|
||||
def set_context_float(key, value, sequence, prefix=""):
|
||||
sequence.context.feature[merge_prefix(prefix, key)].float_list.value[:] = (
|
||||
value,)
|
||||
|
||||
|
||||
def get_context_float(key, sequence, prefix=""):
|
||||
return sequence.context.feature[merge_prefix(
|
||||
prefix, key)].float_list.value[0]
|
||||
|
||||
|
||||
def set_context_bytes(key, value, sequence, prefix=""):
|
||||
sequence.context.feature[merge_prefix(
|
||||
prefix, key)].bytes_list.value[:] = (value,)
|
||||
|
||||
|
||||
def get_context_bytes(key, sequence, prefix=""):
|
||||
return sequence.context.feature[merge_prefix(prefix, key)].bytes_list.value[0]
|
||||
|
||||
|
||||
def set_context_int(key, value, sequence, prefix=""):
|
||||
sequence.context.feature[merge_prefix(
|
||||
prefix, key)].int64_list.value[:] = (value,)
|
||||
|
||||
|
||||
def get_context_int(key, sequence, prefix=""):
|
||||
return sequence.context.feature[merge_prefix(prefix, key)].int64_list.value[0]
|
||||
|
||||
|
||||
def set_context_float_list(key, value, sequence, prefix=""):
|
||||
sequence.context.feature[merge_prefix(
|
||||
prefix, key)].float_list.value[:] = value
|
||||
|
||||
|
||||
def get_context_float_list(key, sequence, prefix=""):
|
||||
return sequence.context.feature[merge_prefix(prefix, key)].float_list.value
|
||||
|
||||
|
||||
def set_context_bytes_list(key, value, sequence, prefix=""):
|
||||
sequence.context.feature[merge_prefix(
|
||||
prefix, key)].bytes_list.value[:] = value
|
||||
|
||||
|
||||
def get_context_bytes_list(key, sequence, prefix=""):
|
||||
return sequence.context.feature[merge_prefix(prefix, key)].bytes_list.value
|
||||
|
||||
|
||||
def set_context_int_list(key, value, sequence, prefix=""):
|
||||
sequence.context.feature[merge_prefix(
|
||||
prefix, key)].int64_list.value[:] = value
|
||||
|
||||
|
||||
def get_context_int_list(key, sequence, prefix=""):
|
||||
return sequence.context.feature[merge_prefix(prefix, key)].int64_list.value
|
||||
|
||||
|
||||
def has_feature_list(key, sequence, prefix=""):
|
||||
return merge_prefix(prefix, key) in sequence.feature_lists.feature_list
|
||||
|
||||
|
||||
def get_feature_list_size(key, sequence, prefix=""):
|
||||
if has_feature_list(merge_prefix(prefix, key), sequence):
|
||||
return len(sequence.feature_lists.feature_list[merge_prefix(
|
||||
prefix, key)].feature)
|
||||
else:
|
||||
return 0
|
||||
|
||||
|
||||
def clear_feature_list(key, sequence, prefix=""):
|
||||
del sequence.feature_lists.feature_list[merge_prefix(prefix, key)]
|
||||
|
||||
|
||||
def get_float_list_at(key, index, sequence, prefix=""):
|
||||
return sequence.feature_lists.feature_list[merge_prefix(prefix, key)].feature[
|
||||
index].float_list.value
|
||||
|
||||
|
||||
def get_int_list_at(key, index, sequence, prefix=""):
|
||||
return sequence.feature_lists.feature_list[merge_prefix(prefix, key)].feature[
|
||||
index].int64_list.value
|
||||
|
||||
|
||||
def get_bytes_list_at(key, index, sequence, prefix=""):
|
||||
return sequence.feature_lists.feature_list[merge_prefix(prefix, key)].feature[
|
||||
index].bytes_list.value
|
||||
|
||||
|
||||
def add_float_list(key, values, sequence, prefix=""):
|
||||
sequence.feature_lists.feature_list[merge_prefix(prefix, key)].feature.add(
|
||||
).float_list.value[:] = values
|
||||
|
||||
|
||||
def add_bytes_list(key, values, sequence, prefix=""):
|
||||
sequence.feature_lists.feature_list[merge_prefix(prefix, key)].feature.add(
|
||||
).bytes_list.value[:] = values
|
||||
|
||||
|
||||
def add_int_list(key, values, sequence, prefix=""):
|
||||
sequence.feature_lists.feature_list[merge_prefix(prefix, key)].feature.add(
|
||||
).int64_list.value[:] = values
|
||||
|
||||
|
||||
def get_float_at(key, index, sequence, prefix=""):
|
||||
return sequence.feature_lists.feature_list[merge_prefix(prefix, key)].feature[
|
||||
index].float_list.value[0]
|
||||
|
||||
|
||||
def get_int_at(key, index, sequence, prefix=""):
|
||||
return sequence.feature_lists.feature_list[merge_prefix(prefix, key)].feature[
|
||||
index].int64_list.value[0]
|
||||
|
||||
|
||||
def get_bytes_at(key, index, sequence, prefix=""):
|
||||
return sequence.feature_lists.feature_list[merge_prefix(prefix, key)].feature[
|
||||
index].bytes_list.value[0]
|
||||
|
||||
|
||||
def add_float(key, value, sequence, prefix=""):
|
||||
sequence.feature_lists.feature_list[merge_prefix(prefix, key)].feature.add(
|
||||
).float_list.value[:] = (value,)
|
||||
|
||||
|
||||
def add_bytes(key, value, sequence, prefix=""):
|
||||
sequence.feature_lists.feature_list[merge_prefix(prefix, key)].feature.add(
|
||||
).bytes_list.value[:] = (value,)
|
||||
|
||||
|
||||
def add_int(key, value, sequence, prefix=""):
|
||||
sequence.feature_lists.feature_list[merge_prefix(prefix, key)].feature.add(
|
||||
).int64_list.value[:] = (value,)
|
||||
|
||||
|
||||
def create_bytes_list_context_feature(name, key, prefix="", module_dict=None):
|
||||
"""Creates accessor functions for list of bytes features.
|
||||
|
||||
The provided functions are has_${NAME}, get_${NAME}, set_${NAME} and
|
||||
clear_${NAME}.
|
||||
|
||||
Example:
|
||||
example = tensorflow.train.SequenceExample()
|
||||
set_clip_label_string(["dog", "cat"], example)
|
||||
if has_clip_label_string(example):
|
||||
clip_label_string = get_clip_label_string(example)
|
||||
clear_clip_label_string(example)
|
||||
|
||||
Args:
|
||||
name: the name of the feature to use in function names.
|
||||
key: the key for this feature in the SequenceExample.
|
||||
prefix: a prefix to append to the key in the SequenceExample
|
||||
module_dict: adds the functions to the corresponding module dict.
|
||||
"""
|
||||
def _has(sequence_example, prefix=prefix):
|
||||
return has_context(key, sequence_example, prefix=prefix)
|
||||
|
||||
def _get(sequence_example, prefix=prefix):
|
||||
return get_context_bytes_list(key, sequence_example, prefix=prefix)
|
||||
|
||||
def _clear(sequence_example, prefix=prefix):
|
||||
clear_context(key, sequence_example, prefix=prefix)
|
||||
|
||||
def _set(value, sequence_example, prefix=prefix):
|
||||
set_context_bytes_list(key, value, sequence_example, prefix=prefix)
|
||||
|
||||
def _get_key(prefix=prefix):
|
||||
return merge_prefix(prefix, key)
|
||||
|
||||
def _get_default_parser():
|
||||
return tf.io.VarLenFeature(tf.string)
|
||||
|
||||
function_dict = {
|
||||
"has_" + name: _has,
|
||||
"get_" + name: _get,
|
||||
"clear_" + name: _clear,
|
||||
"set_" + name: _set,
|
||||
"get_" + name + "_key": _get_key,
|
||||
"get_" + name + "_default_parser": _get_default_parser,
|
||||
}
|
||||
add_functions_to_module(function_dict, module_dict)
|
||||
|
||||
|
||||
def create_float_list_context_feature(name, key, prefix="", module_dict=None):
|
||||
"""Creates accessor functions for list of floats features.
|
||||
|
||||
The provided functions are has_${NAME}, get_${NAME}, set_${NAME} and
|
||||
clear_${NAME}.
|
||||
|
||||
Example:
|
||||
example = tensorflow.train.SequenceExample()
|
||||
set_segment_label_confidence([0.47, 0.49], example)
|
||||
if has_segment_label_confidence(example):
|
||||
confidences = get_segment_label_confidence(example)
|
||||
clear_segment_label_confidence(example)
|
||||
|
||||
Args:
|
||||
name: the name of the feature to use in function names.
|
||||
key: the key for this feature in the SequenceExample.
|
||||
prefix: a prefix to append to the key in the SequenceExample
|
||||
module_dict: adds the functions to the corresponding module dict.
|
||||
"""
|
||||
def _has(sequence_example, prefix=prefix):
|
||||
return has_context(key, sequence_example, prefix=prefix)
|
||||
|
||||
def _get(sequence_example, prefix=prefix):
|
||||
return get_context_float_list(key, sequence_example, prefix=prefix)
|
||||
|
||||
def _clear(sequence_example, prefix=prefix):
|
||||
clear_context(key, sequence_example, prefix=prefix)
|
||||
|
||||
def _set(value, sequence_example, prefix=prefix):
|
||||
set_context_float_list(key, value, sequence_example, prefix=prefix)
|
||||
|
||||
def _get_key(prefix=prefix):
|
||||
return merge_prefix(prefix, key)
|
||||
|
||||
def _get_default_parser():
|
||||
return tf.io.VarLenFeature(tf.float32)
|
||||
|
||||
function_dict = {
|
||||
"has_" + name: _has,
|
||||
"get_" + name: _get,
|
||||
"clear_" + name: _clear,
|
||||
"set_" + name: _set,
|
||||
"get_" + name + "_key": _get_key,
|
||||
"get_" + name + "_default_parser": _get_default_parser,
|
||||
}
|
||||
add_functions_to_module(function_dict, module_dict)
|
||||
|
||||
|
||||
def create_int_list_context_feature(name, key, prefix="", module_dict=None):
|
||||
"""Creates accessor functions for list of int64 features.
|
||||
|
||||
The provided functions are has_${NAME}, get_${NAME}, set_${NAME} and
|
||||
clear_${NAME}.
|
||||
|
||||
Example:
|
||||
example = tensorflow.train.SequenceExample()
|
||||
set_clip_label_index([0, 1, 2, 3], example)
|
||||
if has_clip_label_index(example):
|
||||
clip_label_index = get_clip_label_index(example)
|
||||
clear_clip_label_index
|
||||
|
||||
Args:
|
||||
name: the name of the feature to use in function names.
|
||||
key: the key for this feature in the SequenceExample.
|
||||
prefix: a prefix to append to the key in the SequenceExample
|
||||
module_dict: adds the functions to the corresponding module dict.
|
||||
"""
|
||||
def _has(sequence_example, prefix=prefix):
|
||||
return has_context(key, sequence_example, prefix=prefix)
|
||||
|
||||
def _get(sequence_example, prefix=prefix):
|
||||
return get_context_int_list(key, sequence_example, prefix=prefix)
|
||||
|
||||
def _clear(sequence_example, prefix=prefix):
|
||||
clear_context(key, sequence_example, prefix=prefix)
|
||||
|
||||
def _set(value, sequence_example, prefix=prefix):
|
||||
set_context_int_list(key, value, sequence_example, prefix=prefix)
|
||||
|
||||
def _get_key(prefix=prefix):
|
||||
return merge_prefix(prefix, key)
|
||||
|
||||
def _get_default_parser():
|
||||
return tf.io.VarLenFeature(tf.int64)
|
||||
|
||||
function_dict = {
|
||||
"has_" + name: _has,
|
||||
"get_" + name: _get,
|
||||
"clear_" + name: _clear,
|
||||
"set_" + name: _set,
|
||||
"get_" + name + "_key": _get_key,
|
||||
"get_" + name + "_default_parser": _get_default_parser,
|
||||
}
|
||||
add_functions_to_module(function_dict, module_dict)
|
||||
|
||||
|
||||
def create_bytes_context_feature(name, key, prefix="", module_dict=None):
|
||||
"""Creates accessor functions for single bytes features.
|
||||
|
||||
The provided functions are has_${NAME}, get_${NAME}, set_${NAME} and
|
||||
clear_${NAME}.
|
||||
|
||||
Example:
|
||||
example = tensorflow.train.SequenceExample()
|
||||
set_data_path("path_to_a_file", example)
|
||||
if has_data_path(example):
|
||||
path = get_data_path(example)
|
||||
clear_data_path(example)
|
||||
|
||||
Args:
|
||||
name: the name of the feature to use in function names.
|
||||
key: the key for this feature in the SequenceExample.
|
||||
prefix: a prefix to append to the key in the SequenceExample
|
||||
module_dict: adds the functions to the corresponding module dict.
|
||||
"""
|
||||
def _has(sequence_example, prefix=prefix):
|
||||
return has_context(key, sequence_example, prefix=prefix)
|
||||
|
||||
def _get(sequence_example, prefix=prefix):
|
||||
return get_context_bytes(key, sequence_example, prefix=prefix)
|
||||
|
||||
def _clear(sequence_example, prefix=prefix):
|
||||
clear_context(key, sequence_example, prefix=prefix)
|
||||
|
||||
def _set(value, sequence_example, prefix=prefix):
|
||||
set_context_bytes(key, value, sequence_example, prefix=prefix)
|
||||
|
||||
def _get_key(prefix=prefix):
|
||||
return merge_prefix(prefix, key)
|
||||
|
||||
def _get_default_parser():
|
||||
return tf.io.FixedLenFeature((), tf.string)
|
||||
|
||||
function_dict = {
|
||||
"has_" + name: _has,
|
||||
"get_" + name: _get,
|
||||
"clear_" + name: _clear,
|
||||
"set_" + name: _set,
|
||||
"get_" + name + "_key": _get_key,
|
||||
"get_" + name + "_default_parser": _get_default_parser,
|
||||
}
|
||||
add_functions_to_module(function_dict, module_dict)
|
||||
|
||||
|
||||
def create_float_context_feature(name, key, prefix="", module_dict=None):
|
||||
"""Creates accessor functions for single float features.
|
||||
|
||||
The provided functions are has_${NAME}, get_${NAME}, set_${NAME} and
|
||||
clear_${NAME}.
|
||||
|
||||
Example:
|
||||
example = tensorflow.train.SequenceExample()
|
||||
set_image_frame_rate(0.47, example)
|
||||
if has_image_frame_rate(example):
|
||||
image_frame_rate = get_image_frame_rate(example)
|
||||
clear_image_frame_rate(example)
|
||||
|
||||
Args:
|
||||
name: the name of the feature to use in function names.
|
||||
key: the key for this feature in the SequenceExample.
|
||||
prefix: a prefix to append to the key in the SequenceExample
|
||||
module_dict: adds the functions to the corresponding module dict.
|
||||
"""
|
||||
def _has(sequence_example, prefix=prefix):
|
||||
return has_context(key, sequence_example, prefix=prefix)
|
||||
|
||||
def _get(sequence_example, prefix=prefix):
|
||||
return get_context_float(key, sequence_example, prefix=prefix)
|
||||
|
||||
def _clear(sequence_example, prefix=prefix):
|
||||
clear_context(key, sequence_example, prefix=prefix)
|
||||
|
||||
def _set(value, sequence_example, prefix=prefix):
|
||||
set_context_float(key, value, sequence_example, prefix=prefix)
|
||||
|
||||
def _get_key(prefix=prefix):
|
||||
return merge_prefix(prefix, key)
|
||||
|
||||
def _get_default_parser():
|
||||
return tf.io.FixedLenFeature((), tf.float32)
|
||||
|
||||
function_dict = {
|
||||
"has_" + name: _has,
|
||||
"get_" + name: _get,
|
||||
"clear_" + name: _clear,
|
||||
"set_" + name: _set,
|
||||
"get_" + name + "_key": _get_key,
|
||||
"get_" + name + "_default_parser": _get_default_parser,
|
||||
}
|
||||
add_functions_to_module(function_dict, module_dict)
|
||||
|
||||
|
||||
def create_int_context_feature(name, key, prefix="", module_dict=None):
|
||||
"""Creates accessor functions for single int64 features.
|
||||
|
||||
The provided functions are has_${NAME}, get_${NAME}, set_${NAME} and
|
||||
clear_${NAME}.
|
||||
|
||||
Example:
|
||||
example = tensorflow.train.SequenceExample()
|
||||
set_clip_frames(47, example)
|
||||
if has_clip_frames(example):
|
||||
clip_frames = get_clip_frames(example)
|
||||
clear_clip_frames(example)
|
||||
|
||||
Args:
|
||||
name: the name of the feature to use in function names.
|
||||
key: the key for this feature in the SequenceExample.
|
||||
prefix: a prefix to append to the key in the SequenceExample
|
||||
module_dict: adds the functions to the corresponding module dict.
|
||||
"""
|
||||
def _has(sequence_example, prefix=prefix):
|
||||
return has_context(key, sequence_example, prefix=prefix)
|
||||
|
||||
def _get(sequence_example, prefix=prefix):
|
||||
return get_context_int(key, sequence_example, prefix=prefix)
|
||||
|
||||
def _clear(sequence_example, prefix=prefix):
|
||||
clear_context(key, sequence_example, prefix=prefix)
|
||||
|
||||
def _set(value, sequence_example, prefix=prefix):
|
||||
set_context_int(key, value, sequence_example, prefix=prefix)
|
||||
|
||||
def _get_key(prefix=prefix):
|
||||
return merge_prefix(prefix, key)
|
||||
|
||||
def _get_default_parser():
|
||||
return tf.io.FixedLenFeature((), tf.int64)
|
||||
|
||||
function_dict = {
|
||||
"has_" + name: _has,
|
||||
"get_" + name: _get,
|
||||
"clear_" + name: _clear,
|
||||
"set_" + name: _set,
|
||||
"get_" + name + "_key": _get_key,
|
||||
"get_" + name + "_default_parser": _get_default_parser,
|
||||
}
|
||||
add_functions_to_module(function_dict, module_dict)
|
||||
|
||||
|
||||
def create_bytes_feature_list(name, key, prefix="", module_dict=None):
|
||||
"""Creates accessor functions for bytes feature lists.
|
||||
|
||||
The provided functions are has_${NAME}, get_${NAME}_size, get_${NAME}_at,
|
||||
clear_${NAME}, and add_${NAME}.
|
||||
|
||||
example = tensorflow.train.SequenceExample()
|
||||
add_image_encoded(1000000, example)
|
||||
add_image_encoded(2000000, example)
|
||||
if has_image_encoded:
|
||||
for i in range(get_image_encoded_size(example):
|
||||
image_encoded = get_image_encoded_at(i, example)
|
||||
clear_image_encoded(example)
|
||||
|
||||
Args:
|
||||
name: the name of the feature to use in function names.
|
||||
key: the key for this feature in the SequenceExample.
|
||||
prefix: a prefix to append to the key in the SequenceExample
|
||||
module_dict: adds the functions to the corresponding module dict.
|
||||
"""
|
||||
def _has(sequence_example, prefix=prefix):
|
||||
return has_feature_list(key, sequence_example, prefix=prefix)
|
||||
|
||||
def _get_size(sequence_example, prefix=prefix):
|
||||
return get_feature_list_size(key, sequence_example, prefix=prefix)
|
||||
|
||||
def _get_at(index, sequence_example, prefix=prefix):
|
||||
return get_bytes_at(key, index, sequence_example, prefix=prefix)
|
||||
|
||||
def _clear(sequence_example, prefix=prefix):
|
||||
clear_feature_list(key, sequence_example, prefix=prefix)
|
||||
|
||||
def _add(value, sequence_example, prefix=prefix):
|
||||
add_bytes(key, value, sequence_example, prefix=prefix)
|
||||
|
||||
def _get_key(prefix=prefix):
|
||||
return merge_prefix(prefix, key)
|
||||
|
||||
def _get_default_parser():
|
||||
return tf.io.FixedLenSequenceFeature((), tf.string)
|
||||
|
||||
function_dict = {
|
||||
"has_" + name: _has,
|
||||
"get_" + name + "_size": _get_size,
|
||||
"get_" + name + "_at": _get_at,
|
||||
"clear_" + name: _clear,
|
||||
"add_" + name: _add,
|
||||
"get_" + name + "_key": _get_key,
|
||||
"get_" + name + "_default_parser": _get_default_parser,
|
||||
}
|
||||
add_functions_to_module(function_dict, module_dict)
|
||||
|
||||
|
||||
def create_float_feature_list(name, key, prefix="", module_dict=None):
|
||||
"""Creates accessor functions for float feature lists.
|
||||
|
||||
The provided functions are has_${NAME}, get_${NAME}_size, get_${NAME}_at,
|
||||
clear_${NAME}, and add_${NAME}.
|
||||
|
||||
example = tensorflow.train.SequenceExample()
|
||||
add_confidence(0.47, example)
|
||||
add_confidence(0.49, example)
|
||||
if has_confidence:
|
||||
for i in range(get_confidence_size(example):
|
||||
confidence = get_confidence_at(i, example)
|
||||
clear_confidence(example)
|
||||
|
||||
Args:
|
||||
name: the name of the feature to use in function names.
|
||||
key: the key for this feature in the SequenceExample.
|
||||
prefix: a prefix to append to the key in the SequenceExample
|
||||
module_dict: adds the functions to the corresponding module dict.
|
||||
"""
|
||||
def _has(sequence_example, prefix=prefix):
|
||||
return has_feature_list(key, sequence_example, prefix=prefix)
|
||||
|
||||
def _get_size(sequence_example, prefix=prefix):
|
||||
return get_feature_list_size(key, sequence_example, prefix=prefix)
|
||||
|
||||
def _get_at(index, sequence_example, prefix=prefix):
|
||||
return get_float_at(key, index, sequence_example, prefix=prefix)
|
||||
|
||||
def _clear(sequence_example, prefix=prefix):
|
||||
clear_feature_list(key, sequence_example, prefix=prefix)
|
||||
|
||||
def _add(value, sequence_example, prefix=prefix):
|
||||
add_float(key, value, sequence_example, prefix=prefix)
|
||||
|
||||
def _get_key(prefix=prefix):
|
||||
return merge_prefix(prefix, key)
|
||||
|
||||
def _get_default_parser():
|
||||
return tf.io.FixedLenSequenceFeature((), tf.float32)
|
||||
|
||||
function_dict = {
|
||||
"has_" + name: _has,
|
||||
"get_" + name + "_size": _get_size,
|
||||
"get_" + name + "_at": _get_at,
|
||||
"clear_" + name: _clear,
|
||||
"add_" + name: _add,
|
||||
"get_" + name + "_key": _get_key,
|
||||
"get_" + name + "_default_parser": _get_default_parser,
|
||||
}
|
||||
add_functions_to_module(function_dict, module_dict)
|
||||
|
||||
|
||||
def create_int_feature_list(name, key, prefix="", module_dict=None):
|
||||
"""Creates accessor functions for bytes feature lists.
|
||||
|
||||
The provided functions are has_${NAME}, get_${NAME}_size, get_${NAME}_at,
|
||||
clear_${NAME}, and add_${NAME}.
|
||||
|
||||
example = tensorflow.train.SequenceExample()
|
||||
add_image_timestamp(1000000, example)
|
||||
add_image_timestamp(2000000, example)
|
||||
if has_image_timestamp:
|
||||
for i in range(get_image_timestamp_size(example):
|
||||
timestamp = get_image_timestamp_at(i, example)
|
||||
clear_image_timestamp(example)
|
||||
|
||||
Args:
|
||||
name: the name of the feature to use in function names.
|
||||
key: the key for this feature in the SequenceExample.
|
||||
prefix: a prefix to append to the key in the SequenceExample
|
||||
module_dict: adds the functions to the corresponding module dict.
|
||||
"""
|
||||
def _has(sequence_example, prefix=prefix):
|
||||
return has_feature_list(key, sequence_example, prefix=prefix)
|
||||
|
||||
def _get_size(sequence_example, prefix=prefix):
|
||||
return get_feature_list_size(key, sequence_example, prefix=prefix)
|
||||
|
||||
def _get_at(index, sequence_example, prefix=prefix):
|
||||
return get_int_at(key, index, sequence_example, prefix=prefix)
|
||||
|
||||
def _clear(sequence_example, prefix=prefix):
|
||||
clear_feature_list(key, sequence_example, prefix=prefix)
|
||||
|
||||
def _add(value, sequence_example, prefix=prefix):
|
||||
add_int(key, value, sequence_example, prefix=prefix)
|
||||
|
||||
def _get_key(prefix=prefix):
|
||||
return merge_prefix(prefix, key)
|
||||
|
||||
def _get_default_parser():
|
||||
return tf.io.FixedLenSequenceFeature((), tf.int64)
|
||||
|
||||
function_dict = {
|
||||
"has_" + name: _has,
|
||||
"get_" + name + "_size": _get_size,
|
||||
"get_" + name + "_at": _get_at,
|
||||
"clear_" + name: _clear,
|
||||
"add_" + name: _add,
|
||||
"get_" + name + "_key": _get_key,
|
||||
"get_" + name + "_default_parser": _get_default_parser,
|
||||
}
|
||||
add_functions_to_module(function_dict, module_dict)
|
||||
|
||||
|
||||
def create_bytes_list_feature_list(name, key, prefix="", module_dict=None):
|
||||
"""Creates accessor functions for list of bytes feature lists.
|
||||
|
||||
The provided functions are has_${NAME}, get_${NAME}_size, get_${NAME}_at,
|
||||
clear_${NAME}, and add_${NAME}.
|
||||
|
||||
example = tensorflow.train.SequenceExample()
|
||||
add_bbox_label_string(["dog", "cat"], example)
|
||||
add_bbox_label_string(["cat", "dog"], example)
|
||||
if has_bbox_label_string:
|
||||
for i in range(get_bbox_label_string_size(example):
|
||||
timestamp = get_bbox_label_string_at(i, example)
|
||||
clear_bbox_label_string(example)
|
||||
|
||||
Args:
|
||||
name: the name of the feature to use in function names.
|
||||
key: the key for this feature in the SequenceExample.
|
||||
prefix: a prefix to append to the key in the SequenceExample
|
||||
module_dict: adds the functions to the corresponding module dict.
|
||||
"""
|
||||
def _has(sequence_example, prefix=prefix):
|
||||
return has_feature_list(key, sequence_example, prefix=prefix)
|
||||
|
||||
def _get_size(sequence_example, prefix=prefix):
|
||||
return get_feature_list_size(key, sequence_example, prefix=prefix)
|
||||
|
||||
def _get_at(index, sequence_example, prefix=prefix):
|
||||
return get_bytes_list_at(key, index, sequence_example, prefix=prefix)
|
||||
|
||||
def _clear(sequence_example, prefix=prefix):
|
||||
clear_feature_list(key, sequence_example, prefix=prefix)
|
||||
|
||||
def _add(value, sequence_example, prefix=prefix):
|
||||
add_bytes_list(key, value, sequence_example, prefix=prefix)
|
||||
|
||||
def _get_key(prefix=prefix):
|
||||
return merge_prefix(prefix, key)
|
||||
|
||||
def _get_default_parser():
|
||||
return tf.io.VarLenFeature(tf.string)
|
||||
|
||||
function_dict = {
|
||||
"has_" + name: _has,
|
||||
"get_" + name + "_size": _get_size,
|
||||
"get_" + name + "_at": _get_at,
|
||||
"clear_" + name: _clear,
|
||||
"add_" + name: _add,
|
||||
"get_" + name + "_key": _get_key,
|
||||
"get_" + name + "_default_parser": _get_default_parser,
|
||||
}
|
||||
add_functions_to_module(function_dict, module_dict)
|
||||
|
||||
|
||||
def create_float_list_feature_list(name, key, prefix="", module_dict=None):
|
||||
"""Creates accessor functions for list of float feature lists.
|
||||
|
||||
The provided functions are has_${NAME}, get_${NAME}_size, get_${NAME}_at,
|
||||
clear_${NAME}, and add_${NAME}.
|
||||
|
||||
example = tensorflow.train.SequenceExample()
|
||||
add_bbox_ymin([0.47, 0.49], example)
|
||||
add_bbox_ymin([0.49, 0.47], example)
|
||||
if has_bbox_ymin:
|
||||
for i in range(get_bbox_ymin_size(example):
|
||||
bbox_ymin = get_bbox_ymin_at(i, example)
|
||||
clear_bbox_ymin(example)
|
||||
|
||||
Args:
|
||||
name: the name of the feature to use in function names.
|
||||
key: the key for this feature in the SequenceExample.
|
||||
prefix: a prefix to append to the key in the SequenceExample
|
||||
module_dict: adds the functions to the corresponding module dict.
|
||||
"""
|
||||
def _has(sequence_example, prefix=prefix):
|
||||
return has_feature_list(key, sequence_example, prefix=prefix)
|
||||
|
||||
def _get_size(sequence_example, prefix=prefix):
|
||||
return get_feature_list_size(key, sequence_example, prefix=prefix)
|
||||
|
||||
def _get_at(index, sequence_example, prefix=prefix):
|
||||
return get_float_list_at(key, index, sequence_example, prefix=prefix)
|
||||
|
||||
def _clear(sequence_example, prefix=prefix):
|
||||
clear_feature_list(key, sequence_example, prefix=prefix)
|
||||
|
||||
def _add(value, sequence_example, prefix=prefix):
|
||||
add_float_list(key, value, sequence_example, prefix=prefix)
|
||||
|
||||
def _get_key(prefix=prefix):
|
||||
return merge_prefix(prefix, key)
|
||||
|
||||
def _get_default_parser():
|
||||
return tf.io.VarLenFeature(tf.float32)
|
||||
|
||||
function_dict = {
|
||||
"has_" + name: _has,
|
||||
"get_" + name + "_size": _get_size,
|
||||
"get_" + name + "_at": _get_at,
|
||||
"clear_" + name: _clear,
|
||||
"add_" + name: _add,
|
||||
"get_" + name + "_key": _get_key,
|
||||
"get_" + name + "_default_parser": _get_default_parser,
|
||||
}
|
||||
add_functions_to_module(function_dict, module_dict)
|
||||
|
||||
|
||||
def create_int_list_feature_list(name, key, prefix="", module_dict=None):
|
||||
"""Creates accessor functions for list of int64 feature lists.
|
||||
|
||||
The provided functions are has_${NAME}, get_${NAME}_size, get_${NAME}_at,
|
||||
clear_${NAME}, and add_${NAME}.
|
||||
|
||||
example = tensorflow.train.SequenceExample()
|
||||
add_bbox_label_index([47, 49], example)
|
||||
add_bbox_label_index([49, 47], example)
|
||||
if has_bbox_label_index:
|
||||
for i in range(get_bbox_label_index_size(example):
|
||||
bbox_label_index = get_bbox_label_index_at(i, example)
|
||||
clear_bbox_label_index(example)
|
||||
|
||||
Args:
|
||||
name: the name of the feature to use in function names.
|
||||
key: the key for this feature in the SequenceExample.
|
||||
prefix: a prefix to append to the key in the SequenceExample
|
||||
module_dict: adds the functions to the corresponding module dict.
|
||||
"""
|
||||
def _has(sequence_example, prefix=prefix):
|
||||
return has_feature_list(key, sequence_example, prefix=prefix)
|
||||
|
||||
def _get_size(sequence_example, prefix=prefix):
|
||||
return get_feature_list_size(key, sequence_example, prefix=prefix)
|
||||
|
||||
def _get_at(index, sequence_example, prefix=prefix):
|
||||
return get_int_list_at(key, index, sequence_example, prefix=prefix)
|
||||
|
||||
def _clear(sequence_example, prefix=prefix):
|
||||
clear_feature_list(key, sequence_example, prefix=prefix)
|
||||
|
||||
def _add(value, sequence_example, prefix=prefix):
|
||||
add_int_list(key, value, sequence_example, prefix=prefix)
|
||||
|
||||
def _get_key(prefix=prefix):
|
||||
return merge_prefix(prefix, key)
|
||||
|
||||
def _get_default_parser():
|
||||
return tf.io.VarLenFeature(tf.int64)
|
||||
|
||||
function_dict = {
|
||||
"has_" + name: _has,
|
||||
"get_" + name + "_size": _get_size,
|
||||
"get_" + name + "_at": _get_at,
|
||||
"clear_" + name: _clear,
|
||||
"add_" + name: _add,
|
||||
"get_" + name + "_key": _get_key,
|
||||
"get_" + name + "_default_parser": _get_default_parser,
|
||||
}
|
||||
add_functions_to_module(function_dict, module_dict)
|
||||
@@ -0,0 +1,389 @@
|
||||
"""Copyright 2019 The MediaPipe Authors.
|
||||
|
||||
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.
|
||||
|
||||
Tests for media_sequence_util.py.
|
||||
"""
|
||||
|
||||
from __future__ import absolute_import
|
||||
from __future__ import division
|
||||
from __future__ import print_function
|
||||
|
||||
import tensorflow.compat.v1 as tf
|
||||
|
||||
from mediapipe.util.sequence import media_sequence_util as msu
|
||||
|
||||
# run this code at the module level to procedurally define functions for test
|
||||
# cases.
|
||||
msu.create_bytes_context_feature(
|
||||
"string_context", "string_feature", module_dict=msu.__dict__)
|
||||
msu.create_float_context_feature(
|
||||
"float_context", "float_feature", module_dict=msu.__dict__)
|
||||
msu.create_int_context_feature(
|
||||
"int64_context", "int64_feature", module_dict=msu.__dict__)
|
||||
msu.create_bytes_list_context_feature(
|
||||
"string_list_context", "string_vector_feature", module_dict=msu.__dict__)
|
||||
msu.create_float_list_context_feature(
|
||||
"float_list_context", "float_vector_feature", module_dict=msu.__dict__)
|
||||
msu.create_int_list_context_feature(
|
||||
"int64_list_context", "int64_vector_feature", module_dict=msu.__dict__)
|
||||
msu.create_bytes_feature_list(
|
||||
"string_feature_list", "string_feature_list", module_dict=msu.__dict__)
|
||||
msu.create_float_feature_list(
|
||||
"float_feature_list", "float_feature_list", module_dict=msu.__dict__)
|
||||
msu.create_int_feature_list(
|
||||
"int64_feature_list", "int64_feature_list", module_dict=msu.__dict__)
|
||||
msu.create_bytes_list_feature_list(
|
||||
"string_list_feature_list", "string_list_feature_list",
|
||||
module_dict=msu.__dict__)
|
||||
msu.create_float_list_feature_list(
|
||||
"float_list_feature_list", "float_list_feature_list",
|
||||
module_dict=msu.__dict__)
|
||||
msu.create_int_list_feature_list(
|
||||
"int64_list_feature_list", "int64_list_feature_list",
|
||||
module_dict=msu.__dict__)
|
||||
|
||||
|
||||
class MediaSequenceTest(tf.test.TestCase):
|
||||
|
||||
def test_set_context_float(self):
|
||||
example = tf.train.SequenceExample()
|
||||
key = "test_float"
|
||||
msu.set_context_float(key, 0.1, example)
|
||||
self.assertAlmostEqual(0.1,
|
||||
example.context.feature[key].float_list.value[0])
|
||||
|
||||
def test_set_context_bytes(self):
|
||||
example = tf.train.SequenceExample()
|
||||
key = "test_string"
|
||||
msu.set_context_bytes(key, b"test", example)
|
||||
self.assertEqual(b"test",
|
||||
example.context.feature[key].bytes_list.value[0])
|
||||
|
||||
def test_set_context_int(self):
|
||||
example = tf.train.SequenceExample()
|
||||
key = "test_int"
|
||||
msu.set_context_int(key, 47, example)
|
||||
self.assertEqual(47,
|
||||
example.context.feature[key].int64_list.value[0])
|
||||
|
||||
def test_set_context_float_list(self):
|
||||
example = tf.train.SequenceExample()
|
||||
key = "test_float_list"
|
||||
msu.set_context_float_list(key, [0.0, 0.1], example)
|
||||
self.assertSequenceAlmostEqual(
|
||||
[0.0, 0.1], example.context.feature[key].float_list.value)
|
||||
|
||||
def test_set_context_bytes_list(self):
|
||||
example = tf.train.SequenceExample()
|
||||
key = "test_string_list"
|
||||
msu.set_context_bytes_list(key, [b"test", b"test"], example)
|
||||
self.assertSequenceAlmostEqual(
|
||||
[b"test", b"test"], example.context.feature[key].bytes_list.value)
|
||||
|
||||
def test_set_context_int_list(self):
|
||||
example = tf.train.SequenceExample()
|
||||
key = "test_int_list"
|
||||
msu.set_context_int_list(key, [0, 1], example)
|
||||
self.assertSequenceAlmostEqual(
|
||||
[0, 1], example.context.feature[key].int64_list.value)
|
||||
|
||||
def test_round_trip_float_list_feature(self):
|
||||
example = tf.train.SequenceExample()
|
||||
key = "test_float_features"
|
||||
msu.add_float_list(key, [0.0, 0.1], example)
|
||||
msu.add_float_list(key, [0.1, 0.2], example)
|
||||
self.assertEqual(2, msu.get_feature_list_size(key, example))
|
||||
self.assertTrue(msu.has_feature_list(key, example))
|
||||
self.assertAllClose([0.0, 0.1], msu.get_float_list_at(key, 0, example))
|
||||
self.assertAllClose([0.1, 0.2], msu.get_float_list_at(key, 1, example))
|
||||
msu.clear_feature_list(key, example)
|
||||
self.assertEqual(0, msu.get_feature_list_size(key, example))
|
||||
self.assertFalse(msu.has_feature_list(key, example))
|
||||
|
||||
def test_round_trip_bytes_list_feature(self):
|
||||
example = tf.train.SequenceExample()
|
||||
key = "test_bytes_features"
|
||||
msu.add_bytes_list(key, [b"test0", b"test1"], example)
|
||||
msu.add_bytes_list(key, [b"test1", b"test2"], example)
|
||||
self.assertEqual(2, msu.get_feature_list_size(key, example))
|
||||
self.assertTrue(msu.has_feature_list(key, example))
|
||||
self.assertAllEqual([b"test0", b"test1"],
|
||||
msu.get_bytes_list_at(key, 0, example))
|
||||
self.assertAllEqual([b"test1", b"test2"],
|
||||
msu.get_bytes_list_at(key, 1, example))
|
||||
msu.clear_feature_list(key, example)
|
||||
self.assertEqual(0, msu.get_feature_list_size(key, example))
|
||||
self.assertFalse(msu.has_feature_list(key, example))
|
||||
|
||||
def test_round_trip_int_list_feature(self):
|
||||
example = tf.train.SequenceExample()
|
||||
key = "test_ints_features"
|
||||
msu.add_int_list(key, [0, 1], example)
|
||||
msu.add_int_list(key, [1, 2], example)
|
||||
self.assertEqual(2, msu.get_feature_list_size(key, example))
|
||||
self.assertTrue(msu.has_feature_list(key, example))
|
||||
self.assertAllClose([0, 1], msu.get_int_list_at(key, 0, example))
|
||||
self.assertAllClose([1, 2], msu.get_int_list_at(key, 1, example))
|
||||
msu.clear_feature_list(key, example)
|
||||
self.assertEqual(0, msu.get_feature_list_size(key, example))
|
||||
self.assertFalse(msu.has_feature_list(key, example))
|
||||
|
||||
def test_round_trip_string_context(self):
|
||||
example = tf.train.SequenceExample()
|
||||
key = "string_feature"
|
||||
msu.set_string_context(b"test", example)
|
||||
self.assertEqual(b"test", msu.get_string_context(example))
|
||||
self.assertTrue(msu.has_string_context(example))
|
||||
self.assertEqual(b"test",
|
||||
example.context.feature[key].bytes_list.value[0])
|
||||
msu.clear_string_context(example)
|
||||
self.assertFalse(msu.has_string_context(example))
|
||||
self.assertEqual("string_feature", msu.get_string_context_key())
|
||||
|
||||
def test_round_trip_float_context(self):
|
||||
example = tf.train.SequenceExample()
|
||||
key = "float_feature"
|
||||
msu.set_float_context(0.47, example)
|
||||
self.assertAlmostEqual(0.47, msu.get_float_context(example))
|
||||
self.assertTrue(msu.has_float_context(example))
|
||||
self.assertAlmostEqual(0.47,
|
||||
example.context.feature[key].float_list.value[0])
|
||||
msu.clear_float_context(example)
|
||||
self.assertFalse(msu.has_float_context(example))
|
||||
self.assertEqual("float_feature", msu.get_float_context_key())
|
||||
|
||||
def test_round_trip_int_context(self):
|
||||
example = tf.train.SequenceExample()
|
||||
key = "int64_feature"
|
||||
msu.set_int64_context(47, example)
|
||||
self.assertEqual(47, msu.get_int64_context(example))
|
||||
self.assertTrue(msu.has_int64_context(example))
|
||||
self.assertEqual(47,
|
||||
example.context.feature[key].int64_list.value[0])
|
||||
msu.clear_int64_context(example)
|
||||
self.assertFalse(msu.has_int64_context(example))
|
||||
self.assertEqual("int64_feature", msu.get_int64_context_key())
|
||||
|
||||
def test_round_trip_string_list_context(self):
|
||||
example = tf.train.SequenceExample()
|
||||
msu.set_string_list_context([b"test0", b"test1"], example)
|
||||
self.assertSequenceEqual([b"test0", b"test1"],
|
||||
msu.get_string_list_context(example))
|
||||
self.assertTrue(msu.has_string_list_context(example))
|
||||
msu.clear_string_list_context(example)
|
||||
self.assertFalse(msu.has_string_list_context(example))
|
||||
self.assertEqual("string_vector_feature",
|
||||
msu.get_string_list_context_key())
|
||||
|
||||
def test_round_trip_float_list_context(self):
|
||||
example = tf.train.SequenceExample()
|
||||
msu.set_float_list_context([0.47, 0.49], example)
|
||||
self.assertSequenceAlmostEqual([0.47, 0.49],
|
||||
msu.get_float_list_context(example))
|
||||
self.assertTrue(msu.has_float_list_context(example))
|
||||
msu.clear_float_list_context(example)
|
||||
self.assertFalse(msu.has_float_list_context(example))
|
||||
self.assertEqual("float_vector_feature", msu.get_float_list_context_key())
|
||||
|
||||
def test_round_trip_int_list_context(self):
|
||||
example = tf.train.SequenceExample()
|
||||
msu.set_int64_list_context([47, 49], example)
|
||||
self.assertSequenceEqual([47, 49], msu.get_int64_list_context(example))
|
||||
self.assertTrue(msu.has_int64_list_context(example))
|
||||
msu.clear_int64_list_context(example)
|
||||
self.assertFalse(msu.has_int64_list_context(example))
|
||||
self.assertEqual("int64_vector_feature", msu.get_int64_list_context_key())
|
||||
|
||||
def test_round_trip_string_feature_list(self):
|
||||
example = tf.train.SequenceExample()
|
||||
msu.add_string_feature_list(b"test0", example)
|
||||
msu.add_string_feature_list(b"test1", example)
|
||||
self.assertEqual(b"test0", msu.get_string_feature_list_at(0, example))
|
||||
self.assertEqual(b"test1", msu.get_string_feature_list_at(1, example))
|
||||
self.assertTrue(msu.has_string_feature_list(example))
|
||||
self.assertEqual(2, msu.get_string_feature_list_size(example))
|
||||
msu.clear_string_feature_list(example)
|
||||
self.assertFalse(msu.has_string_feature_list(example))
|
||||
self.assertEqual(0, msu.get_string_feature_list_size(example))
|
||||
self.assertEqual("string_feature_list", msu.get_string_feature_list_key())
|
||||
|
||||
def test_round_trip_float_feature_list(self):
|
||||
example = tf.train.SequenceExample()
|
||||
msu.add_float_feature_list(0.47, example)
|
||||
msu.add_float_feature_list(0.49, example)
|
||||
self.assertAlmostEqual(0.47, msu.get_float_feature_list_at(0, example))
|
||||
self.assertAlmostEqual(0.49, msu.get_float_feature_list_at(1, example))
|
||||
self.assertTrue(msu.has_float_feature_list(example))
|
||||
self.assertEqual(2, msu.get_float_feature_list_size(example))
|
||||
msu.clear_float_feature_list(example)
|
||||
self.assertFalse(msu.has_float_feature_list(example))
|
||||
self.assertEqual(0, msu.get_float_feature_list_size(example))
|
||||
self.assertEqual("float_feature_list", msu.get_float_feature_list_key())
|
||||
|
||||
def test_round_trip_int_feature_list(self):
|
||||
example = tf.train.SequenceExample()
|
||||
msu.add_int64_feature_list(47, example)
|
||||
msu.add_int64_feature_list(49, example)
|
||||
self.assertEqual(47, msu.get_int64_feature_list_at(0, example))
|
||||
self.assertEqual(49, msu.get_int64_feature_list_at(1, example))
|
||||
self.assertTrue(msu.has_int64_feature_list(example))
|
||||
self.assertEqual(2, msu.get_int64_feature_list_size(example))
|
||||
msu.clear_int64_feature_list(example)
|
||||
self.assertFalse(msu.has_int64_feature_list(example))
|
||||
self.assertEqual(0, msu.get_int64_feature_list_size(example))
|
||||
self.assertEqual("int64_feature_list", msu.get_int64_feature_list_key())
|
||||
|
||||
def test_round_trip_string_list_feature_list(self):
|
||||
example = tf.train.SequenceExample()
|
||||
msu.add_string_list_feature_list([b"test0", b"test1"], example)
|
||||
msu.add_string_list_feature_list([b"test1", b"test2"], example)
|
||||
self.assertSequenceEqual([b"test0", b"test1"],
|
||||
msu.get_string_list_feature_list_at(0, example))
|
||||
self.assertSequenceEqual([b"test1", b"test2"],
|
||||
msu.get_string_list_feature_list_at(1, example))
|
||||
self.assertTrue(msu.has_string_list_feature_list(example))
|
||||
self.assertEqual(2, msu.get_string_list_feature_list_size(example))
|
||||
msu.clear_string_list_feature_list(example)
|
||||
self.assertFalse(msu.has_string_list_feature_list(example))
|
||||
self.assertEqual(0, msu.get_string_list_feature_list_size(example))
|
||||
self.assertEqual("string_list_feature_list",
|
||||
msu.get_string_list_feature_list_key())
|
||||
|
||||
def test_round_trip_float_list_feature_list(self):
|
||||
example = tf.train.SequenceExample()
|
||||
msu.add_float_list_feature_list([0.47, 0.49], example)
|
||||
msu.add_float_list_feature_list([0.49, 0.50], example)
|
||||
self.assertSequenceAlmostEqual(
|
||||
[0.47, 0.49], msu.get_float_list_feature_list_at(0, example))
|
||||
self.assertSequenceAlmostEqual(
|
||||
[0.49, 0.50], msu.get_float_list_feature_list_at(1, example))
|
||||
self.assertTrue(msu.has_float_list_feature_list(example))
|
||||
self.assertEqual(2, msu.get_float_list_feature_list_size(example))
|
||||
msu.clear_float_list_feature_list(example)
|
||||
self.assertFalse(msu.has_float_list_feature_list(example))
|
||||
self.assertEqual(0, msu.get_float_list_feature_list_size(example))
|
||||
self.assertEqual("float_list_feature_list",
|
||||
msu.get_float_list_feature_list_key())
|
||||
|
||||
def test_round_trip_int_list_feature_list(self):
|
||||
example = tf.train.SequenceExample()
|
||||
msu.add_int64_list_feature_list([47, 49], example)
|
||||
msu.add_int64_list_feature_list([49, 50], example)
|
||||
self.assertSequenceEqual(
|
||||
[47, 49], msu.get_int64_list_feature_list_at(0, example))
|
||||
self.assertSequenceEqual(
|
||||
[49, 50], msu.get_int64_list_feature_list_at(1, example))
|
||||
self.assertTrue(msu.has_int64_list_feature_list(example))
|
||||
self.assertEqual(2, msu.get_int64_list_feature_list_size(example))
|
||||
msu.clear_int64_list_feature_list(example)
|
||||
self.assertFalse(msu.has_int64_list_feature_list(example))
|
||||
self.assertEqual(0, msu.get_int64_list_feature_list_size(example))
|
||||
self.assertEqual("int64_list_feature_list",
|
||||
msu.get_int64_list_feature_list_key())
|
||||
|
||||
def test_prefix_int64_context(self):
|
||||
example = tf.train.SequenceExample()
|
||||
msu.set_int64_context(47, example, prefix="magic")
|
||||
self.assertFalse(msu.has_int64_context(example))
|
||||
self.assertTrue(msu.has_int64_context(example, prefix="magic"))
|
||||
self.assertEqual(47, msu.get_int64_context(example, prefix="magic"))
|
||||
|
||||
def test_prefix_float_context(self):
|
||||
example = tf.train.SequenceExample()
|
||||
msu.set_float_context(47., example, prefix="magic")
|
||||
self.assertFalse(msu.has_float_context(example))
|
||||
self.assertTrue(msu.has_float_context(example, prefix="magic"))
|
||||
self.assertAlmostEqual(47., msu.get_float_context(example, prefix="magic"))
|
||||
|
||||
def test_prefix_string_context(self):
|
||||
example = tf.train.SequenceExample()
|
||||
msu.set_string_context(b"47", example, prefix="magic")
|
||||
self.assertFalse(msu.has_string_context(example))
|
||||
self.assertTrue(msu.has_string_context(example, prefix="magic"))
|
||||
self.assertEqual(b"47", msu.get_string_context(example, prefix="magic"))
|
||||
|
||||
def test_prefix_int64_list_context(self):
|
||||
example = tf.train.SequenceExample()
|
||||
msu.set_int64_list_context((47,), example, prefix="magic")
|
||||
self.assertFalse(msu.has_int64_list_context(example))
|
||||
self.assertTrue(msu.has_int64_list_context(example, prefix="magic"))
|
||||
self.assertEqual([47,], msu.get_int64_list_context(example, prefix="magic"))
|
||||
|
||||
def test_prefix_float_list_context(self):
|
||||
example = tf.train.SequenceExample()
|
||||
msu.set_float_list_context((47.,), example, prefix="magic")
|
||||
self.assertFalse(msu.has_float_list_context(example))
|
||||
self.assertTrue(msu.has_float_list_context(example, prefix="magic"))
|
||||
self.assertAlmostEqual([47.,],
|
||||
msu.get_float_list_context(example, prefix="magic"))
|
||||
|
||||
def test_prefix_string_list_context(self):
|
||||
example = tf.train.SequenceExample()
|
||||
msu.set_string_list_context((b"47",), example, prefix="magic")
|
||||
self.assertFalse(msu.has_string_list_context(example))
|
||||
self.assertTrue(msu.has_string_list_context(example, prefix="magic"))
|
||||
self.assertEqual([b"47",],
|
||||
msu.get_string_list_context(example, prefix="magic"))
|
||||
|
||||
def test_prefix_int64_feature_list(self):
|
||||
example = tf.train.SequenceExample()
|
||||
msu.add_int64_feature_list(47, example, prefix="magic")
|
||||
self.assertFalse(msu.has_int64_feature_list(example))
|
||||
self.assertTrue(msu.has_int64_feature_list(example, prefix="magic"))
|
||||
self.assertEqual(47,
|
||||
msu.get_int64_feature_list_at(0, example, prefix="magic"))
|
||||
|
||||
def test_prefix_float_feature_list(self):
|
||||
example = tf.train.SequenceExample()
|
||||
msu.add_float_feature_list(47., example, prefix="magic")
|
||||
self.assertFalse(msu.has_float_feature_list(example))
|
||||
self.assertTrue(msu.has_float_feature_list(example, prefix="magic"))
|
||||
self.assertAlmostEqual(
|
||||
47., msu.get_float_feature_list_at(0, example, prefix="magic"))
|
||||
|
||||
def test_prefix_string_feature_list(self):
|
||||
example = tf.train.SequenceExample()
|
||||
msu.add_string_feature_list(b"47", example, prefix="magic")
|
||||
self.assertFalse(msu.has_string_feature_list(example))
|
||||
self.assertTrue(msu.has_string_feature_list(example, prefix="magic"))
|
||||
self.assertEqual(
|
||||
b"47", msu.get_string_feature_list_at(0, example, prefix="magic"))
|
||||
|
||||
def test_prefix_int64_list_feature_list(self):
|
||||
example = tf.train.SequenceExample()
|
||||
msu.add_int64_list_feature_list((47,), example, prefix="magic")
|
||||
self.assertFalse(msu.has_int64_list_feature_list(example))
|
||||
self.assertTrue(msu.has_int64_list_feature_list(example, prefix="magic"))
|
||||
self.assertEqual(
|
||||
[47,], msu.get_int64_list_feature_list_at(0, example, prefix="magic"))
|
||||
|
||||
def test_prefix_float_list_feature_list(self):
|
||||
example = tf.train.SequenceExample()
|
||||
msu.add_float_list_feature_list((47.,), example, prefix="magic")
|
||||
self.assertFalse(msu.has_float_list_feature_list(example))
|
||||
self.assertTrue(msu.has_float_list_feature_list(example, prefix="magic"))
|
||||
self.assertAlmostEqual(
|
||||
[47.,], msu.get_float_list_feature_list_at(0, example, prefix="magic"))
|
||||
|
||||
def test_prefix_string_list_feature_list(self):
|
||||
example = tf.train.SequenceExample()
|
||||
msu.add_string_list_feature_list((b"47",), example, prefix="magic")
|
||||
self.assertFalse(msu.has_string_list_feature_list(example))
|
||||
self.assertTrue(msu.has_string_list_feature_list(example, prefix="magic"))
|
||||
self.assertEqual(
|
||||
[b"47",],
|
||||
msu.get_string_list_feature_list_at(0, example, prefix="magic"))
|
||||
|
||||
if __name__ == "__main__":
|
||||
tf.test.main()
|
||||
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
@@ -0,0 +1,351 @@
|
||||
# -*- coding: utf-8 -*-
|
||||
# Generated by the protocol buffer compiler. DO NOT EDIT!
|
||||
# source: mediapipe/util/tracking/box_detector.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.util.tracking import box_tracker_pb2 as mediapipe_dot_util_dot_tracking_dot_box__tracker__pb2
|
||||
from mediapipe.util.tracking import region_flow_pb2 as mediapipe_dot_util_dot_tracking_dot_region__flow__pb2
|
||||
|
||||
|
||||
DESCRIPTOR = _descriptor.FileDescriptor(
|
||||
name='mediapipe/util/tracking/box_detector.proto',
|
||||
package='mediapipe',
|
||||
syntax='proto2',
|
||||
serialized_options=_b('\n\035com.google.mediapipe.trackingB\020BoxDetectorProto'),
|
||||
serialized_pb=_b('\n*mediapipe/util/tracking/box_detector.proto\x12\tmediapipe\x1a)mediapipe/util/tracking/box_tracker.proto\x1a)mediapipe/util/tracking/region_flow.proto\"\xec\x04\n\x12\x42oxDetectorOptions\x12\x46\n\nindex_type\x18\x01 \x01(\x0e\x32\'.mediapipe.BoxDetectorOptions.IndexType:\tOPENCV_BF\x12\x1f\n\x14\x64\x65tect_every_n_frame\x18\x02 \x01(\x05:\x01\x30\x12 \n\x11\x64\x65tect_out_of_fov\x18\x04 \x01(\x08:\x05\x66\x61lse\x12N\n\x14image_query_settings\x18\x03 \x01(\x0b\x32\x30.mediapipe.BoxDetectorOptions.ImageQuerySettings\x12\x1b\n\x0f\x64\x65scriptor_dims\x18\x05 \x01(\x05:\x02\x34\x30\x12!\n\x16min_num_correspondence\x18\x06 \x01(\x05:\x01\x35\x12,\n\x1dransac_reprojection_threshold\x18\x07 \x01(\x02:\x05\x30.005\x12\x1f\n\x12max_match_distance\x18\x08 \x01(\x02:\x03\x30.9\x12#\n\x16max_perspective_factor\x18\t \x01(\x02:\x03\x30.1\x1a\x93\x01\n\x12ImageQuerySettings\x12 \n\x13pyramid_bottom_size\x18\x01 \x01(\x05:\x03\x36\x34\x30\x12!\n\x14pyramid_scale_factor\x18\x02 \x01(\x02:\x03\x31.2\x12\x1d\n\x12max_pyramid_levels\x18\x03 \x01(\x05:\x01\x34\x12\x19\n\x0cmax_features\x18\x04 \x01(\x05:\x03\x35\x30\x30\"1\n\tIndexType\x12\x15\n\x11INDEX_UNSPECIFIED\x10\x00\x12\r\n\tOPENCV_BF\x10\x01\"\x9f\x02\n\x10\x42oxDetectorIndex\x12\x37\n\tbox_entry\x18\x01 \x03(\x0b\x32$.mediapipe.BoxDetectorIndex.BoxEntry\x1a\xd1\x01\n\x08\x42oxEntry\x12\x44\n\x0b\x66rame_entry\x18\x01 \x03(\x0b\x32/.mediapipe.BoxDetectorIndex.BoxEntry.FrameEntry\x1a\x7f\n\nFrameEntry\x12%\n\x03\x62ox\x18\x01 \x01(\x0b\x32\x18.mediapipe.TimedBoxProto\x12\x11\n\tkeypoints\x18\x02 \x03(\x02\x12\x37\n\x0b\x64\x65scriptors\x18\x03 \x03(\x0b\x32\".mediapipe.BinaryFeatureDescriptorB1\n\x1d\x63om.google.mediapipe.trackingB\x10\x42oxDetectorProto')
|
||||
,
|
||||
dependencies=[mediapipe_dot_util_dot_tracking_dot_box__tracker__pb2.DESCRIPTOR,mediapipe_dot_util_dot_tracking_dot_region__flow__pb2.DESCRIPTOR,])
|
||||
|
||||
|
||||
|
||||
_BOXDETECTOROPTIONS_INDEXTYPE = _descriptor.EnumDescriptor(
|
||||
name='IndexType',
|
||||
full_name='mediapipe.BoxDetectorOptions.IndexType',
|
||||
filename=None,
|
||||
file=DESCRIPTOR,
|
||||
values=[
|
||||
_descriptor.EnumValueDescriptor(
|
||||
name='INDEX_UNSPECIFIED', index=0, number=0,
|
||||
serialized_options=None,
|
||||
type=None),
|
||||
_descriptor.EnumValueDescriptor(
|
||||
name='OPENCV_BF', index=1, number=1,
|
||||
serialized_options=None,
|
||||
type=None),
|
||||
],
|
||||
containing_type=None,
|
||||
serialized_options=None,
|
||||
serialized_start=715,
|
||||
serialized_end=764,
|
||||
)
|
||||
_sym_db.RegisterEnumDescriptor(_BOXDETECTOROPTIONS_INDEXTYPE)
|
||||
|
||||
|
||||
_BOXDETECTOROPTIONS_IMAGEQUERYSETTINGS = _descriptor.Descriptor(
|
||||
name='ImageQuerySettings',
|
||||
full_name='mediapipe.BoxDetectorOptions.ImageQuerySettings',
|
||||
filename=None,
|
||||
file=DESCRIPTOR,
|
||||
containing_type=None,
|
||||
fields=[
|
||||
_descriptor.FieldDescriptor(
|
||||
name='pyramid_bottom_size', full_name='mediapipe.BoxDetectorOptions.ImageQuerySettings.pyramid_bottom_size', index=0,
|
||||
number=1, type=5, cpp_type=1, label=1,
|
||||
has_default_value=True, default_value=640,
|
||||
message_type=None, enum_type=None, containing_type=None,
|
||||
is_extension=False, extension_scope=None,
|
||||
serialized_options=None, file=DESCRIPTOR),
|
||||
_descriptor.FieldDescriptor(
|
||||
name='pyramid_scale_factor', full_name='mediapipe.BoxDetectorOptions.ImageQuerySettings.pyramid_scale_factor', index=1,
|
||||
number=2, type=2, cpp_type=6, label=1,
|
||||
has_default_value=True, default_value=float(1.2),
|
||||
message_type=None, enum_type=None, containing_type=None,
|
||||
is_extension=False, extension_scope=None,
|
||||
serialized_options=None, file=DESCRIPTOR),
|
||||
_descriptor.FieldDescriptor(
|
||||
name='max_pyramid_levels', full_name='mediapipe.BoxDetectorOptions.ImageQuerySettings.max_pyramid_levels', index=2,
|
||||
number=3, type=5, cpp_type=1, label=1,
|
||||
has_default_value=True, default_value=4,
|
||||
message_type=None, enum_type=None, containing_type=None,
|
||||
is_extension=False, extension_scope=None,
|
||||
serialized_options=None, file=DESCRIPTOR),
|
||||
_descriptor.FieldDescriptor(
|
||||
name='max_features', full_name='mediapipe.BoxDetectorOptions.ImageQuerySettings.max_features', index=3,
|
||||
number=4, type=5, cpp_type=1, label=1,
|
||||
has_default_value=True, default_value=500,
|
||||
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=566,
|
||||
serialized_end=713,
|
||||
)
|
||||
|
||||
_BOXDETECTOROPTIONS = _descriptor.Descriptor(
|
||||
name='BoxDetectorOptions',
|
||||
full_name='mediapipe.BoxDetectorOptions',
|
||||
filename=None,
|
||||
file=DESCRIPTOR,
|
||||
containing_type=None,
|
||||
fields=[
|
||||
_descriptor.FieldDescriptor(
|
||||
name='index_type', full_name='mediapipe.BoxDetectorOptions.index_type', index=0,
|
||||
number=1, type=14, cpp_type=8, 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='detect_every_n_frame', full_name='mediapipe.BoxDetectorOptions.detect_every_n_frame', index=1,
|
||||
number=2, 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='detect_out_of_fov', full_name='mediapipe.BoxDetectorOptions.detect_out_of_fov', index=2,
|
||||
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='image_query_settings', full_name='mediapipe.BoxDetectorOptions.image_query_settings', index=3,
|
||||
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='descriptor_dims', full_name='mediapipe.BoxDetectorOptions.descriptor_dims', index=4,
|
||||
number=5, type=5, cpp_type=1, label=1,
|
||||
has_default_value=True, default_value=40,
|
||||
message_type=None, enum_type=None, containing_type=None,
|
||||
is_extension=False, extension_scope=None,
|
||||
serialized_options=None, file=DESCRIPTOR),
|
||||
_descriptor.FieldDescriptor(
|
||||
name='min_num_correspondence', full_name='mediapipe.BoxDetectorOptions.min_num_correspondence', index=5,
|
||||
number=6, type=5, cpp_type=1, label=1,
|
||||
has_default_value=True, default_value=5,
|
||||
message_type=None, enum_type=None, containing_type=None,
|
||||
is_extension=False, extension_scope=None,
|
||||
serialized_options=None, file=DESCRIPTOR),
|
||||
_descriptor.FieldDescriptor(
|
||||
name='ransac_reprojection_threshold', full_name='mediapipe.BoxDetectorOptions.ransac_reprojection_threshold', index=6,
|
||||
number=7, type=2, cpp_type=6, label=1,
|
||||
has_default_value=True, default_value=float(0.005),
|
||||
message_type=None, enum_type=None, containing_type=None,
|
||||
is_extension=False, extension_scope=None,
|
||||
serialized_options=None, file=DESCRIPTOR),
|
||||
_descriptor.FieldDescriptor(
|
||||
name='max_match_distance', full_name='mediapipe.BoxDetectorOptions.max_match_distance', index=7,
|
||||
number=8, type=2, cpp_type=6, label=1,
|
||||
has_default_value=True, default_value=float(0.9),
|
||||
message_type=None, enum_type=None, containing_type=None,
|
||||
is_extension=False, extension_scope=None,
|
||||
serialized_options=None, file=DESCRIPTOR),
|
||||
_descriptor.FieldDescriptor(
|
||||
name='max_perspective_factor', full_name='mediapipe.BoxDetectorOptions.max_perspective_factor', index=8,
|
||||
number=9, type=2, cpp_type=6, label=1,
|
||||
has_default_value=True, default_value=float(0.1),
|
||||
message_type=None, enum_type=None, containing_type=None,
|
||||
is_extension=False, extension_scope=None,
|
||||
serialized_options=None, file=DESCRIPTOR),
|
||||
],
|
||||
extensions=[
|
||||
],
|
||||
nested_types=[_BOXDETECTOROPTIONS_IMAGEQUERYSETTINGS, ],
|
||||
enum_types=[
|
||||
_BOXDETECTOROPTIONS_INDEXTYPE,
|
||||
],
|
||||
serialized_options=None,
|
||||
is_extendable=False,
|
||||
syntax='proto2',
|
||||
extension_ranges=[],
|
||||
oneofs=[
|
||||
],
|
||||
serialized_start=144,
|
||||
serialized_end=764,
|
||||
)
|
||||
|
||||
|
||||
_BOXDETECTORINDEX_BOXENTRY_FRAMEENTRY = _descriptor.Descriptor(
|
||||
name='FrameEntry',
|
||||
full_name='mediapipe.BoxDetectorIndex.BoxEntry.FrameEntry',
|
||||
filename=None,
|
||||
file=DESCRIPTOR,
|
||||
containing_type=None,
|
||||
fields=[
|
||||
_descriptor.FieldDescriptor(
|
||||
name='box', full_name='mediapipe.BoxDetectorIndex.BoxEntry.FrameEntry.box', 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='keypoints', full_name='mediapipe.BoxDetectorIndex.BoxEntry.FrameEntry.keypoints', index=1,
|
||||
number=2, 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='descriptors', full_name='mediapipe.BoxDetectorIndex.BoxEntry.FrameEntry.descriptors', index=2,
|
||||
number=3, type=11, cpp_type=10, 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),
|
||||
],
|
||||
extensions=[
|
||||
],
|
||||
nested_types=[],
|
||||
enum_types=[
|
||||
],
|
||||
serialized_options=None,
|
||||
is_extendable=False,
|
||||
syntax='proto2',
|
||||
extension_ranges=[],
|
||||
oneofs=[
|
||||
],
|
||||
serialized_start=927,
|
||||
serialized_end=1054,
|
||||
)
|
||||
|
||||
_BOXDETECTORINDEX_BOXENTRY = _descriptor.Descriptor(
|
||||
name='BoxEntry',
|
||||
full_name='mediapipe.BoxDetectorIndex.BoxEntry',
|
||||
filename=None,
|
||||
file=DESCRIPTOR,
|
||||
containing_type=None,
|
||||
fields=[
|
||||
_descriptor.FieldDescriptor(
|
||||
name='frame_entry', full_name='mediapipe.BoxDetectorIndex.BoxEntry.frame_entry', index=0,
|
||||
number=1, type=11, cpp_type=10, 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),
|
||||
],
|
||||
extensions=[
|
||||
],
|
||||
nested_types=[_BOXDETECTORINDEX_BOXENTRY_FRAMEENTRY, ],
|
||||
enum_types=[
|
||||
],
|
||||
serialized_options=None,
|
||||
is_extendable=False,
|
||||
syntax='proto2',
|
||||
extension_ranges=[],
|
||||
oneofs=[
|
||||
],
|
||||
serialized_start=845,
|
||||
serialized_end=1054,
|
||||
)
|
||||
|
||||
_BOXDETECTORINDEX = _descriptor.Descriptor(
|
||||
name='BoxDetectorIndex',
|
||||
full_name='mediapipe.BoxDetectorIndex',
|
||||
filename=None,
|
||||
file=DESCRIPTOR,
|
||||
containing_type=None,
|
||||
fields=[
|
||||
_descriptor.FieldDescriptor(
|
||||
name='box_entry', full_name='mediapipe.BoxDetectorIndex.box_entry', index=0,
|
||||
number=1, type=11, cpp_type=10, 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),
|
||||
],
|
||||
extensions=[
|
||||
],
|
||||
nested_types=[_BOXDETECTORINDEX_BOXENTRY, ],
|
||||
enum_types=[
|
||||
],
|
||||
serialized_options=None,
|
||||
is_extendable=False,
|
||||
syntax='proto2',
|
||||
extension_ranges=[],
|
||||
oneofs=[
|
||||
],
|
||||
serialized_start=767,
|
||||
serialized_end=1054,
|
||||
)
|
||||
|
||||
_BOXDETECTOROPTIONS_IMAGEQUERYSETTINGS.containing_type = _BOXDETECTOROPTIONS
|
||||
_BOXDETECTOROPTIONS.fields_by_name['index_type'].enum_type = _BOXDETECTOROPTIONS_INDEXTYPE
|
||||
_BOXDETECTOROPTIONS.fields_by_name['image_query_settings'].message_type = _BOXDETECTOROPTIONS_IMAGEQUERYSETTINGS
|
||||
_BOXDETECTOROPTIONS_INDEXTYPE.containing_type = _BOXDETECTOROPTIONS
|
||||
_BOXDETECTORINDEX_BOXENTRY_FRAMEENTRY.fields_by_name['box'].message_type = mediapipe_dot_util_dot_tracking_dot_box__tracker__pb2._TIMEDBOXPROTO
|
||||
_BOXDETECTORINDEX_BOXENTRY_FRAMEENTRY.fields_by_name['descriptors'].message_type = mediapipe_dot_util_dot_tracking_dot_region__flow__pb2._BINARYFEATUREDESCRIPTOR
|
||||
_BOXDETECTORINDEX_BOXENTRY_FRAMEENTRY.containing_type = _BOXDETECTORINDEX_BOXENTRY
|
||||
_BOXDETECTORINDEX_BOXENTRY.fields_by_name['frame_entry'].message_type = _BOXDETECTORINDEX_BOXENTRY_FRAMEENTRY
|
||||
_BOXDETECTORINDEX_BOXENTRY.containing_type = _BOXDETECTORINDEX
|
||||
_BOXDETECTORINDEX.fields_by_name['box_entry'].message_type = _BOXDETECTORINDEX_BOXENTRY
|
||||
DESCRIPTOR.message_types_by_name['BoxDetectorOptions'] = _BOXDETECTOROPTIONS
|
||||
DESCRIPTOR.message_types_by_name['BoxDetectorIndex'] = _BOXDETECTORINDEX
|
||||
_sym_db.RegisterFileDescriptor(DESCRIPTOR)
|
||||
|
||||
BoxDetectorOptions = _reflection.GeneratedProtocolMessageType('BoxDetectorOptions', (_message.Message,), dict(
|
||||
|
||||
ImageQuerySettings = _reflection.GeneratedProtocolMessageType('ImageQuerySettings', (_message.Message,), dict(
|
||||
DESCRIPTOR = _BOXDETECTOROPTIONS_IMAGEQUERYSETTINGS,
|
||||
__module__ = 'mediapipe.util.tracking.box_detector_pb2'
|
||||
# @@protoc_insertion_point(class_scope:mediapipe.BoxDetectorOptions.ImageQuerySettings)
|
||||
))
|
||||
,
|
||||
DESCRIPTOR = _BOXDETECTOROPTIONS,
|
||||
__module__ = 'mediapipe.util.tracking.box_detector_pb2'
|
||||
# @@protoc_insertion_point(class_scope:mediapipe.BoxDetectorOptions)
|
||||
))
|
||||
_sym_db.RegisterMessage(BoxDetectorOptions)
|
||||
_sym_db.RegisterMessage(BoxDetectorOptions.ImageQuerySettings)
|
||||
|
||||
BoxDetectorIndex = _reflection.GeneratedProtocolMessageType('BoxDetectorIndex', (_message.Message,), dict(
|
||||
|
||||
BoxEntry = _reflection.GeneratedProtocolMessageType('BoxEntry', (_message.Message,), dict(
|
||||
|
||||
FrameEntry = _reflection.GeneratedProtocolMessageType('FrameEntry', (_message.Message,), dict(
|
||||
DESCRIPTOR = _BOXDETECTORINDEX_BOXENTRY_FRAMEENTRY,
|
||||
__module__ = 'mediapipe.util.tracking.box_detector_pb2'
|
||||
# @@protoc_insertion_point(class_scope:mediapipe.BoxDetectorIndex.BoxEntry.FrameEntry)
|
||||
))
|
||||
,
|
||||
DESCRIPTOR = _BOXDETECTORINDEX_BOXENTRY,
|
||||
__module__ = 'mediapipe.util.tracking.box_detector_pb2'
|
||||
# @@protoc_insertion_point(class_scope:mediapipe.BoxDetectorIndex.BoxEntry)
|
||||
))
|
||||
,
|
||||
DESCRIPTOR = _BOXDETECTORINDEX,
|
||||
__module__ = 'mediapipe.util.tracking.box_detector_pb2'
|
||||
# @@protoc_insertion_point(class_scope:mediapipe.BoxDetectorIndex)
|
||||
))
|
||||
_sym_db.RegisterMessage(BoxDetectorIndex)
|
||||
_sym_db.RegisterMessage(BoxDetectorIndex.BoxEntry)
|
||||
_sym_db.RegisterMessage(BoxDetectorIndex.BoxEntry.FrameEntry)
|
||||
|
||||
|
||||
DESCRIPTOR._options = None
|
||||
# @@protoc_insertion_point(module_scope)
|
||||
@@ -0,0 +1,273 @@
|
||||
# -*- coding: utf-8 -*-
|
||||
# Generated by the protocol buffer compiler. DO NOT EDIT!
|
||||
# source: mediapipe/util/tracking/box_tracker.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.util.tracking import tracking_pb2 as mediapipe_dot_util_dot_tracking_dot_tracking__pb2
|
||||
|
||||
|
||||
DESCRIPTOR = _descriptor.FileDescriptor(
|
||||
name='mediapipe/util/tracking/box_tracker.proto',
|
||||
package='mediapipe',
|
||||
syntax='proto2',
|
||||
serialized_options=_b('\n\035com.google.mediapipe.trackingB\017BoxTrackerProto'),
|
||||
serialized_pb=_b('\n)mediapipe/util/tracking/box_tracker.proto\x12\tmediapipe\x1a&mediapipe/util/tracking/tracking.proto\"\x86\x02\n\x11\x42oxTrackerOptions\x12%\n\x17\x63\x61\x63hing_chunk_size_msec\x18\x01 \x01(\x05:\x04\x32\x35\x30\x30\x12%\n\x11\x63\x61\x63he_file_format\x18\x02 \x01(\t:\nchunk_%04d\x12\x1f\n\x14num_tracking_workers\x18\x03 \x01(\x05:\x01\x38\x12&\n\x17read_chunk_timeout_msec\x18\x04 \x01(\x05:\x05\x36\x30\x30\x30\x30\x12!\n\x12record_path_states\x18\x05 \x01(\x08:\x05\x66\x61lse\x12\x37\n\x12track_step_options\x18\x06 \x01(\x0b\x32\x1b.mediapipe.TrackStepOptions\"\xa7\x02\n\rTimedBoxProto\x12\x0b\n\x03top\x18\x01 \x01(\x02\x12\x0c\n\x04left\x18\x02 \x01(\x02\x12\x0e\n\x06\x62ottom\x18\x03 \x01(\x02\x12\r\n\x05right\x18\x04 \x01(\x02\x12\x10\n\x08rotation\x18\x07 \x01(\x02\x12,\n\x04quad\x18\t \x01(\x0b\x32\x1e.mediapipe.MotionBoxState.Quad\x12\x14\n\ttime_msec\x18\x05 \x01(\x03:\x01\x30\x12\x0e\n\x02id\x18\x06 \x01(\x05:\x02-1\x12\r\n\x05label\x18\r \x01(\t\x12\x12\n\nconfidence\x18\x08 \x01(\x02\x12\x14\n\x0c\x61spect_ratio\x18\n \x01(\x02\x12\x1c\n\rreacquisition\x18\x0b \x01(\x08:\x05\x66\x61lse\x12\x1f\n\x10request_grouping\x18\x0c \x01(\x08:\x05\x66\x61lse\":\n\x11TimedBoxProtoList\x12%\n\x03\x62ox\x18\x01 \x03(\x0b\x32\x18.mediapipe.TimedBoxProtoB0\n\x1d\x63om.google.mediapipe.trackingB\x0f\x42oxTrackerProto')
|
||||
,
|
||||
dependencies=[mediapipe_dot_util_dot_tracking_dot_tracking__pb2.DESCRIPTOR,])
|
||||
|
||||
|
||||
|
||||
|
||||
_BOXTRACKEROPTIONS = _descriptor.Descriptor(
|
||||
name='BoxTrackerOptions',
|
||||
full_name='mediapipe.BoxTrackerOptions',
|
||||
filename=None,
|
||||
file=DESCRIPTOR,
|
||||
containing_type=None,
|
||||
fields=[
|
||||
_descriptor.FieldDescriptor(
|
||||
name='caching_chunk_size_msec', full_name='mediapipe.BoxTrackerOptions.caching_chunk_size_msec', index=0,
|
||||
number=1, type=5, cpp_type=1, label=1,
|
||||
has_default_value=True, default_value=2500,
|
||||
message_type=None, enum_type=None, containing_type=None,
|
||||
is_extension=False, extension_scope=None,
|
||||
serialized_options=None, file=DESCRIPTOR),
|
||||
_descriptor.FieldDescriptor(
|
||||
name='cache_file_format', full_name='mediapipe.BoxTrackerOptions.cache_file_format', index=1,
|
||||
number=2, type=9, cpp_type=9, label=1,
|
||||
has_default_value=True, default_value=_b("chunk_%04d").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='num_tracking_workers', full_name='mediapipe.BoxTrackerOptions.num_tracking_workers', index=2,
|
||||
number=3, type=5, cpp_type=1, label=1,
|
||||
has_default_value=True, default_value=8,
|
||||
message_type=None, enum_type=None, containing_type=None,
|
||||
is_extension=False, extension_scope=None,
|
||||
serialized_options=None, file=DESCRIPTOR),
|
||||
_descriptor.FieldDescriptor(
|
||||
name='read_chunk_timeout_msec', full_name='mediapipe.BoxTrackerOptions.read_chunk_timeout_msec', index=3,
|
||||
number=4, type=5, cpp_type=1, label=1,
|
||||
has_default_value=True, default_value=60000,
|
||||
message_type=None, enum_type=None, containing_type=None,
|
||||
is_extension=False, extension_scope=None,
|
||||
serialized_options=None, file=DESCRIPTOR),
|
||||
_descriptor.FieldDescriptor(
|
||||
name='record_path_states', full_name='mediapipe.BoxTrackerOptions.record_path_states', index=4,
|
||||
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='track_step_options', full_name='mediapipe.BoxTrackerOptions.track_step_options', index=5,
|
||||
number=6, 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=[],
|
||||
enum_types=[
|
||||
],
|
||||
serialized_options=None,
|
||||
is_extendable=False,
|
||||
syntax='proto2',
|
||||
extension_ranges=[],
|
||||
oneofs=[
|
||||
],
|
||||
serialized_start=97,
|
||||
serialized_end=359,
|
||||
)
|
||||
|
||||
|
||||
_TIMEDBOXPROTO = _descriptor.Descriptor(
|
||||
name='TimedBoxProto',
|
||||
full_name='mediapipe.TimedBoxProto',
|
||||
filename=None,
|
||||
file=DESCRIPTOR,
|
||||
containing_type=None,
|
||||
fields=[
|
||||
_descriptor.FieldDescriptor(
|
||||
name='top', full_name='mediapipe.TimedBoxProto.top', 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='left', full_name='mediapipe.TimedBoxProto.left', 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),
|
||||
_descriptor.FieldDescriptor(
|
||||
name='bottom', full_name='mediapipe.TimedBoxProto.bottom', index=2,
|
||||
number=3, 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='right', full_name='mediapipe.TimedBoxProto.right', index=3,
|
||||
number=4, 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='rotation', full_name='mediapipe.TimedBoxProto.rotation', index=4,
|
||||
number=7, 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='quad', full_name='mediapipe.TimedBoxProto.quad', index=5,
|
||||
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),
|
||||
_descriptor.FieldDescriptor(
|
||||
name='time_msec', full_name='mediapipe.TimedBoxProto.time_msec', index=6,
|
||||
number=5, type=3, cpp_type=2, 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='id', full_name='mediapipe.TimedBoxProto.id', index=7,
|
||||
number=6, 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='label', full_name='mediapipe.TimedBoxProto.label', index=8,
|
||||
number=13, 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='confidence', full_name='mediapipe.TimedBoxProto.confidence', index=9,
|
||||
number=8, 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='aspect_ratio', full_name='mediapipe.TimedBoxProto.aspect_ratio', index=10,
|
||||
number=10, 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='reacquisition', full_name='mediapipe.TimedBoxProto.reacquisition', index=11,
|
||||
number=11, 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='request_grouping', full_name='mediapipe.TimedBoxProto.request_grouping', index=12,
|
||||
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),
|
||||
],
|
||||
extensions=[
|
||||
],
|
||||
nested_types=[],
|
||||
enum_types=[
|
||||
],
|
||||
serialized_options=None,
|
||||
is_extendable=False,
|
||||
syntax='proto2',
|
||||
extension_ranges=[],
|
||||
oneofs=[
|
||||
],
|
||||
serialized_start=362,
|
||||
serialized_end=657,
|
||||
)
|
||||
|
||||
|
||||
_TIMEDBOXPROTOLIST = _descriptor.Descriptor(
|
||||
name='TimedBoxProtoList',
|
||||
full_name='mediapipe.TimedBoxProtoList',
|
||||
filename=None,
|
||||
file=DESCRIPTOR,
|
||||
containing_type=None,
|
||||
fields=[
|
||||
_descriptor.FieldDescriptor(
|
||||
name='box', full_name='mediapipe.TimedBoxProtoList.box', index=0,
|
||||
number=1, type=11, cpp_type=10, 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),
|
||||
],
|
||||
extensions=[
|
||||
],
|
||||
nested_types=[],
|
||||
enum_types=[
|
||||
],
|
||||
serialized_options=None,
|
||||
is_extendable=False,
|
||||
syntax='proto2',
|
||||
extension_ranges=[],
|
||||
oneofs=[
|
||||
],
|
||||
serialized_start=659,
|
||||
serialized_end=717,
|
||||
)
|
||||
|
||||
_BOXTRACKEROPTIONS.fields_by_name['track_step_options'].message_type = mediapipe_dot_util_dot_tracking_dot_tracking__pb2._TRACKSTEPOPTIONS
|
||||
_TIMEDBOXPROTO.fields_by_name['quad'].message_type = mediapipe_dot_util_dot_tracking_dot_tracking__pb2._MOTIONBOXSTATE_QUAD
|
||||
_TIMEDBOXPROTOLIST.fields_by_name['box'].message_type = _TIMEDBOXPROTO
|
||||
DESCRIPTOR.message_types_by_name['BoxTrackerOptions'] = _BOXTRACKEROPTIONS
|
||||
DESCRIPTOR.message_types_by_name['TimedBoxProto'] = _TIMEDBOXPROTO
|
||||
DESCRIPTOR.message_types_by_name['TimedBoxProtoList'] = _TIMEDBOXPROTOLIST
|
||||
_sym_db.RegisterFileDescriptor(DESCRIPTOR)
|
||||
|
||||
BoxTrackerOptions = _reflection.GeneratedProtocolMessageType('BoxTrackerOptions', (_message.Message,), dict(
|
||||
DESCRIPTOR = _BOXTRACKEROPTIONS,
|
||||
__module__ = 'mediapipe.util.tracking.box_tracker_pb2'
|
||||
# @@protoc_insertion_point(class_scope:mediapipe.BoxTrackerOptions)
|
||||
))
|
||||
_sym_db.RegisterMessage(BoxTrackerOptions)
|
||||
|
||||
TimedBoxProto = _reflection.GeneratedProtocolMessageType('TimedBoxProto', (_message.Message,), dict(
|
||||
DESCRIPTOR = _TIMEDBOXPROTO,
|
||||
__module__ = 'mediapipe.util.tracking.box_tracker_pb2'
|
||||
# @@protoc_insertion_point(class_scope:mediapipe.TimedBoxProto)
|
||||
))
|
||||
_sym_db.RegisterMessage(TimedBoxProto)
|
||||
|
||||
TimedBoxProtoList = _reflection.GeneratedProtocolMessageType('TimedBoxProtoList', (_message.Message,), dict(
|
||||
DESCRIPTOR = _TIMEDBOXPROTOLIST,
|
||||
__module__ = 'mediapipe.util.tracking.box_tracker_pb2'
|
||||
# @@protoc_insertion_point(class_scope:mediapipe.TimedBoxProtoList)
|
||||
))
|
||||
_sym_db.RegisterMessage(TimedBoxProtoList)
|
||||
|
||||
|
||||
DESCRIPTOR._options = None
|
||||
# @@protoc_insertion_point(module_scope)
|
||||
@@ -0,0 +1,368 @@
|
||||
# -*- coding: utf-8 -*-
|
||||
# Generated by the protocol buffer compiler. DO NOT EDIT!
|
||||
# source: mediapipe/util/tracking/camera_motion.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.util.tracking import motion_models_pb2 as mediapipe_dot_util_dot_tracking_dot_motion__models__pb2
|
||||
|
||||
|
||||
DESCRIPTOR = _descriptor.FileDescriptor(
|
||||
name='mediapipe/util/tracking/camera_motion.proto',
|
||||
package='mediapipe',
|
||||
syntax='proto2',
|
||||
serialized_options=None,
|
||||
serialized_pb=_b('\n+mediapipe/util/tracking/camera_motion.proto\x12\tmediapipe\x1a+mediapipe/util/tracking/motion_models.proto\"\x86\x0b\n\x0c\x43\x61meraMotion\x12\x30\n\x0btranslation\x18\x01 \x01(\x0b\x32\x1b.mediapipe.TranslationModel\x12.\n\nsimilarity\x18\x02 \x01(\x0b\x32\x1a.mediapipe.SimilarityModel\x12;\n\x11linear_similarity\x18\x03 \x01(\x0b\x32 .mediapipe.LinearSimilarityModel\x12&\n\x06\x61\x66\x66ine\x18\x04 \x01(\x0b\x32\x16.mediapipe.AffineModel\x12)\n\nhomography\x18\x05 \x01(\x0b\x32\x15.mediapipe.Homography\x12\x38\n\x12mixture_homography\x18\x08 \x01(\x0b\x32\x1c.mediapipe.MixtureHomography\x12\x13\n\x0b\x66rame_width\x18\x1f \x01(\x02\x12\x14\n\x0c\x66rame_height\x18 \x01(\x02\x12\x41\n\x1bmixture_homography_spectrum\x18\x17 \x03(\x0b\x32\x1c.mediapipe.MixtureHomography\x12\x19\n\x11mixture_row_sigma\x18\n \x01(\x02\x12\x1c\n\x11\x61verage_magnitude\x18\x18 \x01(\x02:\x01\x30\x12\x1f\n\x14translation_variance\x18\x19 \x01(\x02:\x01\x30\x12\"\n\x17similarity_inlier_ratio\x18\x1d \x01(\x02:\x01\x30\x12)\n\x1esimilarity_strict_inlier_ratio\x18\x1e \x01(\x02:\x01\x30\x12 \n\x18\x61verage_homography_error\x18\x0b \x01(\x02\x12\"\n\x1ahomography_inlier_coverage\x18\x0c \x01(\x02\x12)\n!homography_strict_inlier_coverage\x18\x16 \x01(\x02\x12\x1f\n\x17mixture_inlier_coverage\x18\r \x03(\x02\x12\x1d\n\x15rolling_shutter_guess\x18\x0e \x01(\x02\x12(\n\x1crolling_shutter_motion_index\x18\x10 \x01(\x05:\x02-1\x12\x17\n\x0foverlay_indices\x18\x11 \x03(\x05\x12\x1a\n\x0eoverlay_domain\x18\x12 \x01(\x05:\x02\x31\x30\x12\x31\n\x04type\x18\x06 \x01(\x0e\x32\x1c.mediapipe.CameraMotion.Type:\x05VALID\x12<\n\x0foverridden_type\x18\x0f \x01(\x0e\x32\x1c.mediapipe.CameraMotion.Type:\x05VALID\x12\x10\n\x05\x66lags\x18\x13 \x01(\x05:\x01\x30\x12\x12\n\nblur_score\x18\x14 \x01(\x02\x12\x14\n\tbluriness\x18\x15 \x01(\x02:\x01\x30\x12#\n\x1b\x66rac_long_features_rejected\x18\x1a \x01(\x02\x12\x19\n\x0etimestamp_usec\x18\x1b \x01(\x03:\x01\x30\x12\x16\n\x0bmatch_frame\x18\x1c \x01(\x05:\x01\x30\"R\n\x04Type\x12\t\n\x05VALID\x10\x00\x12\x12\n\x0eUNSTABLE_HOMOG\x10\x01\x12\x10\n\x0cUNSTABLE_SIM\x10\x02\x12\x0c\n\x08UNSTABLE\x10\x03\x12\x0b\n\x07INVALID\x10\x04\"\xc3\x01\n\x05\x46lags\x12\x16\n\x12\x46LAG_SHOT_BOUNDARY\x10\x01\x12\x15\n\x11\x46LAG_BLURRY_FRAME\x10\x02\x12\x16\n\x12\x46LAG_MAJOR_OVERLAY\x10\x04\x12\x14\n\x10\x46LAG_SHARP_FRAME\x10\x08\x12\x1c\n\x18\x46LAG_SINGULAR_ESTIMATION\x10\x10\x12\x12\n\x0e\x46LAG_SHOT_FADE\x10 \x12\x13\n\x0f\x46LAG_DUPLICATED\x10@\x12\x16\n\x11\x46LAG_CENTER_FRAME\x10\x80\x01*\x04\x08\t\x10\n')
|
||||
,
|
||||
dependencies=[mediapipe_dot_util_dot_tracking_dot_motion__models__pb2.DESCRIPTOR,])
|
||||
|
||||
|
||||
|
||||
_CAMERAMOTION_TYPE = _descriptor.EnumDescriptor(
|
||||
name='Type',
|
||||
full_name='mediapipe.CameraMotion.Type',
|
||||
filename=None,
|
||||
file=DESCRIPTOR,
|
||||
values=[
|
||||
_descriptor.EnumValueDescriptor(
|
||||
name='VALID', index=0, number=0,
|
||||
serialized_options=None,
|
||||
type=None),
|
||||
_descriptor.EnumValueDescriptor(
|
||||
name='UNSTABLE_HOMOG', index=1, number=1,
|
||||
serialized_options=None,
|
||||
type=None),
|
||||
_descriptor.EnumValueDescriptor(
|
||||
name='UNSTABLE_SIM', index=2, number=2,
|
||||
serialized_options=None,
|
||||
type=None),
|
||||
_descriptor.EnumValueDescriptor(
|
||||
name='UNSTABLE', index=3, number=3,
|
||||
serialized_options=None,
|
||||
type=None),
|
||||
_descriptor.EnumValueDescriptor(
|
||||
name='INVALID', index=4, number=4,
|
||||
serialized_options=None,
|
||||
type=None),
|
||||
],
|
||||
containing_type=None,
|
||||
serialized_options=None,
|
||||
serialized_start=1232,
|
||||
serialized_end=1314,
|
||||
)
|
||||
_sym_db.RegisterEnumDescriptor(_CAMERAMOTION_TYPE)
|
||||
|
||||
_CAMERAMOTION_FLAGS = _descriptor.EnumDescriptor(
|
||||
name='Flags',
|
||||
full_name='mediapipe.CameraMotion.Flags',
|
||||
filename=None,
|
||||
file=DESCRIPTOR,
|
||||
values=[
|
||||
_descriptor.EnumValueDescriptor(
|
||||
name='FLAG_SHOT_BOUNDARY', index=0, number=1,
|
||||
serialized_options=None,
|
||||
type=None),
|
||||
_descriptor.EnumValueDescriptor(
|
||||
name='FLAG_BLURRY_FRAME', index=1, number=2,
|
||||
serialized_options=None,
|
||||
type=None),
|
||||
_descriptor.EnumValueDescriptor(
|
||||
name='FLAG_MAJOR_OVERLAY', index=2, number=4,
|
||||
serialized_options=None,
|
||||
type=None),
|
||||
_descriptor.EnumValueDescriptor(
|
||||
name='FLAG_SHARP_FRAME', index=3, number=8,
|
||||
serialized_options=None,
|
||||
type=None),
|
||||
_descriptor.EnumValueDescriptor(
|
||||
name='FLAG_SINGULAR_ESTIMATION', index=4, number=16,
|
||||
serialized_options=None,
|
||||
type=None),
|
||||
_descriptor.EnumValueDescriptor(
|
||||
name='FLAG_SHOT_FADE', index=5, number=32,
|
||||
serialized_options=None,
|
||||
type=None),
|
||||
_descriptor.EnumValueDescriptor(
|
||||
name='FLAG_DUPLICATED', index=6, number=64,
|
||||
serialized_options=None,
|
||||
type=None),
|
||||
_descriptor.EnumValueDescriptor(
|
||||
name='FLAG_CENTER_FRAME', index=7, number=128,
|
||||
serialized_options=None,
|
||||
type=None),
|
||||
],
|
||||
containing_type=None,
|
||||
serialized_options=None,
|
||||
serialized_start=1317,
|
||||
serialized_end=1512,
|
||||
)
|
||||
_sym_db.RegisterEnumDescriptor(_CAMERAMOTION_FLAGS)
|
||||
|
||||
|
||||
_CAMERAMOTION = _descriptor.Descriptor(
|
||||
name='CameraMotion',
|
||||
full_name='mediapipe.CameraMotion',
|
||||
filename=None,
|
||||
file=DESCRIPTOR,
|
||||
containing_type=None,
|
||||
fields=[
|
||||
_descriptor.FieldDescriptor(
|
||||
name='translation', full_name='mediapipe.CameraMotion.translation', 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='similarity', full_name='mediapipe.CameraMotion.similarity', 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='linear_similarity', full_name='mediapipe.CameraMotion.linear_similarity', 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='affine', full_name='mediapipe.CameraMotion.affine', 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),
|
||||
_descriptor.FieldDescriptor(
|
||||
name='homography', full_name='mediapipe.CameraMotion.homography', 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),
|
||||
_descriptor.FieldDescriptor(
|
||||
name='mixture_homography', full_name='mediapipe.CameraMotion.mixture_homography', index=5,
|
||||
number=8, 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='frame_width', full_name='mediapipe.CameraMotion.frame_width', index=6,
|
||||
number=31, 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='frame_height', full_name='mediapipe.CameraMotion.frame_height', index=7,
|
||||
number=32, 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='mixture_homography_spectrum', full_name='mediapipe.CameraMotion.mixture_homography_spectrum', index=8,
|
||||
number=23, type=11, cpp_type=10, 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='mixture_row_sigma', full_name='mediapipe.CameraMotion.mixture_row_sigma', index=9,
|
||||
number=10, 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='average_magnitude', full_name='mediapipe.CameraMotion.average_magnitude', index=10,
|
||||
number=24, 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='translation_variance', full_name='mediapipe.CameraMotion.translation_variance', index=11,
|
||||
number=25, 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='similarity_inlier_ratio', full_name='mediapipe.CameraMotion.similarity_inlier_ratio', index=12,
|
||||
number=29, 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='similarity_strict_inlier_ratio', full_name='mediapipe.CameraMotion.similarity_strict_inlier_ratio', index=13,
|
||||
number=30, 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='average_homography_error', full_name='mediapipe.CameraMotion.average_homography_error', index=14,
|
||||
number=11, 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='homography_inlier_coverage', full_name='mediapipe.CameraMotion.homography_inlier_coverage', index=15,
|
||||
number=12, 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='homography_strict_inlier_coverage', full_name='mediapipe.CameraMotion.homography_strict_inlier_coverage', index=16,
|
||||
number=22, 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='mixture_inlier_coverage', full_name='mediapipe.CameraMotion.mixture_inlier_coverage', index=17,
|
||||
number=13, 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='rolling_shutter_guess', full_name='mediapipe.CameraMotion.rolling_shutter_guess', index=18,
|
||||
number=14, 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='rolling_shutter_motion_index', full_name='mediapipe.CameraMotion.rolling_shutter_motion_index', index=19,
|
||||
number=16, 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='overlay_indices', full_name='mediapipe.CameraMotion.overlay_indices', index=20,
|
||||
number=17, 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='overlay_domain', full_name='mediapipe.CameraMotion.overlay_domain', index=21,
|
||||
number=18, type=5, cpp_type=1, label=1,
|
||||
has_default_value=True, default_value=10,
|
||||
message_type=None, enum_type=None, containing_type=None,
|
||||
is_extension=False, extension_scope=None,
|
||||
serialized_options=None, file=DESCRIPTOR),
|
||||
_descriptor.FieldDescriptor(
|
||||
name='type', full_name='mediapipe.CameraMotion.type', index=22,
|
||||
number=6, 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='overridden_type', full_name='mediapipe.CameraMotion.overridden_type', index=23,
|
||||
number=15, 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='flags', full_name='mediapipe.CameraMotion.flags', index=24,
|
||||
number=19, 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='blur_score', full_name='mediapipe.CameraMotion.blur_score', index=25,
|
||||
number=20, 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='bluriness', full_name='mediapipe.CameraMotion.bluriness', index=26,
|
||||
number=21, 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='frac_long_features_rejected', full_name='mediapipe.CameraMotion.frac_long_features_rejected', index=27,
|
||||
number=26, 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='timestamp_usec', full_name='mediapipe.CameraMotion.timestamp_usec', index=28,
|
||||
number=27, type=3, cpp_type=2, 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='match_frame', full_name='mediapipe.CameraMotion.match_frame', index=29,
|
||||
number=28, 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),
|
||||
],
|
||||
extensions=[
|
||||
],
|
||||
nested_types=[],
|
||||
enum_types=[
|
||||
_CAMERAMOTION_TYPE,
|
||||
_CAMERAMOTION_FLAGS,
|
||||
],
|
||||
serialized_options=None,
|
||||
is_extendable=True,
|
||||
syntax='proto2',
|
||||
extension_ranges=[(9, 10), ],
|
||||
oneofs=[
|
||||
],
|
||||
serialized_start=104,
|
||||
serialized_end=1518,
|
||||
)
|
||||
|
||||
_CAMERAMOTION.fields_by_name['translation'].message_type = mediapipe_dot_util_dot_tracking_dot_motion__models__pb2._TRANSLATIONMODEL
|
||||
_CAMERAMOTION.fields_by_name['similarity'].message_type = mediapipe_dot_util_dot_tracking_dot_motion__models__pb2._SIMILARITYMODEL
|
||||
_CAMERAMOTION.fields_by_name['linear_similarity'].message_type = mediapipe_dot_util_dot_tracking_dot_motion__models__pb2._LINEARSIMILARITYMODEL
|
||||
_CAMERAMOTION.fields_by_name['affine'].message_type = mediapipe_dot_util_dot_tracking_dot_motion__models__pb2._AFFINEMODEL
|
||||
_CAMERAMOTION.fields_by_name['homography'].message_type = mediapipe_dot_util_dot_tracking_dot_motion__models__pb2._HOMOGRAPHY
|
||||
_CAMERAMOTION.fields_by_name['mixture_homography'].message_type = mediapipe_dot_util_dot_tracking_dot_motion__models__pb2._MIXTUREHOMOGRAPHY
|
||||
_CAMERAMOTION.fields_by_name['mixture_homography_spectrum'].message_type = mediapipe_dot_util_dot_tracking_dot_motion__models__pb2._MIXTUREHOMOGRAPHY
|
||||
_CAMERAMOTION.fields_by_name['type'].enum_type = _CAMERAMOTION_TYPE
|
||||
_CAMERAMOTION.fields_by_name['overridden_type'].enum_type = _CAMERAMOTION_TYPE
|
||||
_CAMERAMOTION_TYPE.containing_type = _CAMERAMOTION
|
||||
_CAMERAMOTION_FLAGS.containing_type = _CAMERAMOTION
|
||||
DESCRIPTOR.message_types_by_name['CameraMotion'] = _CAMERAMOTION
|
||||
_sym_db.RegisterFileDescriptor(DESCRIPTOR)
|
||||
|
||||
CameraMotion = _reflection.GeneratedProtocolMessageType('CameraMotion', (_message.Message,), dict(
|
||||
DESCRIPTOR = _CAMERAMOTION,
|
||||
__module__ = 'mediapipe.util.tracking.camera_motion_pb2'
|
||||
# @@protoc_insertion_point(class_scope:mediapipe.CameraMotion)
|
||||
))
|
||||
_sym_db.RegisterMessage(CameraMotion)
|
||||
|
||||
|
||||
# @@protoc_insertion_point(module_scope)
|
||||
@@ -0,0 +1,763 @@
|
||||
# -*- coding: utf-8 -*-
|
||||
# Generated by the protocol buffer compiler. DO NOT EDIT!
|
||||
# source: mediapipe/util/tracking/flow_packager.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.util.tracking import motion_models_pb2 as mediapipe_dot_util_dot_tracking_dot_motion__models__pb2
|
||||
from mediapipe.util.tracking import region_flow_pb2 as mediapipe_dot_util_dot_tracking_dot_region__flow__pb2
|
||||
|
||||
|
||||
DESCRIPTOR = _descriptor.FileDescriptor(
|
||||
name='mediapipe/util/tracking/flow_packager.proto',
|
||||
package='mediapipe',
|
||||
syntax='proto2',
|
||||
serialized_options=None,
|
||||
serialized_pb=_b('\n+mediapipe/util/tracking/flow_packager.proto\x12\tmediapipe\x1a+mediapipe/util/tracking/motion_models.proto\x1a)mediapipe/util/tracking/region_flow.proto\"\xb1\x05\n\x0cTrackingData\x12\x16\n\x0b\x66rame_flags\x18\x01 \x01(\x05:\x01\x30\x12\x14\n\x0c\x64omain_width\x18\x02 \x01(\x05\x12\x15\n\rdomain_height\x18\x03 \x01(\x05\x12\x17\n\x0c\x66rame_aspect\x18\x06 \x01(\x02:\x01\x31\x12/\n\x10\x62\x61\x63kground_model\x18\x04 \x01(\x0b\x32\x15.mediapipe.Homography\x12\x37\n\x0bmotion_data\x18\x05 \x01(\x0b\x32\".mediapipe.TrackingData.MotionData\x12\x1c\n\x14global_feature_count\x18\x07 \x01(\r\x12 \n\x18\x61verage_motion_magnitude\x18\x08 \x01(\x02\x1a\xeb\x01\n\nMotionData\x12\x14\n\x0cnum_elements\x18\x01 \x01(\x05\x12\x17\n\x0bvector_data\x18\x02 \x03(\x02\x42\x02\x10\x01\x12\x14\n\x08track_id\x18\x03 \x03(\x05\x42\x02\x10\x01\x12\x17\n\x0brow_indices\x18\x04 \x03(\x05\x42\x02\x10\x01\x12\x16\n\ncol_starts\x18\x05 \x03(\x05\x42\x02\x10\x01\x12?\n\x13\x66\x65\x61ture_descriptors\x18\x06 \x03(\x0b\x32\".mediapipe.BinaryFeatureDescriptor\x12&\n\x1e\x61\x63tively_discarded_tracked_ids\x18\x07 \x03(\x05\"\xaa\x01\n\nFrameFlags\x12\x19\n\x15\x46LAG_PROFILE_BASELINE\x10\x00\x12\x15\n\x11\x46LAG_PROFILE_HIGH\x10\x01\x12\x1e\n\x1a\x46LAG_HIGH_FIDELITY_VECTORS\x10\x02\x12\x1c\n\x18\x46LAG_BACKGROUND_UNSTABLE\x10\x04\x12\x13\n\x0f\x46LAG_DUPLICATED\x10\x08\x12\x17\n\x13\x46LAG_CHUNK_BOUNDARY\x10\x10\"\xfb\x01\n\x11TrackingDataChunk\x12/\n\x04item\x18\x01 \x03(\x0b\x32!.mediapipe.TrackingDataChunk.Item\x12\x19\n\nlast_chunk\x18\x02 \x01(\x08:\x05\x66\x61lse\x12\x1a\n\x0b\x66irst_chunk\x18\x03 \x01(\x08:\x05\x66\x61lse\x1a~\n\x04Item\x12.\n\rtracking_data\x18\x01 \x01(\x0b\x32\x17.mediapipe.TrackingData\x12\x11\n\tframe_idx\x18\x02 \x01(\x05\x12\x16\n\x0etimestamp_usec\x18\x03 \x01(\x03\x12\x1b\n\x13prev_timestamp_usec\x18\x04 \x01(\x03\"\"\n\x12\x42inaryTrackingData\x12\x0c\n\x04\x64\x61ta\x18\x01 \x01(\x0c\"\x8a\x01\n\x08MetaData\x12\x12\n\nnum_frames\x18\x02 \x01(\x07\x12\x36\n\rtrack_offsets\x18\x03 \x03(\x0b\x32\x1f.mediapipe.MetaData.TrackOffset\x1a\x32\n\x0bTrackOffset\x12\x0c\n\x04msec\x18\x01 \x01(\x07\x12\x15\n\rstream_offset\x18\x02 \x01(\x07\"S\n\x11TrackingContainer\x12\x0e\n\x06header\x18\x01 \x01(\t\x12\x12\n\x07version\x18\x02 \x01(\x07:\x01\x31\x12\x0c\n\x04size\x18\x03 \x01(\x07\x12\x0c\n\x04\x64\x61ta\x18\x04 \x01(\x0c\"\xad\x01\n\x17TrackingContainerFormat\x12/\n\tmeta_data\x18\x01 \x01(\x0b\x32\x1c.mediapipe.TrackingContainer\x12\x30\n\ntrack_data\x18\x02 \x03(\x0b\x32\x1c.mediapipe.TrackingContainer\x12/\n\tterm_data\x18\x03 \x01(\x0b\x32\x1c.mediapipe.TrackingContainer\"s\n\x16TrackingContainerProto\x12&\n\tmeta_data\x18\x01 \x01(\x0b\x32\x13.mediapipe.MetaData\x12\x31\n\ntrack_data\x18\x02 \x03(\x0b\x32\x1d.mediapipe.BinaryTrackingData\"\xc1\x02\n\x13\x46lowPackagerOptions\x12\x19\n\x0c\x64omain_width\x18\x01 \x01(\x05:\x03\x32\x35\x36\x12\x1a\n\rdomain_height\x18\x02 \x01(\x05:\x03\x31\x39\x32\x12*\n\x1c\x62inary_tracking_data_support\x18\x06 \x01(\x08:\x04true\x12\x1f\n\x10use_high_profile\x18\x03 \x01(\x08:\x05\x66\x61lse\x12(\n\x1ahigh_fidelity_16bit_encode\x18\x04 \x01(\x08:\x04true\x12)\n\x1chigh_profile_reuse_threshold\x18\x05 \x01(\x02:\x03\x30.5\"Q\n\x13HighProfileEncoding\x12\x11\n\x0c\x41\x44VANCE_FLAG\x10\x80\x01\x12\x17\n\x13\x44OUBLE_INDEX_ENCODE\x10@\x12\x0e\n\nINDEX_MASK\x10?')
|
||||
,
|
||||
dependencies=[mediapipe_dot_util_dot_tracking_dot_motion__models__pb2.DESCRIPTOR,mediapipe_dot_util_dot_tracking_dot_region__flow__pb2.DESCRIPTOR,])
|
||||
|
||||
|
||||
|
||||
_TRACKINGDATA_FRAMEFLAGS = _descriptor.EnumDescriptor(
|
||||
name='FrameFlags',
|
||||
full_name='mediapipe.TrackingData.FrameFlags',
|
||||
filename=None,
|
||||
file=DESCRIPTOR,
|
||||
values=[
|
||||
_descriptor.EnumValueDescriptor(
|
||||
name='FLAG_PROFILE_BASELINE', index=0, number=0,
|
||||
serialized_options=None,
|
||||
type=None),
|
||||
_descriptor.EnumValueDescriptor(
|
||||
name='FLAG_PROFILE_HIGH', index=1, number=1,
|
||||
serialized_options=None,
|
||||
type=None),
|
||||
_descriptor.EnumValueDescriptor(
|
||||
name='FLAG_HIGH_FIDELITY_VECTORS', index=2, number=2,
|
||||
serialized_options=None,
|
||||
type=None),
|
||||
_descriptor.EnumValueDescriptor(
|
||||
name='FLAG_BACKGROUND_UNSTABLE', index=3, number=4,
|
||||
serialized_options=None,
|
||||
type=None),
|
||||
_descriptor.EnumValueDescriptor(
|
||||
name='FLAG_DUPLICATED', index=4, number=8,
|
||||
serialized_options=None,
|
||||
type=None),
|
||||
_descriptor.EnumValueDescriptor(
|
||||
name='FLAG_CHUNK_BOUNDARY', index=5, number=16,
|
||||
serialized_options=None,
|
||||
type=None),
|
||||
],
|
||||
containing_type=None,
|
||||
serialized_options=None,
|
||||
serialized_start=666,
|
||||
serialized_end=836,
|
||||
)
|
||||
_sym_db.RegisterEnumDescriptor(_TRACKINGDATA_FRAMEFLAGS)
|
||||
|
||||
_FLOWPACKAGEROPTIONS_HIGHPROFILEENCODING = _descriptor.EnumDescriptor(
|
||||
name='HighProfileEncoding',
|
||||
full_name='mediapipe.FlowPackagerOptions.HighProfileEncoding',
|
||||
filename=None,
|
||||
file=DESCRIPTOR,
|
||||
values=[
|
||||
_descriptor.EnumValueDescriptor(
|
||||
name='ADVANCE_FLAG', index=0, number=128,
|
||||
serialized_options=None,
|
||||
type=None),
|
||||
_descriptor.EnumValueDescriptor(
|
||||
name='DOUBLE_INDEX_ENCODE', index=1, number=64,
|
||||
serialized_options=None,
|
||||
type=None),
|
||||
_descriptor.EnumValueDescriptor(
|
||||
name='INDEX_MASK', index=2, number=63,
|
||||
serialized_options=None,
|
||||
type=None),
|
||||
],
|
||||
containing_type=None,
|
||||
serialized_options=None,
|
||||
serialized_start=1888,
|
||||
serialized_end=1969,
|
||||
)
|
||||
_sym_db.RegisterEnumDescriptor(_FLOWPACKAGEROPTIONS_HIGHPROFILEENCODING)
|
||||
|
||||
|
||||
_TRACKINGDATA_MOTIONDATA = _descriptor.Descriptor(
|
||||
name='MotionData',
|
||||
full_name='mediapipe.TrackingData.MotionData',
|
||||
filename=None,
|
||||
file=DESCRIPTOR,
|
||||
containing_type=None,
|
||||
fields=[
|
||||
_descriptor.FieldDescriptor(
|
||||
name='num_elements', full_name='mediapipe.TrackingData.MotionData.num_elements', index=0,
|
||||
number=1, 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='vector_data', full_name='mediapipe.TrackingData.MotionData.vector_data', index=1,
|
||||
number=2, 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=_b('\020\001'), file=DESCRIPTOR),
|
||||
_descriptor.FieldDescriptor(
|
||||
name='track_id', full_name='mediapipe.TrackingData.MotionData.track_id', index=2,
|
||||
number=3, 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=_b('\020\001'), file=DESCRIPTOR),
|
||||
_descriptor.FieldDescriptor(
|
||||
name='row_indices', full_name='mediapipe.TrackingData.MotionData.row_indices', index=3,
|
||||
number=4, 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=_b('\020\001'), file=DESCRIPTOR),
|
||||
_descriptor.FieldDescriptor(
|
||||
name='col_starts', full_name='mediapipe.TrackingData.MotionData.col_starts', index=4,
|
||||
number=5, 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=_b('\020\001'), file=DESCRIPTOR),
|
||||
_descriptor.FieldDescriptor(
|
||||
name='feature_descriptors', full_name='mediapipe.TrackingData.MotionData.feature_descriptors', index=5,
|
||||
number=6, type=11, cpp_type=10, 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='actively_discarded_tracked_ids', full_name='mediapipe.TrackingData.MotionData.actively_discarded_tracked_ids', index=6,
|
||||
number=7, 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),
|
||||
],
|
||||
extensions=[
|
||||
],
|
||||
nested_types=[],
|
||||
enum_types=[
|
||||
],
|
||||
serialized_options=None,
|
||||
is_extendable=False,
|
||||
syntax='proto2',
|
||||
extension_ranges=[],
|
||||
oneofs=[
|
||||
],
|
||||
serialized_start=428,
|
||||
serialized_end=663,
|
||||
)
|
||||
|
||||
_TRACKINGDATA = _descriptor.Descriptor(
|
||||
name='TrackingData',
|
||||
full_name='mediapipe.TrackingData',
|
||||
filename=None,
|
||||
file=DESCRIPTOR,
|
||||
containing_type=None,
|
||||
fields=[
|
||||
_descriptor.FieldDescriptor(
|
||||
name='frame_flags', full_name='mediapipe.TrackingData.frame_flags', index=0,
|
||||
number=1, 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='domain_width', full_name='mediapipe.TrackingData.domain_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='domain_height', full_name='mediapipe.TrackingData.domain_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='frame_aspect', full_name='mediapipe.TrackingData.frame_aspect', index=3,
|
||||
number=6, 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='background_model', full_name='mediapipe.TrackingData.background_model', index=4,
|
||||
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),
|
||||
_descriptor.FieldDescriptor(
|
||||
name='motion_data', full_name='mediapipe.TrackingData.motion_data', index=5,
|
||||
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),
|
||||
_descriptor.FieldDescriptor(
|
||||
name='global_feature_count', full_name='mediapipe.TrackingData.global_feature_count', index=6,
|
||||
number=7, type=13, cpp_type=3, 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='average_motion_magnitude', full_name='mediapipe.TrackingData.average_motion_magnitude', index=7,
|
||||
number=8, 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=[_TRACKINGDATA_MOTIONDATA, ],
|
||||
enum_types=[
|
||||
_TRACKINGDATA_FRAMEFLAGS,
|
||||
],
|
||||
serialized_options=None,
|
||||
is_extendable=False,
|
||||
syntax='proto2',
|
||||
extension_ranges=[],
|
||||
oneofs=[
|
||||
],
|
||||
serialized_start=147,
|
||||
serialized_end=836,
|
||||
)
|
||||
|
||||
|
||||
_TRACKINGDATACHUNK_ITEM = _descriptor.Descriptor(
|
||||
name='Item',
|
||||
full_name='mediapipe.TrackingDataChunk.Item',
|
||||
filename=None,
|
||||
file=DESCRIPTOR,
|
||||
containing_type=None,
|
||||
fields=[
|
||||
_descriptor.FieldDescriptor(
|
||||
name='tracking_data', full_name='mediapipe.TrackingDataChunk.Item.tracking_data', 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='frame_idx', full_name='mediapipe.TrackingDataChunk.Item.frame_idx', 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='timestamp_usec', full_name='mediapipe.TrackingDataChunk.Item.timestamp_usec', index=2,
|
||||
number=3, type=3, cpp_type=2, 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='prev_timestamp_usec', full_name='mediapipe.TrackingDataChunk.Item.prev_timestamp_usec', index=3,
|
||||
number=4, type=3, cpp_type=2, 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),
|
||||
],
|
||||
extensions=[
|
||||
],
|
||||
nested_types=[],
|
||||
enum_types=[
|
||||
],
|
||||
serialized_options=None,
|
||||
is_extendable=False,
|
||||
syntax='proto2',
|
||||
extension_ranges=[],
|
||||
oneofs=[
|
||||
],
|
||||
serialized_start=964,
|
||||
serialized_end=1090,
|
||||
)
|
||||
|
||||
_TRACKINGDATACHUNK = _descriptor.Descriptor(
|
||||
name='TrackingDataChunk',
|
||||
full_name='mediapipe.TrackingDataChunk',
|
||||
filename=None,
|
||||
file=DESCRIPTOR,
|
||||
containing_type=None,
|
||||
fields=[
|
||||
_descriptor.FieldDescriptor(
|
||||
name='item', full_name='mediapipe.TrackingDataChunk.item', index=0,
|
||||
number=1, type=11, cpp_type=10, 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='last_chunk', full_name='mediapipe.TrackingDataChunk.last_chunk', 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=None, file=DESCRIPTOR),
|
||||
_descriptor.FieldDescriptor(
|
||||
name='first_chunk', full_name='mediapipe.TrackingDataChunk.first_chunk', 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=None, file=DESCRIPTOR),
|
||||
],
|
||||
extensions=[
|
||||
],
|
||||
nested_types=[_TRACKINGDATACHUNK_ITEM, ],
|
||||
enum_types=[
|
||||
],
|
||||
serialized_options=None,
|
||||
is_extendable=False,
|
||||
syntax='proto2',
|
||||
extension_ranges=[],
|
||||
oneofs=[
|
||||
],
|
||||
serialized_start=839,
|
||||
serialized_end=1090,
|
||||
)
|
||||
|
||||
|
||||
_BINARYTRACKINGDATA = _descriptor.Descriptor(
|
||||
name='BinaryTrackingData',
|
||||
full_name='mediapipe.BinaryTrackingData',
|
||||
filename=None,
|
||||
file=DESCRIPTOR,
|
||||
containing_type=None,
|
||||
fields=[
|
||||
_descriptor.FieldDescriptor(
|
||||
name='data', full_name='mediapipe.BinaryTrackingData.data', index=0,
|
||||
number=1, type=12, cpp_type=9, label=1,
|
||||
has_default_value=False, default_value=_b(""),
|
||||
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=1092,
|
||||
serialized_end=1126,
|
||||
)
|
||||
|
||||
|
||||
_METADATA_TRACKOFFSET = _descriptor.Descriptor(
|
||||
name='TrackOffset',
|
||||
full_name='mediapipe.MetaData.TrackOffset',
|
||||
filename=None,
|
||||
file=DESCRIPTOR,
|
||||
containing_type=None,
|
||||
fields=[
|
||||
_descriptor.FieldDescriptor(
|
||||
name='msec', full_name='mediapipe.MetaData.TrackOffset.msec', index=0,
|
||||
number=1, type=7, cpp_type=3, 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='stream_offset', full_name='mediapipe.MetaData.TrackOffset.stream_offset', index=1,
|
||||
number=2, type=7, cpp_type=3, 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),
|
||||
],
|
||||
extensions=[
|
||||
],
|
||||
nested_types=[],
|
||||
enum_types=[
|
||||
],
|
||||
serialized_options=None,
|
||||
is_extendable=False,
|
||||
syntax='proto2',
|
||||
extension_ranges=[],
|
||||
oneofs=[
|
||||
],
|
||||
serialized_start=1217,
|
||||
serialized_end=1267,
|
||||
)
|
||||
|
||||
_METADATA = _descriptor.Descriptor(
|
||||
name='MetaData',
|
||||
full_name='mediapipe.MetaData',
|
||||
filename=None,
|
||||
file=DESCRIPTOR,
|
||||
containing_type=None,
|
||||
fields=[
|
||||
_descriptor.FieldDescriptor(
|
||||
name='num_frames', full_name='mediapipe.MetaData.num_frames', index=0,
|
||||
number=2, type=7, cpp_type=3, 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='track_offsets', full_name='mediapipe.MetaData.track_offsets', index=1,
|
||||
number=3, type=11, cpp_type=10, 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),
|
||||
],
|
||||
extensions=[
|
||||
],
|
||||
nested_types=[_METADATA_TRACKOFFSET, ],
|
||||
enum_types=[
|
||||
],
|
||||
serialized_options=None,
|
||||
is_extendable=False,
|
||||
syntax='proto2',
|
||||
extension_ranges=[],
|
||||
oneofs=[
|
||||
],
|
||||
serialized_start=1129,
|
||||
serialized_end=1267,
|
||||
)
|
||||
|
||||
|
||||
_TRACKINGCONTAINER = _descriptor.Descriptor(
|
||||
name='TrackingContainer',
|
||||
full_name='mediapipe.TrackingContainer',
|
||||
filename=None,
|
||||
file=DESCRIPTOR,
|
||||
containing_type=None,
|
||||
fields=[
|
||||
_descriptor.FieldDescriptor(
|
||||
name='header', full_name='mediapipe.TrackingContainer.header', 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='version', full_name='mediapipe.TrackingContainer.version', index=1,
|
||||
number=2, type=7, cpp_type=3, 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='size', full_name='mediapipe.TrackingContainer.size', index=2,
|
||||
number=3, type=7, cpp_type=3, 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='data', full_name='mediapipe.TrackingContainer.data', index=3,
|
||||
number=4, type=12, cpp_type=9, label=1,
|
||||
has_default_value=False, default_value=_b(""),
|
||||
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=1269,
|
||||
serialized_end=1352,
|
||||
)
|
||||
|
||||
|
||||
_TRACKINGCONTAINERFORMAT = _descriptor.Descriptor(
|
||||
name='TrackingContainerFormat',
|
||||
full_name='mediapipe.TrackingContainerFormat',
|
||||
filename=None,
|
||||
file=DESCRIPTOR,
|
||||
containing_type=None,
|
||||
fields=[
|
||||
_descriptor.FieldDescriptor(
|
||||
name='meta_data', full_name='mediapipe.TrackingContainerFormat.meta_data', 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='track_data', full_name='mediapipe.TrackingContainerFormat.track_data', index=1,
|
||||
number=2, type=11, cpp_type=10, 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='term_data', full_name='mediapipe.TrackingContainerFormat.term_data', 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),
|
||||
],
|
||||
extensions=[
|
||||
],
|
||||
nested_types=[],
|
||||
enum_types=[
|
||||
],
|
||||
serialized_options=None,
|
||||
is_extendable=False,
|
||||
syntax='proto2',
|
||||
extension_ranges=[],
|
||||
oneofs=[
|
||||
],
|
||||
serialized_start=1355,
|
||||
serialized_end=1528,
|
||||
)
|
||||
|
||||
|
||||
_TRACKINGCONTAINERPROTO = _descriptor.Descriptor(
|
||||
name='TrackingContainerProto',
|
||||
full_name='mediapipe.TrackingContainerProto',
|
||||
filename=None,
|
||||
file=DESCRIPTOR,
|
||||
containing_type=None,
|
||||
fields=[
|
||||
_descriptor.FieldDescriptor(
|
||||
name='meta_data', full_name='mediapipe.TrackingContainerProto.meta_data', 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='track_data', full_name='mediapipe.TrackingContainerProto.track_data', index=1,
|
||||
number=2, type=11, cpp_type=10, 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),
|
||||
],
|
||||
extensions=[
|
||||
],
|
||||
nested_types=[],
|
||||
enum_types=[
|
||||
],
|
||||
serialized_options=None,
|
||||
is_extendable=False,
|
||||
syntax='proto2',
|
||||
extension_ranges=[],
|
||||
oneofs=[
|
||||
],
|
||||
serialized_start=1530,
|
||||
serialized_end=1645,
|
||||
)
|
||||
|
||||
|
||||
_FLOWPACKAGEROPTIONS = _descriptor.Descriptor(
|
||||
name='FlowPackagerOptions',
|
||||
full_name='mediapipe.FlowPackagerOptions',
|
||||
filename=None,
|
||||
file=DESCRIPTOR,
|
||||
containing_type=None,
|
||||
fields=[
|
||||
_descriptor.FieldDescriptor(
|
||||
name='domain_width', full_name='mediapipe.FlowPackagerOptions.domain_width', index=0,
|
||||
number=1, type=5, cpp_type=1, label=1,
|
||||
has_default_value=True, default_value=256,
|
||||
message_type=None, enum_type=None, containing_type=None,
|
||||
is_extension=False, extension_scope=None,
|
||||
serialized_options=None, file=DESCRIPTOR),
|
||||
_descriptor.FieldDescriptor(
|
||||
name='domain_height', full_name='mediapipe.FlowPackagerOptions.domain_height', index=1,
|
||||
number=2, type=5, cpp_type=1, label=1,
|
||||
has_default_value=True, default_value=192,
|
||||
message_type=None, enum_type=None, containing_type=None,
|
||||
is_extension=False, extension_scope=None,
|
||||
serialized_options=None, file=DESCRIPTOR),
|
||||
_descriptor.FieldDescriptor(
|
||||
name='binary_tracking_data_support', full_name='mediapipe.FlowPackagerOptions.binary_tracking_data_support', index=2,
|
||||
number=6, 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_high_profile', full_name='mediapipe.FlowPackagerOptions.use_high_profile', index=3,
|
||||
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=None, file=DESCRIPTOR),
|
||||
_descriptor.FieldDescriptor(
|
||||
name='high_fidelity_16bit_encode', full_name='mediapipe.FlowPackagerOptions.high_fidelity_16bit_encode', index=4,
|
||||
number=4, 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='high_profile_reuse_threshold', full_name='mediapipe.FlowPackagerOptions.high_profile_reuse_threshold', index=5,
|
||||
number=5, type=2, cpp_type=6, label=1,
|
||||
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),
|
||||
],
|
||||
extensions=[
|
||||
],
|
||||
nested_types=[],
|
||||
enum_types=[
|
||||
_FLOWPACKAGEROPTIONS_HIGHPROFILEENCODING,
|
||||
],
|
||||
serialized_options=None,
|
||||
is_extendable=False,
|
||||
syntax='proto2',
|
||||
extension_ranges=[],
|
||||
oneofs=[
|
||||
],
|
||||
serialized_start=1648,
|
||||
serialized_end=1969,
|
||||
)
|
||||
|
||||
_TRACKINGDATA_MOTIONDATA.fields_by_name['feature_descriptors'].message_type = mediapipe_dot_util_dot_tracking_dot_region__flow__pb2._BINARYFEATUREDESCRIPTOR
|
||||
_TRACKINGDATA_MOTIONDATA.containing_type = _TRACKINGDATA
|
||||
_TRACKINGDATA.fields_by_name['background_model'].message_type = mediapipe_dot_util_dot_tracking_dot_motion__models__pb2._HOMOGRAPHY
|
||||
_TRACKINGDATA.fields_by_name['motion_data'].message_type = _TRACKINGDATA_MOTIONDATA
|
||||
_TRACKINGDATA_FRAMEFLAGS.containing_type = _TRACKINGDATA
|
||||
_TRACKINGDATACHUNK_ITEM.fields_by_name['tracking_data'].message_type = _TRACKINGDATA
|
||||
_TRACKINGDATACHUNK_ITEM.containing_type = _TRACKINGDATACHUNK
|
||||
_TRACKINGDATACHUNK.fields_by_name['item'].message_type = _TRACKINGDATACHUNK_ITEM
|
||||
_METADATA_TRACKOFFSET.containing_type = _METADATA
|
||||
_METADATA.fields_by_name['track_offsets'].message_type = _METADATA_TRACKOFFSET
|
||||
_TRACKINGCONTAINERFORMAT.fields_by_name['meta_data'].message_type = _TRACKINGCONTAINER
|
||||
_TRACKINGCONTAINERFORMAT.fields_by_name['track_data'].message_type = _TRACKINGCONTAINER
|
||||
_TRACKINGCONTAINERFORMAT.fields_by_name['term_data'].message_type = _TRACKINGCONTAINER
|
||||
_TRACKINGCONTAINERPROTO.fields_by_name['meta_data'].message_type = _METADATA
|
||||
_TRACKINGCONTAINERPROTO.fields_by_name['track_data'].message_type = _BINARYTRACKINGDATA
|
||||
_FLOWPACKAGEROPTIONS_HIGHPROFILEENCODING.containing_type = _FLOWPACKAGEROPTIONS
|
||||
DESCRIPTOR.message_types_by_name['TrackingData'] = _TRACKINGDATA
|
||||
DESCRIPTOR.message_types_by_name['TrackingDataChunk'] = _TRACKINGDATACHUNK
|
||||
DESCRIPTOR.message_types_by_name['BinaryTrackingData'] = _BINARYTRACKINGDATA
|
||||
DESCRIPTOR.message_types_by_name['MetaData'] = _METADATA
|
||||
DESCRIPTOR.message_types_by_name['TrackingContainer'] = _TRACKINGCONTAINER
|
||||
DESCRIPTOR.message_types_by_name['TrackingContainerFormat'] = _TRACKINGCONTAINERFORMAT
|
||||
DESCRIPTOR.message_types_by_name['TrackingContainerProto'] = _TRACKINGCONTAINERPROTO
|
||||
DESCRIPTOR.message_types_by_name['FlowPackagerOptions'] = _FLOWPACKAGEROPTIONS
|
||||
_sym_db.RegisterFileDescriptor(DESCRIPTOR)
|
||||
|
||||
TrackingData = _reflection.GeneratedProtocolMessageType('TrackingData', (_message.Message,), dict(
|
||||
|
||||
MotionData = _reflection.GeneratedProtocolMessageType('MotionData', (_message.Message,), dict(
|
||||
DESCRIPTOR = _TRACKINGDATA_MOTIONDATA,
|
||||
__module__ = 'mediapipe.util.tracking.flow_packager_pb2'
|
||||
# @@protoc_insertion_point(class_scope:mediapipe.TrackingData.MotionData)
|
||||
))
|
||||
,
|
||||
DESCRIPTOR = _TRACKINGDATA,
|
||||
__module__ = 'mediapipe.util.tracking.flow_packager_pb2'
|
||||
# @@protoc_insertion_point(class_scope:mediapipe.TrackingData)
|
||||
))
|
||||
_sym_db.RegisterMessage(TrackingData)
|
||||
_sym_db.RegisterMessage(TrackingData.MotionData)
|
||||
|
||||
TrackingDataChunk = _reflection.GeneratedProtocolMessageType('TrackingDataChunk', (_message.Message,), dict(
|
||||
|
||||
Item = _reflection.GeneratedProtocolMessageType('Item', (_message.Message,), dict(
|
||||
DESCRIPTOR = _TRACKINGDATACHUNK_ITEM,
|
||||
__module__ = 'mediapipe.util.tracking.flow_packager_pb2'
|
||||
# @@protoc_insertion_point(class_scope:mediapipe.TrackingDataChunk.Item)
|
||||
))
|
||||
,
|
||||
DESCRIPTOR = _TRACKINGDATACHUNK,
|
||||
__module__ = 'mediapipe.util.tracking.flow_packager_pb2'
|
||||
# @@protoc_insertion_point(class_scope:mediapipe.TrackingDataChunk)
|
||||
))
|
||||
_sym_db.RegisterMessage(TrackingDataChunk)
|
||||
_sym_db.RegisterMessage(TrackingDataChunk.Item)
|
||||
|
||||
BinaryTrackingData = _reflection.GeneratedProtocolMessageType('BinaryTrackingData', (_message.Message,), dict(
|
||||
DESCRIPTOR = _BINARYTRACKINGDATA,
|
||||
__module__ = 'mediapipe.util.tracking.flow_packager_pb2'
|
||||
# @@protoc_insertion_point(class_scope:mediapipe.BinaryTrackingData)
|
||||
))
|
||||
_sym_db.RegisterMessage(BinaryTrackingData)
|
||||
|
||||
MetaData = _reflection.GeneratedProtocolMessageType('MetaData', (_message.Message,), dict(
|
||||
|
||||
TrackOffset = _reflection.GeneratedProtocolMessageType('TrackOffset', (_message.Message,), dict(
|
||||
DESCRIPTOR = _METADATA_TRACKOFFSET,
|
||||
__module__ = 'mediapipe.util.tracking.flow_packager_pb2'
|
||||
# @@protoc_insertion_point(class_scope:mediapipe.MetaData.TrackOffset)
|
||||
))
|
||||
,
|
||||
DESCRIPTOR = _METADATA,
|
||||
__module__ = 'mediapipe.util.tracking.flow_packager_pb2'
|
||||
# @@protoc_insertion_point(class_scope:mediapipe.MetaData)
|
||||
))
|
||||
_sym_db.RegisterMessage(MetaData)
|
||||
_sym_db.RegisterMessage(MetaData.TrackOffset)
|
||||
|
||||
TrackingContainer = _reflection.GeneratedProtocolMessageType('TrackingContainer', (_message.Message,), dict(
|
||||
DESCRIPTOR = _TRACKINGCONTAINER,
|
||||
__module__ = 'mediapipe.util.tracking.flow_packager_pb2'
|
||||
# @@protoc_insertion_point(class_scope:mediapipe.TrackingContainer)
|
||||
))
|
||||
_sym_db.RegisterMessage(TrackingContainer)
|
||||
|
||||
TrackingContainerFormat = _reflection.GeneratedProtocolMessageType('TrackingContainerFormat', (_message.Message,), dict(
|
||||
DESCRIPTOR = _TRACKINGCONTAINERFORMAT,
|
||||
__module__ = 'mediapipe.util.tracking.flow_packager_pb2'
|
||||
# @@protoc_insertion_point(class_scope:mediapipe.TrackingContainerFormat)
|
||||
))
|
||||
_sym_db.RegisterMessage(TrackingContainerFormat)
|
||||
|
||||
TrackingContainerProto = _reflection.GeneratedProtocolMessageType('TrackingContainerProto', (_message.Message,), dict(
|
||||
DESCRIPTOR = _TRACKINGCONTAINERPROTO,
|
||||
__module__ = 'mediapipe.util.tracking.flow_packager_pb2'
|
||||
# @@protoc_insertion_point(class_scope:mediapipe.TrackingContainerProto)
|
||||
))
|
||||
_sym_db.RegisterMessage(TrackingContainerProto)
|
||||
|
||||
FlowPackagerOptions = _reflection.GeneratedProtocolMessageType('FlowPackagerOptions', (_message.Message,), dict(
|
||||
DESCRIPTOR = _FLOWPACKAGEROPTIONS,
|
||||
__module__ = 'mediapipe.util.tracking.flow_packager_pb2'
|
||||
# @@protoc_insertion_point(class_scope:mediapipe.FlowPackagerOptions)
|
||||
))
|
||||
_sym_db.RegisterMessage(FlowPackagerOptions)
|
||||
|
||||
|
||||
_TRACKINGDATA_MOTIONDATA.fields_by_name['vector_data']._options = None
|
||||
_TRACKINGDATA_MOTIONDATA.fields_by_name['track_id']._options = None
|
||||
_TRACKINGDATA_MOTIONDATA.fields_by_name['row_indices']._options = None
|
||||
_TRACKINGDATA_MOTIONDATA.fields_by_name['col_starts']._options = None
|
||||
# @@protoc_insertion_point(module_scope)
|
||||
@@ -0,0 +1,272 @@
|
||||
# -*- coding: utf-8 -*-
|
||||
# Generated by the protocol buffer compiler. DO NOT EDIT!
|
||||
# source: mediapipe/util/tracking/frame_selection.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.util.tracking import camera_motion_pb2 as mediapipe_dot_util_dot_tracking_dot_camera__motion__pb2
|
||||
from mediapipe.util.tracking import frame_selection_solution_evaluator_pb2 as mediapipe_dot_util_dot_tracking_dot_frame__selection__solution__evaluator__pb2
|
||||
from mediapipe.util.tracking import region_flow_pb2 as mediapipe_dot_util_dot_tracking_dot_region__flow__pb2
|
||||
|
||||
|
||||
DESCRIPTOR = _descriptor.FileDescriptor(
|
||||
name='mediapipe/util/tracking/frame_selection.proto',
|
||||
package='mediapipe',
|
||||
syntax='proto2',
|
||||
serialized_options=None,
|
||||
serialized_pb=_b('\n-mediapipe/util/tracking/frame_selection.proto\x12\tmediapipe\x1a+mediapipe/util/tracking/camera_motion.proto\x1a@mediapipe/util/tracking/frame_selection_solution_evaluator.proto\x1a)mediapipe/util/tracking/region_flow.proto\"e\n\x17\x46rameSelectionTimestamp\x12\x11\n\ttimestamp\x18\x01 \x01(\x03\x12\x11\n\tframe_idx\x18\x02 \x01(\x05\x12$\n\x18processed_from_timestamp\x18\x03 \x01(\x03:\x02-1\"\xc6\x01\n\x14\x46rameSelectionResult\x12\x11\n\ttimestamp\x18\x01 \x01(\x03\x12\x11\n\tframe_idx\x18\x02 \x01(\x05\x12.\n\rcamera_motion\x18\x03 \x01(\x0b\x32\x17.mediapipe.CameraMotion\x12\x32\n\x08\x66\x65\x61tures\x18\x04 \x01(\x0b\x32 .mediapipe.RegionFlowFeatureList\x12$\n\x18processed_from_timestamp\x18\x05 \x01(\x03:\x02-1\"\xdc\x01\n\x17\x46rameSelectionCriterion\x12\x18\n\rsampling_rate\x18\x01 \x01(\x05:\x01\x30\x12\x1c\n\x10\x62\x61ndwidth_frames\x18\x02 \x01(\x02:\x02\x35\x30\x12\x1f\n\x14search_radius_frames\x18\x03 \x01(\x05:\x01\x31\x12J\n\x12solution_evaluator\x18\x04 \x01(\x0b\x32..mediapipe.FrameSelectionSolutionEvaluatorType\x12\x1c\n\x11max_output_frames\x18\x05 \x01(\x05:\x01\x30\"g\n\x15\x46rameSelectionOptions\x12\x35\n\tcriterion\x18\x01 \x03(\x0b\x32\".mediapipe.FrameSelectionCriterion\x12\x17\n\nchunk_size\x18\x02 \x01(\x05:\x03\x31\x30\x30')
|
||||
,
|
||||
dependencies=[mediapipe_dot_util_dot_tracking_dot_camera__motion__pb2.DESCRIPTOR,mediapipe_dot_util_dot_tracking_dot_frame__selection__solution__evaluator__pb2.DESCRIPTOR,mediapipe_dot_util_dot_tracking_dot_region__flow__pb2.DESCRIPTOR,])
|
||||
|
||||
|
||||
|
||||
|
||||
_FRAMESELECTIONTIMESTAMP = _descriptor.Descriptor(
|
||||
name='FrameSelectionTimestamp',
|
||||
full_name='mediapipe.FrameSelectionTimestamp',
|
||||
filename=None,
|
||||
file=DESCRIPTOR,
|
||||
containing_type=None,
|
||||
fields=[
|
||||
_descriptor.FieldDescriptor(
|
||||
name='timestamp', full_name='mediapipe.FrameSelectionTimestamp.timestamp', index=0,
|
||||
number=1, type=3, cpp_type=2, 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='frame_idx', full_name='mediapipe.FrameSelectionTimestamp.frame_idx', 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='processed_from_timestamp', full_name='mediapipe.FrameSelectionTimestamp.processed_from_timestamp', index=2,
|
||||
number=3, type=3, cpp_type=2, 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=214,
|
||||
serialized_end=315,
|
||||
)
|
||||
|
||||
|
||||
_FRAMESELECTIONRESULT = _descriptor.Descriptor(
|
||||
name='FrameSelectionResult',
|
||||
full_name='mediapipe.FrameSelectionResult',
|
||||
filename=None,
|
||||
file=DESCRIPTOR,
|
||||
containing_type=None,
|
||||
fields=[
|
||||
_descriptor.FieldDescriptor(
|
||||
name='timestamp', full_name='mediapipe.FrameSelectionResult.timestamp', index=0,
|
||||
number=1, type=3, cpp_type=2, 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='frame_idx', full_name='mediapipe.FrameSelectionResult.frame_idx', 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='camera_motion', full_name='mediapipe.FrameSelectionResult.camera_motion', 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='features', full_name='mediapipe.FrameSelectionResult.features', 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),
|
||||
_descriptor.FieldDescriptor(
|
||||
name='processed_from_timestamp', full_name='mediapipe.FrameSelectionResult.processed_from_timestamp', index=4,
|
||||
number=5, type=3, cpp_type=2, 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=318,
|
||||
serialized_end=516,
|
||||
)
|
||||
|
||||
|
||||
_FRAMESELECTIONCRITERION = _descriptor.Descriptor(
|
||||
name='FrameSelectionCriterion',
|
||||
full_name='mediapipe.FrameSelectionCriterion',
|
||||
filename=None,
|
||||
file=DESCRIPTOR,
|
||||
containing_type=None,
|
||||
fields=[
|
||||
_descriptor.FieldDescriptor(
|
||||
name='sampling_rate', full_name='mediapipe.FrameSelectionCriterion.sampling_rate', index=0,
|
||||
number=1, 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='bandwidth_frames', full_name='mediapipe.FrameSelectionCriterion.bandwidth_frames', index=1,
|
||||
number=2, type=2, cpp_type=6, label=1,
|
||||
has_default_value=True, default_value=float(50),
|
||||
message_type=None, enum_type=None, containing_type=None,
|
||||
is_extension=False, extension_scope=None,
|
||||
serialized_options=None, file=DESCRIPTOR),
|
||||
_descriptor.FieldDescriptor(
|
||||
name='search_radius_frames', full_name='mediapipe.FrameSelectionCriterion.search_radius_frames', index=2,
|
||||
number=3, 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='solution_evaluator', full_name='mediapipe.FrameSelectionCriterion.solution_evaluator', 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),
|
||||
_descriptor.FieldDescriptor(
|
||||
name='max_output_frames', full_name='mediapipe.FrameSelectionCriterion.max_output_frames', index=4,
|
||||
number=5, 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),
|
||||
],
|
||||
extensions=[
|
||||
],
|
||||
nested_types=[],
|
||||
enum_types=[
|
||||
],
|
||||
serialized_options=None,
|
||||
is_extendable=False,
|
||||
syntax='proto2',
|
||||
extension_ranges=[],
|
||||
oneofs=[
|
||||
],
|
||||
serialized_start=519,
|
||||
serialized_end=739,
|
||||
)
|
||||
|
||||
|
||||
_FRAMESELECTIONOPTIONS = _descriptor.Descriptor(
|
||||
name='FrameSelectionOptions',
|
||||
full_name='mediapipe.FrameSelectionOptions',
|
||||
filename=None,
|
||||
file=DESCRIPTOR,
|
||||
containing_type=None,
|
||||
fields=[
|
||||
_descriptor.FieldDescriptor(
|
||||
name='criterion', full_name='mediapipe.FrameSelectionOptions.criterion', index=0,
|
||||
number=1, type=11, cpp_type=10, 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='chunk_size', full_name='mediapipe.FrameSelectionOptions.chunk_size', index=1,
|
||||
number=2, type=5, cpp_type=1, label=1,
|
||||
has_default_value=True, default_value=100,
|
||||
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=741,
|
||||
serialized_end=844,
|
||||
)
|
||||
|
||||
_FRAMESELECTIONRESULT.fields_by_name['camera_motion'].message_type = mediapipe_dot_util_dot_tracking_dot_camera__motion__pb2._CAMERAMOTION
|
||||
_FRAMESELECTIONRESULT.fields_by_name['features'].message_type = mediapipe_dot_util_dot_tracking_dot_region__flow__pb2._REGIONFLOWFEATURELIST
|
||||
_FRAMESELECTIONCRITERION.fields_by_name['solution_evaluator'].message_type = mediapipe_dot_util_dot_tracking_dot_frame__selection__solution__evaluator__pb2._FRAMESELECTIONSOLUTIONEVALUATORTYPE
|
||||
_FRAMESELECTIONOPTIONS.fields_by_name['criterion'].message_type = _FRAMESELECTIONCRITERION
|
||||
DESCRIPTOR.message_types_by_name['FrameSelectionTimestamp'] = _FRAMESELECTIONTIMESTAMP
|
||||
DESCRIPTOR.message_types_by_name['FrameSelectionResult'] = _FRAMESELECTIONRESULT
|
||||
DESCRIPTOR.message_types_by_name['FrameSelectionCriterion'] = _FRAMESELECTIONCRITERION
|
||||
DESCRIPTOR.message_types_by_name['FrameSelectionOptions'] = _FRAMESELECTIONOPTIONS
|
||||
_sym_db.RegisterFileDescriptor(DESCRIPTOR)
|
||||
|
||||
FrameSelectionTimestamp = _reflection.GeneratedProtocolMessageType('FrameSelectionTimestamp', (_message.Message,), dict(
|
||||
DESCRIPTOR = _FRAMESELECTIONTIMESTAMP,
|
||||
__module__ = 'mediapipe.util.tracking.frame_selection_pb2'
|
||||
# @@protoc_insertion_point(class_scope:mediapipe.FrameSelectionTimestamp)
|
||||
))
|
||||
_sym_db.RegisterMessage(FrameSelectionTimestamp)
|
||||
|
||||
FrameSelectionResult = _reflection.GeneratedProtocolMessageType('FrameSelectionResult', (_message.Message,), dict(
|
||||
DESCRIPTOR = _FRAMESELECTIONRESULT,
|
||||
__module__ = 'mediapipe.util.tracking.frame_selection_pb2'
|
||||
# @@protoc_insertion_point(class_scope:mediapipe.FrameSelectionResult)
|
||||
))
|
||||
_sym_db.RegisterMessage(FrameSelectionResult)
|
||||
|
||||
FrameSelectionCriterion = _reflection.GeneratedProtocolMessageType('FrameSelectionCriterion', (_message.Message,), dict(
|
||||
DESCRIPTOR = _FRAMESELECTIONCRITERION,
|
||||
__module__ = 'mediapipe.util.tracking.frame_selection_pb2'
|
||||
# @@protoc_insertion_point(class_scope:mediapipe.FrameSelectionCriterion)
|
||||
))
|
||||
_sym_db.RegisterMessage(FrameSelectionCriterion)
|
||||
|
||||
FrameSelectionOptions = _reflection.GeneratedProtocolMessageType('FrameSelectionOptions', (_message.Message,), dict(
|
||||
DESCRIPTOR = _FRAMESELECTIONOPTIONS,
|
||||
__module__ = 'mediapipe.util.tracking.frame_selection_pb2'
|
||||
# @@protoc_insertion_point(class_scope:mediapipe.FrameSelectionOptions)
|
||||
))
|
||||
_sym_db.RegisterMessage(FrameSelectionOptions)
|
||||
|
||||
|
||||
# @@protoc_insertion_point(module_scope)
|
||||
@@ -0,0 +1,110 @@
|
||||
# -*- coding: utf-8 -*-
|
||||
# Generated by the protocol buffer compiler. DO NOT EDIT!
|
||||
# source: mediapipe/util/tracking/frame_selection_solution_evaluator.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()
|
||||
|
||||
|
||||
|
||||
|
||||
DESCRIPTOR = _descriptor.FileDescriptor(
|
||||
name='mediapipe/util/tracking/frame_selection_solution_evaluator.proto',
|
||||
package='mediapipe',
|
||||
syntax='proto2',
|
||||
serialized_options=None,
|
||||
serialized_pb=_b('\n@mediapipe/util/tracking/frame_selection_solution_evaluator.proto\x12\tmediapipe\"4\n&FrameSelectionSolutionEvaluatorOptions*\n\x08\xa0\x9c\x01\x10\x80\x80\x80\x80\x02\"\x9e\x01\n#FrameSelectionSolutionEvaluatorType\x12\x33\n\nclass_name\x18\x01 \x01(\t:\x1f\x46rameSelectionSolutionEvaluator\x12\x42\n\x07options\x18\x02 \x01(\x0b\x32\x31.mediapipe.FrameSelectionSolutionEvaluatorOptions')
|
||||
)
|
||||
|
||||
|
||||
|
||||
|
||||
_FRAMESELECTIONSOLUTIONEVALUATOROPTIONS = _descriptor.Descriptor(
|
||||
name='FrameSelectionSolutionEvaluatorOptions',
|
||||
full_name='mediapipe.FrameSelectionSolutionEvaluatorOptions',
|
||||
filename=None,
|
||||
file=DESCRIPTOR,
|
||||
containing_type=None,
|
||||
fields=[
|
||||
],
|
||||
extensions=[
|
||||
],
|
||||
nested_types=[],
|
||||
enum_types=[
|
||||
],
|
||||
serialized_options=None,
|
||||
is_extendable=True,
|
||||
syntax='proto2',
|
||||
extension_ranges=[(20000, 536870912), ],
|
||||
oneofs=[
|
||||
],
|
||||
serialized_start=79,
|
||||
serialized_end=131,
|
||||
)
|
||||
|
||||
|
||||
_FRAMESELECTIONSOLUTIONEVALUATORTYPE = _descriptor.Descriptor(
|
||||
name='FrameSelectionSolutionEvaluatorType',
|
||||
full_name='mediapipe.FrameSelectionSolutionEvaluatorType',
|
||||
filename=None,
|
||||
file=DESCRIPTOR,
|
||||
containing_type=None,
|
||||
fields=[
|
||||
_descriptor.FieldDescriptor(
|
||||
name='class_name', full_name='mediapipe.FrameSelectionSolutionEvaluatorType.class_name', index=0,
|
||||
number=1, type=9, cpp_type=9, label=1,
|
||||
has_default_value=True, default_value=_b("FrameSelectionSolutionEvaluator").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='options', full_name='mediapipe.FrameSelectionSolutionEvaluatorType.options', 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),
|
||||
],
|
||||
extensions=[
|
||||
],
|
||||
nested_types=[],
|
||||
enum_types=[
|
||||
],
|
||||
serialized_options=None,
|
||||
is_extendable=False,
|
||||
syntax='proto2',
|
||||
extension_ranges=[],
|
||||
oneofs=[
|
||||
],
|
||||
serialized_start=134,
|
||||
serialized_end=292,
|
||||
)
|
||||
|
||||
_FRAMESELECTIONSOLUTIONEVALUATORTYPE.fields_by_name['options'].message_type = _FRAMESELECTIONSOLUTIONEVALUATOROPTIONS
|
||||
DESCRIPTOR.message_types_by_name['FrameSelectionSolutionEvaluatorOptions'] = _FRAMESELECTIONSOLUTIONEVALUATOROPTIONS
|
||||
DESCRIPTOR.message_types_by_name['FrameSelectionSolutionEvaluatorType'] = _FRAMESELECTIONSOLUTIONEVALUATORTYPE
|
||||
_sym_db.RegisterFileDescriptor(DESCRIPTOR)
|
||||
|
||||
FrameSelectionSolutionEvaluatorOptions = _reflection.GeneratedProtocolMessageType('FrameSelectionSolutionEvaluatorOptions', (_message.Message,), dict(
|
||||
DESCRIPTOR = _FRAMESELECTIONSOLUTIONEVALUATOROPTIONS,
|
||||
__module__ = 'mediapipe.util.tracking.frame_selection_solution_evaluator_pb2'
|
||||
# @@protoc_insertion_point(class_scope:mediapipe.FrameSelectionSolutionEvaluatorOptions)
|
||||
))
|
||||
_sym_db.RegisterMessage(FrameSelectionSolutionEvaluatorOptions)
|
||||
|
||||
FrameSelectionSolutionEvaluatorType = _reflection.GeneratedProtocolMessageType('FrameSelectionSolutionEvaluatorType', (_message.Message,), dict(
|
||||
DESCRIPTOR = _FRAMESELECTIONSOLUTIONEVALUATORTYPE,
|
||||
__module__ = 'mediapipe.util.tracking.frame_selection_solution_evaluator_pb2'
|
||||
# @@protoc_insertion_point(class_scope:mediapipe.FrameSelectionSolutionEvaluatorType)
|
||||
))
|
||||
_sym_db.RegisterMessage(FrameSelectionSolutionEvaluatorType)
|
||||
|
||||
|
||||
# @@protoc_insertion_point(module_scope)
|
||||
@@ -0,0 +1,348 @@
|
||||
# -*- coding: utf-8 -*-
|
||||
# Generated by the protocol buffer compiler. DO NOT EDIT!
|
||||
# source: mediapipe/util/tracking/motion_analysis.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.util.tracking import motion_estimation_pb2 as mediapipe_dot_util_dot_tracking_dot_motion__estimation__pb2
|
||||
from mediapipe.util.tracking import motion_saliency_pb2 as mediapipe_dot_util_dot_tracking_dot_motion__saliency__pb2
|
||||
from mediapipe.util.tracking import region_flow_computation_pb2 as mediapipe_dot_util_dot_tracking_dot_region__flow__computation__pb2
|
||||
|
||||
|
||||
DESCRIPTOR = _descriptor.FileDescriptor(
|
||||
name='mediapipe/util/tracking/motion_analysis.proto',
|
||||
package='mediapipe',
|
||||
syntax='proto2',
|
||||
serialized_options=None,
|
||||
serialized_pb=_b('\n-mediapipe/util/tracking/motion_analysis.proto\x12\tmediapipe\x1a/mediapipe/util/tracking/motion_estimation.proto\x1a-mediapipe/util/tracking/motion_saliency.proto\x1a\x35mediapipe/util/tracking/region_flow_computation.proto\"\xf8\n\n\x15MotionAnalysisOptions\x12`\n\x0f\x61nalysis_policy\x18\x0e \x01(\x0e\x32/.mediapipe.MotionAnalysisOptions.AnalysisPolicy:\x16\x41NALYSIS_POLICY_LEGACY\x12=\n\x0c\x66low_options\x18\x01 \x01(\x0b\x32\'.mediapipe.RegionFlowComputationOptions\x12:\n\x0emotion_options\x18\x02 \x01(\x0b\x32\".mediapipe.MotionEstimationOptions\x12:\n\x10saliency_options\x18\x03 \x01(\x0b\x32 .mediapipe.MotionSaliencyOptions\x12 \n\x14\x65stimation_clip_size\x18\x04 \x01(\x05:\x02\x31\x36\x12\x33\n$subtract_camera_motion_from_features\x18\x05 \x01(\x08:\x05\x66\x61lse\x12\x16\n\x0btrack_index\x18\x06 \x01(\x05:\x01\x30\x12&\n\x17\x63ompute_motion_saliency\x18\x07 \x01(\x08:\x05\x66\x61lse\x12%\n\x17select_saliency_inliers\x18\x08 \x01(\x08:\x04true\x12\x1d\n\x0f\x66ilter_saliency\x18\t \x01(\x08:\x04true\x12\"\n\x13post_irls_smoothing\x18\n \x01(\x08:\x05\x66\x61lse\x12)\n\x1drejection_transform_threshold\x18\r \x01(\x02:\x02\x32\x30\x12T\n\x15visualization_options\x18\x0b \x01(\x0b\x32\x35.mediapipe.MotionAnalysisOptions.VisualizationOptions\x12N\n\x12\x66oreground_options\x18\x0c \x01(\x0b\x32\x32.mediapipe.MotionAnalysisOptions.ForegroundOptions\x1a\xc5\x02\n\x14VisualizationOptions\x12,\n\x1evisualize_region_flow_features\x18\x01 \x01(\x08:\x04true\x12\'\n\x18visualize_salient_points\x18\x02 \x01(\x08:\x05\x66\x61lse\x12\x19\n\x0eline_thickness\x18\x05 \x01(\x05:\x01\x34\x12&\n\x17\x66oreground_jet_coloring\x18\x03 \x01(\x08:\x05\x66\x61lse\x12-\n\x1evisualize_blur_analysis_region\x18\x04 \x01(\x08:\x05\x66\x61lse\x12\x1d\n\x0fvisualize_stats\x18\x06 \x01(\x08:\x04true\x12!\n\x16min_long_feature_track\x18\x07 \x01(\x05:\x01\x30\x12\"\n\x17max_long_feature_points\x18\x08 \x01(\x05:\x01\x30\x1a}\n\x11\x46oregroundOptions\x12!\n\x14\x66oreground_threshold\x18\x01 \x01(\x02:\x03\x30.5\x12\x1b\n\x10\x66oreground_gamma\x18\x02 \x01(\x02:\x01\x31\x12(\n\x1athreshold_coverage_scaling\x18\x03 \x01(\x08:\x04true\"\xac\x01\n\x0e\x41nalysisPolicy\x12\x1a\n\x16\x41NALYSIS_POLICY_LEGACY\x10\x00\x12\x19\n\x15\x41NALYSIS_POLICY_VIDEO\x10\x01\x12 \n\x1c\x41NALYSIS_POLICY_VIDEO_MOBILE\x10\x02\x12!\n\x1d\x41NALYSIS_POLICY_CAMERA_MOBILE\x10\x03\x12\x1e\n\x1a\x41NALYSIS_POLICY_HYPERLAPSE\x10\x04')
|
||||
,
|
||||
dependencies=[mediapipe_dot_util_dot_tracking_dot_motion__estimation__pb2.DESCRIPTOR,mediapipe_dot_util_dot_tracking_dot_motion__saliency__pb2.DESCRIPTOR,mediapipe_dot_util_dot_tracking_dot_region__flow__computation__pb2.DESCRIPTOR,])
|
||||
|
||||
|
||||
|
||||
_MOTIONANALYSISOPTIONS_ANALYSISPOLICY = _descriptor.EnumDescriptor(
|
||||
name='AnalysisPolicy',
|
||||
full_name='mediapipe.MotionAnalysisOptions.AnalysisPolicy',
|
||||
filename=None,
|
||||
file=DESCRIPTOR,
|
||||
values=[
|
||||
_descriptor.EnumValueDescriptor(
|
||||
name='ANALYSIS_POLICY_LEGACY', index=0, number=0,
|
||||
serialized_options=None,
|
||||
type=None),
|
||||
_descriptor.EnumValueDescriptor(
|
||||
name='ANALYSIS_POLICY_VIDEO', index=1, number=1,
|
||||
serialized_options=None,
|
||||
type=None),
|
||||
_descriptor.EnumValueDescriptor(
|
||||
name='ANALYSIS_POLICY_VIDEO_MOBILE', index=2, number=2,
|
||||
serialized_options=None,
|
||||
type=None),
|
||||
_descriptor.EnumValueDescriptor(
|
||||
name='ANALYSIS_POLICY_CAMERA_MOBILE', index=3, number=3,
|
||||
serialized_options=None,
|
||||
type=None),
|
||||
_descriptor.EnumValueDescriptor(
|
||||
name='ANALYSIS_POLICY_HYPERLAPSE', index=4, number=4,
|
||||
serialized_options=None,
|
||||
type=None),
|
||||
],
|
||||
containing_type=None,
|
||||
serialized_options=None,
|
||||
serialized_start=1440,
|
||||
serialized_end=1612,
|
||||
)
|
||||
_sym_db.RegisterEnumDescriptor(_MOTIONANALYSISOPTIONS_ANALYSISPOLICY)
|
||||
|
||||
|
||||
_MOTIONANALYSISOPTIONS_VISUALIZATIONOPTIONS = _descriptor.Descriptor(
|
||||
name='VisualizationOptions',
|
||||
full_name='mediapipe.MotionAnalysisOptions.VisualizationOptions',
|
||||
filename=None,
|
||||
file=DESCRIPTOR,
|
||||
containing_type=None,
|
||||
fields=[
|
||||
_descriptor.FieldDescriptor(
|
||||
name='visualize_region_flow_features', full_name='mediapipe.MotionAnalysisOptions.VisualizationOptions.visualize_region_flow_features', 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='visualize_salient_points', full_name='mediapipe.MotionAnalysisOptions.VisualizationOptions.visualize_salient_points', 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=None, file=DESCRIPTOR),
|
||||
_descriptor.FieldDescriptor(
|
||||
name='line_thickness', full_name='mediapipe.MotionAnalysisOptions.VisualizationOptions.line_thickness', index=2,
|
||||
number=5, type=5, cpp_type=1, label=1,
|
||||
has_default_value=True, default_value=4,
|
||||
message_type=None, enum_type=None, containing_type=None,
|
||||
is_extension=False, extension_scope=None,
|
||||
serialized_options=None, file=DESCRIPTOR),
|
||||
_descriptor.FieldDescriptor(
|
||||
name='foreground_jet_coloring', full_name='mediapipe.MotionAnalysisOptions.VisualizationOptions.foreground_jet_coloring', index=3,
|
||||
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=None, file=DESCRIPTOR),
|
||||
_descriptor.FieldDescriptor(
|
||||
name='visualize_blur_analysis_region', full_name='mediapipe.MotionAnalysisOptions.VisualizationOptions.visualize_blur_analysis_region', index=4,
|
||||
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='visualize_stats', full_name='mediapipe.MotionAnalysisOptions.VisualizationOptions.visualize_stats', index=5,
|
||||
number=6, 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='min_long_feature_track', full_name='mediapipe.MotionAnalysisOptions.VisualizationOptions.min_long_feature_track', index=6,
|
||||
number=7, 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='max_long_feature_points', full_name='mediapipe.MotionAnalysisOptions.VisualizationOptions.max_long_feature_points', index=7,
|
||||
number=8, 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),
|
||||
],
|
||||
extensions=[
|
||||
],
|
||||
nested_types=[],
|
||||
enum_types=[
|
||||
],
|
||||
serialized_options=None,
|
||||
is_extendable=False,
|
||||
syntax='proto2',
|
||||
extension_ranges=[],
|
||||
oneofs=[
|
||||
],
|
||||
serialized_start=985,
|
||||
serialized_end=1310,
|
||||
)
|
||||
|
||||
_MOTIONANALYSISOPTIONS_FOREGROUNDOPTIONS = _descriptor.Descriptor(
|
||||
name='ForegroundOptions',
|
||||
full_name='mediapipe.MotionAnalysisOptions.ForegroundOptions',
|
||||
filename=None,
|
||||
file=DESCRIPTOR,
|
||||
containing_type=None,
|
||||
fields=[
|
||||
_descriptor.FieldDescriptor(
|
||||
name='foreground_threshold', full_name='mediapipe.MotionAnalysisOptions.ForegroundOptions.foreground_threshold', index=0,
|
||||
number=1, type=2, cpp_type=6, label=1,
|
||||
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='foreground_gamma', full_name='mediapipe.MotionAnalysisOptions.ForegroundOptions.foreground_gamma', index=1,
|
||||
number=2, 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='threshold_coverage_scaling', full_name='mediapipe.MotionAnalysisOptions.ForegroundOptions.threshold_coverage_scaling', 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),
|
||||
],
|
||||
extensions=[
|
||||
],
|
||||
nested_types=[],
|
||||
enum_types=[
|
||||
],
|
||||
serialized_options=None,
|
||||
is_extendable=False,
|
||||
syntax='proto2',
|
||||
extension_ranges=[],
|
||||
oneofs=[
|
||||
],
|
||||
serialized_start=1312,
|
||||
serialized_end=1437,
|
||||
)
|
||||
|
||||
_MOTIONANALYSISOPTIONS = _descriptor.Descriptor(
|
||||
name='MotionAnalysisOptions',
|
||||
full_name='mediapipe.MotionAnalysisOptions',
|
||||
filename=None,
|
||||
file=DESCRIPTOR,
|
||||
containing_type=None,
|
||||
fields=[
|
||||
_descriptor.FieldDescriptor(
|
||||
name='analysis_policy', full_name='mediapipe.MotionAnalysisOptions.analysis_policy', index=0,
|
||||
number=14, 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='flow_options', full_name='mediapipe.MotionAnalysisOptions.flow_options', index=1,
|
||||
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='motion_options', full_name='mediapipe.MotionAnalysisOptions.motion_options', index=2,
|
||||
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='saliency_options', full_name='mediapipe.MotionAnalysisOptions.saliency_options', index=3,
|
||||
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='estimation_clip_size', full_name='mediapipe.MotionAnalysisOptions.estimation_clip_size', index=4,
|
||||
number=4, type=5, cpp_type=1, label=1,
|
||||
has_default_value=True, default_value=16,
|
||||
message_type=None, enum_type=None, containing_type=None,
|
||||
is_extension=False, extension_scope=None,
|
||||
serialized_options=None, file=DESCRIPTOR),
|
||||
_descriptor.FieldDescriptor(
|
||||
name='subtract_camera_motion_from_features', full_name='mediapipe.MotionAnalysisOptions.subtract_camera_motion_from_features', index=5,
|
||||
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='track_index', full_name='mediapipe.MotionAnalysisOptions.track_index', index=6,
|
||||
number=6, 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='compute_motion_saliency', full_name='mediapipe.MotionAnalysisOptions.compute_motion_saliency', index=7,
|
||||
number=7, 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='select_saliency_inliers', full_name='mediapipe.MotionAnalysisOptions.select_saliency_inliers', index=8,
|
||||
number=8, 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='filter_saliency', full_name='mediapipe.MotionAnalysisOptions.filter_saliency', index=9,
|
||||
number=9, 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='post_irls_smoothing', full_name='mediapipe.MotionAnalysisOptions.post_irls_smoothing', index=10,
|
||||
number=10, 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='rejection_transform_threshold', full_name='mediapipe.MotionAnalysisOptions.rejection_transform_threshold', index=11,
|
||||
number=13, type=2, cpp_type=6, label=1,
|
||||
has_default_value=True, default_value=float(20),
|
||||
message_type=None, enum_type=None, containing_type=None,
|
||||
is_extension=False, extension_scope=None,
|
||||
serialized_options=None, file=DESCRIPTOR),
|
||||
_descriptor.FieldDescriptor(
|
||||
name='visualization_options', full_name='mediapipe.MotionAnalysisOptions.visualization_options', index=12,
|
||||
number=11, 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='foreground_options', full_name='mediapipe.MotionAnalysisOptions.foreground_options', index=13,
|
||||
number=12, 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=[_MOTIONANALYSISOPTIONS_VISUALIZATIONOPTIONS, _MOTIONANALYSISOPTIONS_FOREGROUNDOPTIONS, ],
|
||||
enum_types=[
|
||||
_MOTIONANALYSISOPTIONS_ANALYSISPOLICY,
|
||||
],
|
||||
serialized_options=None,
|
||||
is_extendable=False,
|
||||
syntax='proto2',
|
||||
extension_ranges=[],
|
||||
oneofs=[
|
||||
],
|
||||
serialized_start=212,
|
||||
serialized_end=1612,
|
||||
)
|
||||
|
||||
_MOTIONANALYSISOPTIONS_VISUALIZATIONOPTIONS.containing_type = _MOTIONANALYSISOPTIONS
|
||||
_MOTIONANALYSISOPTIONS_FOREGROUNDOPTIONS.containing_type = _MOTIONANALYSISOPTIONS
|
||||
_MOTIONANALYSISOPTIONS.fields_by_name['analysis_policy'].enum_type = _MOTIONANALYSISOPTIONS_ANALYSISPOLICY
|
||||
_MOTIONANALYSISOPTIONS.fields_by_name['flow_options'].message_type = mediapipe_dot_util_dot_tracking_dot_region__flow__computation__pb2._REGIONFLOWCOMPUTATIONOPTIONS
|
||||
_MOTIONANALYSISOPTIONS.fields_by_name['motion_options'].message_type = mediapipe_dot_util_dot_tracking_dot_motion__estimation__pb2._MOTIONESTIMATIONOPTIONS
|
||||
_MOTIONANALYSISOPTIONS.fields_by_name['saliency_options'].message_type = mediapipe_dot_util_dot_tracking_dot_motion__saliency__pb2._MOTIONSALIENCYOPTIONS
|
||||
_MOTIONANALYSISOPTIONS.fields_by_name['visualization_options'].message_type = _MOTIONANALYSISOPTIONS_VISUALIZATIONOPTIONS
|
||||
_MOTIONANALYSISOPTIONS.fields_by_name['foreground_options'].message_type = _MOTIONANALYSISOPTIONS_FOREGROUNDOPTIONS
|
||||
_MOTIONANALYSISOPTIONS_ANALYSISPOLICY.containing_type = _MOTIONANALYSISOPTIONS
|
||||
DESCRIPTOR.message_types_by_name['MotionAnalysisOptions'] = _MOTIONANALYSISOPTIONS
|
||||
_sym_db.RegisterFileDescriptor(DESCRIPTOR)
|
||||
|
||||
MotionAnalysisOptions = _reflection.GeneratedProtocolMessageType('MotionAnalysisOptions', (_message.Message,), dict(
|
||||
|
||||
VisualizationOptions = _reflection.GeneratedProtocolMessageType('VisualizationOptions', (_message.Message,), dict(
|
||||
DESCRIPTOR = _MOTIONANALYSISOPTIONS_VISUALIZATIONOPTIONS,
|
||||
__module__ = 'mediapipe.util.tracking.motion_analysis_pb2'
|
||||
# @@protoc_insertion_point(class_scope:mediapipe.MotionAnalysisOptions.VisualizationOptions)
|
||||
))
|
||||
,
|
||||
|
||||
ForegroundOptions = _reflection.GeneratedProtocolMessageType('ForegroundOptions', (_message.Message,), dict(
|
||||
DESCRIPTOR = _MOTIONANALYSISOPTIONS_FOREGROUNDOPTIONS,
|
||||
__module__ = 'mediapipe.util.tracking.motion_analysis_pb2'
|
||||
# @@protoc_insertion_point(class_scope:mediapipe.MotionAnalysisOptions.ForegroundOptions)
|
||||
))
|
||||
,
|
||||
DESCRIPTOR = _MOTIONANALYSISOPTIONS,
|
||||
__module__ = 'mediapipe.util.tracking.motion_analysis_pb2'
|
||||
# @@protoc_insertion_point(class_scope:mediapipe.MotionAnalysisOptions)
|
||||
))
|
||||
_sym_db.RegisterMessage(MotionAnalysisOptions)
|
||||
_sym_db.RegisterMessage(MotionAnalysisOptions.VisualizationOptions)
|
||||
_sym_db.RegisterMessage(MotionAnalysisOptions.ForegroundOptions)
|
||||
|
||||
|
||||
# @@protoc_insertion_point(module_scope)
|
||||
File diff suppressed because one or more lines are too long
@@ -0,0 +1,519 @@
|
||||
# -*- coding: utf-8 -*-
|
||||
# Generated by the protocol buffer compiler. DO NOT EDIT!
|
||||
# source: mediapipe/util/tracking/motion_models.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()
|
||||
|
||||
|
||||
|
||||
|
||||
DESCRIPTOR = _descriptor.FileDescriptor(
|
||||
name='mediapipe/util/tracking/motion_models.proto',
|
||||
package='mediapipe',
|
||||
syntax='proto2',
|
||||
serialized_options=None,
|
||||
serialized_pb=_b('\n+mediapipe/util/tracking/motion_models.proto\x12\tmediapipe\"0\n\x10TranslationModel\x12\r\n\x02\x64x\x18\x01 \x01(\x02:\x01\x30\x12\r\n\x02\x64y\x18\x02 \x01(\x02:\x01\x30\"V\n\x0fSimilarityModel\x12\r\n\x02\x64x\x18\x01 \x01(\x02:\x01\x30\x12\r\n\x02\x64y\x18\x02 \x01(\x02:\x01\x30\x12\x10\n\x05scale\x18\x03 \x01(\x02:\x01\x31\x12\x13\n\x08rotation\x18\x04 \x01(\x02:\x01\x30\"Q\n\x15LinearSimilarityModel\x12\r\n\x02\x64x\x18\x01 \x01(\x02:\x01\x30\x12\r\n\x02\x64y\x18\x02 \x01(\x02:\x01\x30\x12\x0c\n\x01\x61\x18\x03 \x01(\x02:\x01\x31\x12\x0c\n\x01\x62\x18\x04 \x01(\x02:\x01\x30\"c\n\x0b\x41\x66\x66ineModel\x12\r\n\x02\x64x\x18\x01 \x01(\x02:\x01\x30\x12\r\n\x02\x64y\x18\x02 \x01(\x02:\x01\x30\x12\x0c\n\x01\x61\x18\x03 \x01(\x02:\x01\x31\x12\x0c\n\x01\x62\x18\x04 \x01(\x02:\x01\x30\x12\x0c\n\x01\x63\x18\x05 \x01(\x02:\x01\x30\x12\x0c\n\x01\x64\x18\x06 \x01(\x02:\x01\x31\"\x94\x01\n\nHomography\x12\x0f\n\x04h_00\x18\x01 \x01(\x02:\x01\x31\x12\x0f\n\x04h_01\x18\x02 \x01(\x02:\x01\x30\x12\x0f\n\x04h_02\x18\x03 \x01(\x02:\x01\x30\x12\x0f\n\x04h_10\x18\x04 \x01(\x02:\x01\x30\x12\x0f\n\x04h_11\x18\x05 \x01(\x02:\x01\x31\x12\x0f\n\x04h_12\x18\x06 \x01(\x02:\x01\x30\x12\x0f\n\x04h_20\x18\x07 \x01(\x02:\x01\x30\x12\x0f\n\x04h_21\x18\x08 \x01(\x02:\x01\x30\"J\n\x17MixtureLinearSimilarity\x12/\n\x05model\x18\x01 \x03(\x0b\x32 .mediapipe.LinearSimilarityModel\"6\n\rMixtureAffine\x12%\n\x05model\x18\x01 \x03(\x0b\x32\x16.mediapipe.AffineModel\"\xd0\x01\n\x11MixtureHomography\x12$\n\x05model\x18\x01 \x03(\x0b\x32\x15.mediapipe.Homography\x12>\n\x03\x64of\x18\x02 \x01(\x0e\x32(.mediapipe.MixtureHomography.VariableDOF:\x07\x41LL_DOF\"U\n\x0bVariableDOF\x12\x0b\n\x07\x41LL_DOF\x10\x00\x12\x13\n\x0fTRANSLATION_DOF\x10\x01\x12\x15\n\x11SKEW_ROTATION_DOF\x10\x02\x12\r\n\tCONST_DOF\x10\x03')
|
||||
)
|
||||
|
||||
|
||||
|
||||
_MIXTUREHOMOGRAPHY_VARIABLEDOF = _descriptor.EnumDescriptor(
|
||||
name='VariableDOF',
|
||||
full_name='mediapipe.MixtureHomography.VariableDOF',
|
||||
filename=None,
|
||||
file=DESCRIPTOR,
|
||||
values=[
|
||||
_descriptor.EnumValueDescriptor(
|
||||
name='ALL_DOF', index=0, number=0,
|
||||
serialized_options=None,
|
||||
type=None),
|
||||
_descriptor.EnumValueDescriptor(
|
||||
name='TRANSLATION_DOF', index=1, number=1,
|
||||
serialized_options=None,
|
||||
type=None),
|
||||
_descriptor.EnumValueDescriptor(
|
||||
name='SKEW_ROTATION_DOF', index=2, number=2,
|
||||
serialized_options=None,
|
||||
type=None),
|
||||
_descriptor.EnumValueDescriptor(
|
||||
name='CONST_DOF', index=3, number=3,
|
||||
serialized_options=None,
|
||||
type=None),
|
||||
],
|
||||
containing_type=None,
|
||||
serialized_options=None,
|
||||
serialized_start=787,
|
||||
serialized_end=872,
|
||||
)
|
||||
_sym_db.RegisterEnumDescriptor(_MIXTUREHOMOGRAPHY_VARIABLEDOF)
|
||||
|
||||
|
||||
_TRANSLATIONMODEL = _descriptor.Descriptor(
|
||||
name='TranslationModel',
|
||||
full_name='mediapipe.TranslationModel',
|
||||
filename=None,
|
||||
file=DESCRIPTOR,
|
||||
containing_type=None,
|
||||
fields=[
|
||||
_descriptor.FieldDescriptor(
|
||||
name='dx', full_name='mediapipe.TranslationModel.dx', index=0,
|
||||
number=1, 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='dy', full_name='mediapipe.TranslationModel.dy', index=1,
|
||||
number=2, 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),
|
||||
],
|
||||
extensions=[
|
||||
],
|
||||
nested_types=[],
|
||||
enum_types=[
|
||||
],
|
||||
serialized_options=None,
|
||||
is_extendable=False,
|
||||
syntax='proto2',
|
||||
extension_ranges=[],
|
||||
oneofs=[
|
||||
],
|
||||
serialized_start=58,
|
||||
serialized_end=106,
|
||||
)
|
||||
|
||||
|
||||
_SIMILARITYMODEL = _descriptor.Descriptor(
|
||||
name='SimilarityModel',
|
||||
full_name='mediapipe.SimilarityModel',
|
||||
filename=None,
|
||||
file=DESCRIPTOR,
|
||||
containing_type=None,
|
||||
fields=[
|
||||
_descriptor.FieldDescriptor(
|
||||
name='dx', full_name='mediapipe.SimilarityModel.dx', index=0,
|
||||
number=1, 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='dy', full_name='mediapipe.SimilarityModel.dy', index=1,
|
||||
number=2, 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='scale', full_name='mediapipe.SimilarityModel.scale', index=2,
|
||||
number=3, 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='rotation', full_name='mediapipe.SimilarityModel.rotation', index=3,
|
||||
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),
|
||||
],
|
||||
extensions=[
|
||||
],
|
||||
nested_types=[],
|
||||
enum_types=[
|
||||
],
|
||||
serialized_options=None,
|
||||
is_extendable=False,
|
||||
syntax='proto2',
|
||||
extension_ranges=[],
|
||||
oneofs=[
|
||||
],
|
||||
serialized_start=108,
|
||||
serialized_end=194,
|
||||
)
|
||||
|
||||
|
||||
_LINEARSIMILARITYMODEL = _descriptor.Descriptor(
|
||||
name='LinearSimilarityModel',
|
||||
full_name='mediapipe.LinearSimilarityModel',
|
||||
filename=None,
|
||||
file=DESCRIPTOR,
|
||||
containing_type=None,
|
||||
fields=[
|
||||
_descriptor.FieldDescriptor(
|
||||
name='dx', full_name='mediapipe.LinearSimilarityModel.dx', index=0,
|
||||
number=1, 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='dy', full_name='mediapipe.LinearSimilarityModel.dy', index=1,
|
||||
number=2, 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='a', full_name='mediapipe.LinearSimilarityModel.a', index=2,
|
||||
number=3, 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='b', full_name='mediapipe.LinearSimilarityModel.b', index=3,
|
||||
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),
|
||||
],
|
||||
extensions=[
|
||||
],
|
||||
nested_types=[],
|
||||
enum_types=[
|
||||
],
|
||||
serialized_options=None,
|
||||
is_extendable=False,
|
||||
syntax='proto2',
|
||||
extension_ranges=[],
|
||||
oneofs=[
|
||||
],
|
||||
serialized_start=196,
|
||||
serialized_end=277,
|
||||
)
|
||||
|
||||
|
||||
_AFFINEMODEL = _descriptor.Descriptor(
|
||||
name='AffineModel',
|
||||
full_name='mediapipe.AffineModel',
|
||||
filename=None,
|
||||
file=DESCRIPTOR,
|
||||
containing_type=None,
|
||||
fields=[
|
||||
_descriptor.FieldDescriptor(
|
||||
name='dx', full_name='mediapipe.AffineModel.dx', index=0,
|
||||
number=1, 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='dy', full_name='mediapipe.AffineModel.dy', index=1,
|
||||
number=2, 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='a', full_name='mediapipe.AffineModel.a', index=2,
|
||||
number=3, 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='b', full_name='mediapipe.AffineModel.b', index=3,
|
||||
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='c', full_name='mediapipe.AffineModel.c', index=4,
|
||||
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='d', full_name='mediapipe.AffineModel.d', index=5,
|
||||
number=6, 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),
|
||||
],
|
||||
extensions=[
|
||||
],
|
||||
nested_types=[],
|
||||
enum_types=[
|
||||
],
|
||||
serialized_options=None,
|
||||
is_extendable=False,
|
||||
syntax='proto2',
|
||||
extension_ranges=[],
|
||||
oneofs=[
|
||||
],
|
||||
serialized_start=279,
|
||||
serialized_end=378,
|
||||
)
|
||||
|
||||
|
||||
_HOMOGRAPHY = _descriptor.Descriptor(
|
||||
name='Homography',
|
||||
full_name='mediapipe.Homography',
|
||||
filename=None,
|
||||
file=DESCRIPTOR,
|
||||
containing_type=None,
|
||||
fields=[
|
||||
_descriptor.FieldDescriptor(
|
||||
name='h_00', full_name='mediapipe.Homography.h_00', index=0,
|
||||
number=1, 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='h_01', full_name='mediapipe.Homography.h_01', index=1,
|
||||
number=2, 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_02', full_name='mediapipe.Homography.h_02', index=2,
|
||||
number=3, 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_10', full_name='mediapipe.Homography.h_10', index=3,
|
||||
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='h_11', full_name='mediapipe.Homography.h_11', index=4,
|
||||
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='h_12', full_name='mediapipe.Homography.h_12', index=5,
|
||||
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_20', full_name='mediapipe.Homography.h_20', index=6,
|
||||
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='h_21', full_name='mediapipe.Homography.h_21', index=7,
|
||||
number=8, 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),
|
||||
],
|
||||
extensions=[
|
||||
],
|
||||
nested_types=[],
|
||||
enum_types=[
|
||||
],
|
||||
serialized_options=None,
|
||||
is_extendable=False,
|
||||
syntax='proto2',
|
||||
extension_ranges=[],
|
||||
oneofs=[
|
||||
],
|
||||
serialized_start=381,
|
||||
serialized_end=529,
|
||||
)
|
||||
|
||||
|
||||
_MIXTURELINEARSIMILARITY = _descriptor.Descriptor(
|
||||
name='MixtureLinearSimilarity',
|
||||
full_name='mediapipe.MixtureLinearSimilarity',
|
||||
filename=None,
|
||||
file=DESCRIPTOR,
|
||||
containing_type=None,
|
||||
fields=[
|
||||
_descriptor.FieldDescriptor(
|
||||
name='model', full_name='mediapipe.MixtureLinearSimilarity.model', index=0,
|
||||
number=1, type=11, cpp_type=10, 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),
|
||||
],
|
||||
extensions=[
|
||||
],
|
||||
nested_types=[],
|
||||
enum_types=[
|
||||
],
|
||||
serialized_options=None,
|
||||
is_extendable=False,
|
||||
syntax='proto2',
|
||||
extension_ranges=[],
|
||||
oneofs=[
|
||||
],
|
||||
serialized_start=531,
|
||||
serialized_end=605,
|
||||
)
|
||||
|
||||
|
||||
_MIXTUREAFFINE = _descriptor.Descriptor(
|
||||
name='MixtureAffine',
|
||||
full_name='mediapipe.MixtureAffine',
|
||||
filename=None,
|
||||
file=DESCRIPTOR,
|
||||
containing_type=None,
|
||||
fields=[
|
||||
_descriptor.FieldDescriptor(
|
||||
name='model', full_name='mediapipe.MixtureAffine.model', index=0,
|
||||
number=1, type=11, cpp_type=10, 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),
|
||||
],
|
||||
extensions=[
|
||||
],
|
||||
nested_types=[],
|
||||
enum_types=[
|
||||
],
|
||||
serialized_options=None,
|
||||
is_extendable=False,
|
||||
syntax='proto2',
|
||||
extension_ranges=[],
|
||||
oneofs=[
|
||||
],
|
||||
serialized_start=607,
|
||||
serialized_end=661,
|
||||
)
|
||||
|
||||
|
||||
_MIXTUREHOMOGRAPHY = _descriptor.Descriptor(
|
||||
name='MixtureHomography',
|
||||
full_name='mediapipe.MixtureHomography',
|
||||
filename=None,
|
||||
file=DESCRIPTOR,
|
||||
containing_type=None,
|
||||
fields=[
|
||||
_descriptor.FieldDescriptor(
|
||||
name='model', full_name='mediapipe.MixtureHomography.model', index=0,
|
||||
number=1, type=11, cpp_type=10, 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='dof', full_name='mediapipe.MixtureHomography.dof', index=1,
|
||||
number=2, 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=[
|
||||
],
|
||||
nested_types=[],
|
||||
enum_types=[
|
||||
_MIXTUREHOMOGRAPHY_VARIABLEDOF,
|
||||
],
|
||||
serialized_options=None,
|
||||
is_extendable=False,
|
||||
syntax='proto2',
|
||||
extension_ranges=[],
|
||||
oneofs=[
|
||||
],
|
||||
serialized_start=664,
|
||||
serialized_end=872,
|
||||
)
|
||||
|
||||
_MIXTURELINEARSIMILARITY.fields_by_name['model'].message_type = _LINEARSIMILARITYMODEL
|
||||
_MIXTUREAFFINE.fields_by_name['model'].message_type = _AFFINEMODEL
|
||||
_MIXTUREHOMOGRAPHY.fields_by_name['model'].message_type = _HOMOGRAPHY
|
||||
_MIXTUREHOMOGRAPHY.fields_by_name['dof'].enum_type = _MIXTUREHOMOGRAPHY_VARIABLEDOF
|
||||
_MIXTUREHOMOGRAPHY_VARIABLEDOF.containing_type = _MIXTUREHOMOGRAPHY
|
||||
DESCRIPTOR.message_types_by_name['TranslationModel'] = _TRANSLATIONMODEL
|
||||
DESCRIPTOR.message_types_by_name['SimilarityModel'] = _SIMILARITYMODEL
|
||||
DESCRIPTOR.message_types_by_name['LinearSimilarityModel'] = _LINEARSIMILARITYMODEL
|
||||
DESCRIPTOR.message_types_by_name['AffineModel'] = _AFFINEMODEL
|
||||
DESCRIPTOR.message_types_by_name['Homography'] = _HOMOGRAPHY
|
||||
DESCRIPTOR.message_types_by_name['MixtureLinearSimilarity'] = _MIXTURELINEARSIMILARITY
|
||||
DESCRIPTOR.message_types_by_name['MixtureAffine'] = _MIXTUREAFFINE
|
||||
DESCRIPTOR.message_types_by_name['MixtureHomography'] = _MIXTUREHOMOGRAPHY
|
||||
_sym_db.RegisterFileDescriptor(DESCRIPTOR)
|
||||
|
||||
TranslationModel = _reflection.GeneratedProtocolMessageType('TranslationModel', (_message.Message,), dict(
|
||||
DESCRIPTOR = _TRANSLATIONMODEL,
|
||||
__module__ = 'mediapipe.util.tracking.motion_models_pb2'
|
||||
# @@protoc_insertion_point(class_scope:mediapipe.TranslationModel)
|
||||
))
|
||||
_sym_db.RegisterMessage(TranslationModel)
|
||||
|
||||
SimilarityModel = _reflection.GeneratedProtocolMessageType('SimilarityModel', (_message.Message,), dict(
|
||||
DESCRIPTOR = _SIMILARITYMODEL,
|
||||
__module__ = 'mediapipe.util.tracking.motion_models_pb2'
|
||||
# @@protoc_insertion_point(class_scope:mediapipe.SimilarityModel)
|
||||
))
|
||||
_sym_db.RegisterMessage(SimilarityModel)
|
||||
|
||||
LinearSimilarityModel = _reflection.GeneratedProtocolMessageType('LinearSimilarityModel', (_message.Message,), dict(
|
||||
DESCRIPTOR = _LINEARSIMILARITYMODEL,
|
||||
__module__ = 'mediapipe.util.tracking.motion_models_pb2'
|
||||
# @@protoc_insertion_point(class_scope:mediapipe.LinearSimilarityModel)
|
||||
))
|
||||
_sym_db.RegisterMessage(LinearSimilarityModel)
|
||||
|
||||
AffineModel = _reflection.GeneratedProtocolMessageType('AffineModel', (_message.Message,), dict(
|
||||
DESCRIPTOR = _AFFINEMODEL,
|
||||
__module__ = 'mediapipe.util.tracking.motion_models_pb2'
|
||||
# @@protoc_insertion_point(class_scope:mediapipe.AffineModel)
|
||||
))
|
||||
_sym_db.RegisterMessage(AffineModel)
|
||||
|
||||
Homography = _reflection.GeneratedProtocolMessageType('Homography', (_message.Message,), dict(
|
||||
DESCRIPTOR = _HOMOGRAPHY,
|
||||
__module__ = 'mediapipe.util.tracking.motion_models_pb2'
|
||||
# @@protoc_insertion_point(class_scope:mediapipe.Homography)
|
||||
))
|
||||
_sym_db.RegisterMessage(Homography)
|
||||
|
||||
MixtureLinearSimilarity = _reflection.GeneratedProtocolMessageType('MixtureLinearSimilarity', (_message.Message,), dict(
|
||||
DESCRIPTOR = _MIXTURELINEARSIMILARITY,
|
||||
__module__ = 'mediapipe.util.tracking.motion_models_pb2'
|
||||
# @@protoc_insertion_point(class_scope:mediapipe.MixtureLinearSimilarity)
|
||||
))
|
||||
_sym_db.RegisterMessage(MixtureLinearSimilarity)
|
||||
|
||||
MixtureAffine = _reflection.GeneratedProtocolMessageType('MixtureAffine', (_message.Message,), dict(
|
||||
DESCRIPTOR = _MIXTUREAFFINE,
|
||||
__module__ = 'mediapipe.util.tracking.motion_models_pb2'
|
||||
# @@protoc_insertion_point(class_scope:mediapipe.MixtureAffine)
|
||||
))
|
||||
_sym_db.RegisterMessage(MixtureAffine)
|
||||
|
||||
MixtureHomography = _reflection.GeneratedProtocolMessageType('MixtureHomography', (_message.Message,), dict(
|
||||
DESCRIPTOR = _MIXTUREHOMOGRAPHY,
|
||||
__module__ = 'mediapipe.util.tracking.motion_models_pb2'
|
||||
# @@protoc_insertion_point(class_scope:mediapipe.MixtureHomography)
|
||||
))
|
||||
_sym_db.RegisterMessage(MixtureHomography)
|
||||
|
||||
|
||||
# @@protoc_insertion_point(module_scope)
|
||||
@@ -0,0 +1,175 @@
|
||||
# -*- coding: utf-8 -*-
|
||||
# Generated by the protocol buffer compiler. DO NOT EDIT!
|
||||
# source: mediapipe/util/tracking/motion_saliency.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()
|
||||
|
||||
|
||||
|
||||
|
||||
DESCRIPTOR = _descriptor.FileDescriptor(
|
||||
name='mediapipe/util/tracking/motion_saliency.proto',
|
||||
package='mediapipe',
|
||||
syntax='proto2',
|
||||
serialized_options=None,
|
||||
serialized_pb=_b('\n-mediapipe/util/tracking/motion_saliency.proto\x12\tmediapipe\"\xa5\x04\n\x15MotionSaliencyOptions\x12\x17\n\nbound_left\x18\x01 \x01(\x02:\x03\x30.3\x12\x19\n\x0c\x62ound_bottom\x18\x02 \x01(\x02:\x03\x30.3\x12\x18\n\x0b\x62ound_right\x18\x0f \x01(\x02:\x03\x30.3\x12\x16\n\tbound_top\x18\x10 \x01(\x02:\x03\x30.3\x12\x1b\n\x0fsaliency_weight\x18\x03 \x01(\x02:\x02\x32\x30\x12-\n\x1escale_weight_by_flow_magnitude\x18\x08 \x01(\x08:\x05\x66\x61lse\x12\x17\n\x0cmin_features\x18\x04 \x01(\x05:\x01\x35\x12*\n\x1buse_only_foreground_regions\x18\t \x01(\x08:\x05\x66\x61lse\x12 \n\x14min_irls_mode_weight\x18\n \x01(\x02:\x02\x31\x30\x12\x1d\n\x12num_top_irls_modes\x18\x0b \x01(\x05:\x01\x33\x12\x1c\n\x0fmode_band_width\x18\x0c \x01(\x02:\x03\x30.1\x12!\n\x16selection_frame_radius\x18\x05 \x01(\x05:\x01\x35\x12\'\n\x1aselection_support_distance\x18\x06 \x01(\x02:\x03\x30.2\x12$\n\x19selection_minimum_support\x18\x07 \x01(\x05:\x01\x34\x12#\n\x15\x66iltering_sigma_space\x18\r \x01(\x02:\x04\x30.05\x12\x1f\n\x14\x66iltering_sigma_time\x18\x0e \x01(\x02:\x01\x35')
|
||||
)
|
||||
|
||||
|
||||
|
||||
|
||||
_MOTIONSALIENCYOPTIONS = _descriptor.Descriptor(
|
||||
name='MotionSaliencyOptions',
|
||||
full_name='mediapipe.MotionSaliencyOptions',
|
||||
filename=None,
|
||||
file=DESCRIPTOR,
|
||||
containing_type=None,
|
||||
fields=[
|
||||
_descriptor.FieldDescriptor(
|
||||
name='bound_left', full_name='mediapipe.MotionSaliencyOptions.bound_left', index=0,
|
||||
number=1, type=2, cpp_type=6, label=1,
|
||||
has_default_value=True, default_value=float(0.3),
|
||||
message_type=None, enum_type=None, containing_type=None,
|
||||
is_extension=False, extension_scope=None,
|
||||
serialized_options=None, file=DESCRIPTOR),
|
||||
_descriptor.FieldDescriptor(
|
||||
name='bound_bottom', full_name='mediapipe.MotionSaliencyOptions.bound_bottom', index=1,
|
||||
number=2, type=2, cpp_type=6, label=1,
|
||||
has_default_value=True, default_value=float(0.3),
|
||||
message_type=None, enum_type=None, containing_type=None,
|
||||
is_extension=False, extension_scope=None,
|
||||
serialized_options=None, file=DESCRIPTOR),
|
||||
_descriptor.FieldDescriptor(
|
||||
name='bound_right', full_name='mediapipe.MotionSaliencyOptions.bound_right', index=2,
|
||||
number=15, type=2, cpp_type=6, label=1,
|
||||
has_default_value=True, default_value=float(0.3),
|
||||
message_type=None, enum_type=None, containing_type=None,
|
||||
is_extension=False, extension_scope=None,
|
||||
serialized_options=None, file=DESCRIPTOR),
|
||||
_descriptor.FieldDescriptor(
|
||||
name='bound_top', full_name='mediapipe.MotionSaliencyOptions.bound_top', index=3,
|
||||
number=16, type=2, cpp_type=6, label=1,
|
||||
has_default_value=True, default_value=float(0.3),
|
||||
message_type=None, enum_type=None, containing_type=None,
|
||||
is_extension=False, extension_scope=None,
|
||||
serialized_options=None, file=DESCRIPTOR),
|
||||
_descriptor.FieldDescriptor(
|
||||
name='saliency_weight', full_name='mediapipe.MotionSaliencyOptions.saliency_weight', index=4,
|
||||
number=3, type=2, cpp_type=6, label=1,
|
||||
has_default_value=True, default_value=float(20),
|
||||
message_type=None, enum_type=None, containing_type=None,
|
||||
is_extension=False, extension_scope=None,
|
||||
serialized_options=None, file=DESCRIPTOR),
|
||||
_descriptor.FieldDescriptor(
|
||||
name='scale_weight_by_flow_magnitude', full_name='mediapipe.MotionSaliencyOptions.scale_weight_by_flow_magnitude', index=5,
|
||||
number=8, 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_features', full_name='mediapipe.MotionSaliencyOptions.min_features', index=6,
|
||||
number=4, type=5, cpp_type=1, label=1,
|
||||
has_default_value=True, default_value=5,
|
||||
message_type=None, enum_type=None, containing_type=None,
|
||||
is_extension=False, extension_scope=None,
|
||||
serialized_options=None, file=DESCRIPTOR),
|
||||
_descriptor.FieldDescriptor(
|
||||
name='use_only_foreground_regions', full_name='mediapipe.MotionSaliencyOptions.use_only_foreground_regions', index=7,
|
||||
number=9, 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_irls_mode_weight', full_name='mediapipe.MotionSaliencyOptions.min_irls_mode_weight', index=8,
|
||||
number=10, type=2, cpp_type=6, label=1,
|
||||
has_default_value=True, default_value=float(10),
|
||||
message_type=None, enum_type=None, containing_type=None,
|
||||
is_extension=False, extension_scope=None,
|
||||
serialized_options=None, file=DESCRIPTOR),
|
||||
_descriptor.FieldDescriptor(
|
||||
name='num_top_irls_modes', full_name='mediapipe.MotionSaliencyOptions.num_top_irls_modes', index=9,
|
||||
number=11, 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='mode_band_width', full_name='mediapipe.MotionSaliencyOptions.mode_band_width', index=10,
|
||||
number=12, type=2, cpp_type=6, label=1,
|
||||
has_default_value=True, default_value=float(0.1),
|
||||
message_type=None, enum_type=None, containing_type=None,
|
||||
is_extension=False, extension_scope=None,
|
||||
serialized_options=None, file=DESCRIPTOR),
|
||||
_descriptor.FieldDescriptor(
|
||||
name='selection_frame_radius', full_name='mediapipe.MotionSaliencyOptions.selection_frame_radius', index=11,
|
||||
number=5, type=5, cpp_type=1, label=1,
|
||||
has_default_value=True, default_value=5,
|
||||
message_type=None, enum_type=None, containing_type=None,
|
||||
is_extension=False, extension_scope=None,
|
||||
serialized_options=None, file=DESCRIPTOR),
|
||||
_descriptor.FieldDescriptor(
|
||||
name='selection_support_distance', full_name='mediapipe.MotionSaliencyOptions.selection_support_distance', index=12,
|
||||
number=6, type=2, cpp_type=6, label=1,
|
||||
has_default_value=True, default_value=float(0.2),
|
||||
message_type=None, enum_type=None, containing_type=None,
|
||||
is_extension=False, extension_scope=None,
|
||||
serialized_options=None, file=DESCRIPTOR),
|
||||
_descriptor.FieldDescriptor(
|
||||
name='selection_minimum_support', full_name='mediapipe.MotionSaliencyOptions.selection_minimum_support', index=13,
|
||||
number=7, type=5, cpp_type=1, label=1,
|
||||
has_default_value=True, default_value=4,
|
||||
message_type=None, enum_type=None, containing_type=None,
|
||||
is_extension=False, extension_scope=None,
|
||||
serialized_options=None, file=DESCRIPTOR),
|
||||
_descriptor.FieldDescriptor(
|
||||
name='filtering_sigma_space', full_name='mediapipe.MotionSaliencyOptions.filtering_sigma_space', index=14,
|
||||
number=13, type=2, cpp_type=6, label=1,
|
||||
has_default_value=True, default_value=float(0.05),
|
||||
message_type=None, enum_type=None, containing_type=None,
|
||||
is_extension=False, extension_scope=None,
|
||||
serialized_options=None, file=DESCRIPTOR),
|
||||
_descriptor.FieldDescriptor(
|
||||
name='filtering_sigma_time', full_name='mediapipe.MotionSaliencyOptions.filtering_sigma_time', index=15,
|
||||
number=14, type=2, cpp_type=6, label=1,
|
||||
has_default_value=True, default_value=float(5),
|
||||
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=61,
|
||||
serialized_end=610,
|
||||
)
|
||||
|
||||
DESCRIPTOR.message_types_by_name['MotionSaliencyOptions'] = _MOTIONSALIENCYOPTIONS
|
||||
_sym_db.RegisterFileDescriptor(DESCRIPTOR)
|
||||
|
||||
MotionSaliencyOptions = _reflection.GeneratedProtocolMessageType('MotionSaliencyOptions', (_message.Message,), dict(
|
||||
DESCRIPTOR = _MOTIONSALIENCYOPTIONS,
|
||||
__module__ = 'mediapipe.util.tracking.motion_saliency_pb2'
|
||||
# @@protoc_insertion_point(class_scope:mediapipe.MotionSaliencyOptions)
|
||||
))
|
||||
_sym_db.RegisterMessage(MotionSaliencyOptions)
|
||||
|
||||
|
||||
# @@protoc_insertion_point(module_scope)
|
||||
@@ -0,0 +1,98 @@
|
||||
# -*- coding: utf-8 -*-
|
||||
# Generated by the protocol buffer compiler. DO NOT EDIT!
|
||||
# source: mediapipe/util/tracking/push_pull_filtering.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()
|
||||
|
||||
|
||||
|
||||
|
||||
DESCRIPTOR = _descriptor.FileDescriptor(
|
||||
name='mediapipe/util/tracking/push_pull_filtering.proto',
|
||||
package='mediapipe',
|
||||
syntax='proto2',
|
||||
serialized_options=None,
|
||||
serialized_pb=_b('\n1mediapipe/util/tracking/push_pull_filtering.proto\x12\tmediapipe\"\xc0\x01\n\x0fPushPullOptions\x12\x1b\n\x0f\x62ilateral_sigma\x18\x01 \x01(\x02:\x02\x32\x30\x12!\n\x16pull_propagation_scale\x18\x03 \x01(\x02:\x01\x38\x12!\n\x16push_propagation_scale\x18\x04 \x01(\x02:\x01\x38\x12!\n\x14pull_bilateral_scale\x18\x05 \x01(\x02:\x03\x30.7\x12!\n\x14push_bilateral_scale\x18\x06 \x01(\x02:\x03\x30.9*\x04\x08\x02\x10\x03')
|
||||
)
|
||||
|
||||
|
||||
|
||||
|
||||
_PUSHPULLOPTIONS = _descriptor.Descriptor(
|
||||
name='PushPullOptions',
|
||||
full_name='mediapipe.PushPullOptions',
|
||||
filename=None,
|
||||
file=DESCRIPTOR,
|
||||
containing_type=None,
|
||||
fields=[
|
||||
_descriptor.FieldDescriptor(
|
||||
name='bilateral_sigma', full_name='mediapipe.PushPullOptions.bilateral_sigma', index=0,
|
||||
number=1, type=2, cpp_type=6, label=1,
|
||||
has_default_value=True, default_value=float(20),
|
||||
message_type=None, enum_type=None, containing_type=None,
|
||||
is_extension=False, extension_scope=None,
|
||||
serialized_options=None, file=DESCRIPTOR),
|
||||
_descriptor.FieldDescriptor(
|
||||
name='pull_propagation_scale', full_name='mediapipe.PushPullOptions.pull_propagation_scale', index=1,
|
||||
number=3, type=2, cpp_type=6, label=1,
|
||||
has_default_value=True, default_value=float(8),
|
||||
message_type=None, enum_type=None, containing_type=None,
|
||||
is_extension=False, extension_scope=None,
|
||||
serialized_options=None, file=DESCRIPTOR),
|
||||
_descriptor.FieldDescriptor(
|
||||
name='push_propagation_scale', full_name='mediapipe.PushPullOptions.push_propagation_scale', index=2,
|
||||
number=4, type=2, cpp_type=6, label=1,
|
||||
has_default_value=True, default_value=float(8),
|
||||
message_type=None, enum_type=None, containing_type=None,
|
||||
is_extension=False, extension_scope=None,
|
||||
serialized_options=None, file=DESCRIPTOR),
|
||||
_descriptor.FieldDescriptor(
|
||||
name='pull_bilateral_scale', full_name='mediapipe.PushPullOptions.pull_bilateral_scale', index=3,
|
||||
number=5, type=2, cpp_type=6, label=1,
|
||||
has_default_value=True, default_value=float(0.7),
|
||||
message_type=None, enum_type=None, containing_type=None,
|
||||
is_extension=False, extension_scope=None,
|
||||
serialized_options=None, file=DESCRIPTOR),
|
||||
_descriptor.FieldDescriptor(
|
||||
name='push_bilateral_scale', full_name='mediapipe.PushPullOptions.push_bilateral_scale', index=4,
|
||||
number=6, type=2, cpp_type=6, label=1,
|
||||
has_default_value=True, default_value=float(0.9),
|
||||
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=True,
|
||||
syntax='proto2',
|
||||
extension_ranges=[(2, 3), ],
|
||||
oneofs=[
|
||||
],
|
||||
serialized_start=65,
|
||||
serialized_end=257,
|
||||
)
|
||||
|
||||
DESCRIPTOR.message_types_by_name['PushPullOptions'] = _PUSHPULLOPTIONS
|
||||
_sym_db.RegisterFileDescriptor(DESCRIPTOR)
|
||||
|
||||
PushPullOptions = _reflection.GeneratedProtocolMessageType('PushPullOptions', (_message.Message,), dict(
|
||||
DESCRIPTOR = _PUSHPULLOPTIONS,
|
||||
__module__ = 'mediapipe.util.tracking.push_pull_filtering_pb2'
|
||||
# @@protoc_insertion_point(class_scope:mediapipe.PushPullOptions)
|
||||
))
|
||||
_sym_db.RegisterMessage(PushPullOptions)
|
||||
|
||||
|
||||
# @@protoc_insertion_point(module_scope)
|
||||
File diff suppressed because one or more lines are too long
@@ -0,0 +1,844 @@
|
||||
# -*- coding: utf-8 -*-
|
||||
# Generated by the protocol buffer compiler. DO NOT EDIT!
|
||||
# source: mediapipe/util/tracking/region_flow.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()
|
||||
|
||||
|
||||
|
||||
|
||||
DESCRIPTOR = _descriptor.FileDescriptor(
|
||||
name='mediapipe/util/tracking/region_flow.proto',
|
||||
package='mediapipe',
|
||||
syntax='proto2',
|
||||
serialized_options=_b('\n\035com.google.mediapipe.trackingP\001'),
|
||||
serialized_pb=_b('\n)mediapipe/util/tracking/region_flow.proto\x12\tmediapipe\"\x1f\n\x0fPatchDescriptor\x12\x0c\n\x04\x64\x61ta\x18\x01 \x03(\x02\"\'\n\x17\x42inaryFeatureDescriptor\x12\x0c\n\x04\x64\x61ta\x18\x01 \x01(\x0c\"D\n\x15TemporalIRLSSmoothing\x12\x15\n\nweight_sum\x18\x01 \x01(\x02:\x01\x30\x12\x14\n\tvalue_sum\x18\x02 \x01(\x02:\x01\x30\"\x99\x04\n\x11RegionFlowFeature\x12\x0c\n\x01x\x18\x01 \x01(\x02:\x01\x30\x12\x0c\n\x01y\x18\x02 \x01(\x02:\x01\x30\x12\r\n\x02\x64x\x18\x03 \x01(\x02:\x01\x30\x12\r\n\x02\x64y\x18\x04 \x01(\x02:\x01\x30\x12\x14\n\x08track_id\x18\r \x01(\x05:\x02-1\x12\x19\n\x0etracking_error\x18\x05 \x01(\x02:\x01\x30\x12\x16\n\x0birls_weight\x18\x06 \x01(\x02:\x01\x31\x12\x1a\n\x0f\x63orner_response\x18\x0b \x01(\x02:\x01\x30\x12\x36\n\x12\x66\x65\x61ture_descriptor\x18\x07 \x01(\x0b\x32\x1a.mediapipe.PatchDescriptor\x12<\n\x18\x66\x65\x61ture_match_descriptor\x18\x08 \x01(\x0b\x32\x1a.mediapipe.PatchDescriptor\x12\x37\n\rinternal_irls\x18\n \x01(\x0b\x32 .mediapipe.TemporalIRLSSmoothing\x12\r\n\x05label\x18\x0e \x01(\t\x12\r\n\x05\x66lags\x18\x0f \x01(\x05\x12\x12\n\nfeature_id\x18\x10 \x01(\x05\x12\x11\n\x06octave\x18\x11 \x01(\x05:\x01\x30\x12\x45\n\x19\x62inary_feature_descriptor\x18\x12 \x01(\x0b\x32\".mediapipe.BinaryFeatureDescriptor\"\x1e\n\x05\x46lags\x12\x15\n\x11\x46LAG_BROKEN_TRACK\x10\x01*\x04\x08\t\x10\n*\x04\x08\x0c\x10\r\"\xbd\x04\n\x0fRegionFlowFrame\x12:\n\x0bregion_flow\x18\x01 \x03(\x0b\x32%.mediapipe.RegionFlowFrame.RegionFlow\x12\x1d\n\x12num_total_features\x18\x02 \x01(\x05:\x01\x30\x12\x1d\n\x0eunstable_frame\x18\x04 \x01(\x08:\x05\x66\x61lse\x12\x12\n\nblur_score\x18\x07 \x01(\x02\x12\x13\n\x0b\x66rame_width\x18\x08 \x01(\x05\x12\x14\n\x0c\x66rame_height\x18\t \x01(\x05\x12\x44\n\x10\x62lock_descriptor\x18\n \x01(\x0b\x32*.mediapipe.RegionFlowFrame.BlockDescriptor\x1a\xa8\x01\n\nRegionFlow\x12\x11\n\tregion_id\x18\x01 \x02(\x05\x12\x15\n\ncentroid_x\x18\x02 \x01(\x02:\x01\x30\x12\x15\n\ncentroid_y\x18\x03 \x01(\x02:\x01\x30\x12\x11\n\x06\x66low_x\x18\x04 \x01(\x02:\x01\x30\x12\x11\n\x06\x66low_y\x18\x05 \x01(\x02:\x01\x30\x12-\n\x07\x66\x65\x61ture\x18\x07 \x03(\x0b\x32\x1c.mediapipe.RegionFlowFeature*\x04\x08\x06\x10\x07\x1an\n\x0f\x42lockDescriptor\x12\x13\n\x0b\x62lock_width\x18\x01 \x01(\x05\x12\x14\n\x0c\x62lock_height\x18\x02 \x01(\x05\x12\x17\n\x0cnum_blocks_x\x18\x03 \x01(\x05:\x01\x30\x12\x17\n\x0cnum_blocks_y\x18\x04 \x01(\x05:\x01\x30*\x04\x08\x03\x10\x04*\x04\x08\x05\x10\x06*\x04\x08\x06\x10\x07\"\x9c\x03\n\x15RegionFlowFeatureList\x12-\n\x07\x66\x65\x61ture\x18\x01 \x03(\x0b\x32\x1c.mediapipe.RegionFlowFeature\x12\x13\n\x0b\x66rame_width\x18\x02 \x01(\x05\x12\x14\n\x0c\x66rame_height\x18\x03 \x01(\x05\x12\x17\n\x08unstable\x18\x04 \x01(\x08:\x05\x66\x61lse\x12\x1f\n\x14\x64istance_from_border\x18\x05 \x01(\x05:\x01\x30\x12\x12\n\nblur_score\x18\x06 \x01(\x02\x12\x1a\n\x0blong_tracks\x18\x07 \x01(\x08:\x05\x66\x61lse\x12&\n\x1b\x66rac_long_features_rejected\x18\x08 \x01(\x02:\x01\x30\x12\x1e\n\x12visual_consistency\x18\t \x01(\x02:\x02-1\x12\x19\n\x0etimestamp_usec\x18\n \x01(\x03:\x01\x30\x12\x16\n\x0bmatch_frame\x18\x0b \x01(\x05:\x01\x30\x12\x1c\n\ris_duplicated\x18\x0c \x01(\x08:\x05\x66\x61lse\x12&\n\x1e\x61\x63tively_discarded_tracked_ids\x18\r \x03(\x05\"\x80\x03\n\x0cSalientPoint\x12\x17\n\x0cnorm_point_x\x18\x01 \x01(\x02:\x01\x30\x12\x17\n\x0cnorm_point_y\x18\x02 \x01(\x02:\x01\x30\x12\x44\n\x04type\x18\x0b \x01(\x0e\x32(.mediapipe.SalientPoint.SalientPointType:\x0cTYPE_INCLUDE\x12\x11\n\x04left\x18\x03 \x01(\x02:\x03\x30.3\x12\x13\n\x06\x62ottom\x18\x04 \x01(\x02:\x03\x30.3\x12\x12\n\x05right\x18\t \x01(\x02:\x03\x30.3\x12\x10\n\x03top\x18\n \x01(\x02:\x03\x30.3\x12\x12\n\x06weight\x18\x05 \x01(\x02:\x02\x31\x35\x12\x12\n\nnorm_major\x18\x06 \x01(\x02\x12\x12\n\nnorm_minor\x18\x07 \x01(\x02\x12\r\n\x05\x61ngle\x18\x08 \x01(\x02\"S\n\x10SalientPointType\x12\x10\n\x0cTYPE_INCLUDE\x10\x01\x12\x15\n\x11TYPE_EXCLUDE_LEFT\x10\x02\x12\x16\n\x12TYPE_EXCLUDE_RIGHT\x10\x03*\n\x08\xa0\x9c\x01\x10\x80\x80\x80\x80\x02\"G\n\x11SalientPointFrame\x12&\n\x05point\x18\x01 \x03(\x0b\x32\x17.mediapipe.SalientPoint*\n\x08\xa0\x9c\x01\x10\x80\x80\x80\x80\x02\x42!\n\x1d\x63om.google.mediapipe.trackingP\x01')
|
||||
)
|
||||
|
||||
|
||||
|
||||
_REGIONFLOWFEATURE_FLAGS = _descriptor.EnumDescriptor(
|
||||
name='Flags',
|
||||
full_name='mediapipe.RegionFlowFeature.Flags',
|
||||
filename=None,
|
||||
file=DESCRIPTOR,
|
||||
values=[
|
||||
_descriptor.EnumValueDescriptor(
|
||||
name='FLAG_BROKEN_TRACK', index=0, number=1,
|
||||
serialized_options=None,
|
||||
type=None),
|
||||
],
|
||||
containing_type=None,
|
||||
serialized_options=None,
|
||||
serialized_start=696,
|
||||
serialized_end=726,
|
||||
)
|
||||
_sym_db.RegisterEnumDescriptor(_REGIONFLOWFEATURE_FLAGS)
|
||||
|
||||
_SALIENTPOINT_SALIENTPOINTTYPE = _descriptor.EnumDescriptor(
|
||||
name='SalientPointType',
|
||||
full_name='mediapipe.SalientPoint.SalientPointType',
|
||||
filename=None,
|
||||
file=DESCRIPTOR,
|
||||
values=[
|
||||
_descriptor.EnumValueDescriptor(
|
||||
name='TYPE_INCLUDE', index=0, number=1,
|
||||
serialized_options=None,
|
||||
type=None),
|
||||
_descriptor.EnumValueDescriptor(
|
||||
name='TYPE_EXCLUDE_LEFT', index=1, number=2,
|
||||
serialized_options=None,
|
||||
type=None),
|
||||
_descriptor.EnumValueDescriptor(
|
||||
name='TYPE_EXCLUDE_RIGHT', index=2, number=3,
|
||||
serialized_options=None,
|
||||
type=None),
|
||||
],
|
||||
containing_type=None,
|
||||
serialized_options=None,
|
||||
serialized_start=2021,
|
||||
serialized_end=2104,
|
||||
)
|
||||
_sym_db.RegisterEnumDescriptor(_SALIENTPOINT_SALIENTPOINTTYPE)
|
||||
|
||||
|
||||
_PATCHDESCRIPTOR = _descriptor.Descriptor(
|
||||
name='PatchDescriptor',
|
||||
full_name='mediapipe.PatchDescriptor',
|
||||
filename=None,
|
||||
file=DESCRIPTOR,
|
||||
containing_type=None,
|
||||
fields=[
|
||||
_descriptor.FieldDescriptor(
|
||||
name='data', full_name='mediapipe.PatchDescriptor.data', index=0,
|
||||
number=1, 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),
|
||||
],
|
||||
extensions=[
|
||||
],
|
||||
nested_types=[],
|
||||
enum_types=[
|
||||
],
|
||||
serialized_options=None,
|
||||
is_extendable=False,
|
||||
syntax='proto2',
|
||||
extension_ranges=[],
|
||||
oneofs=[
|
||||
],
|
||||
serialized_start=56,
|
||||
serialized_end=87,
|
||||
)
|
||||
|
||||
|
||||
_BINARYFEATUREDESCRIPTOR = _descriptor.Descriptor(
|
||||
name='BinaryFeatureDescriptor',
|
||||
full_name='mediapipe.BinaryFeatureDescriptor',
|
||||
filename=None,
|
||||
file=DESCRIPTOR,
|
||||
containing_type=None,
|
||||
fields=[
|
||||
_descriptor.FieldDescriptor(
|
||||
name='data', full_name='mediapipe.BinaryFeatureDescriptor.data', index=0,
|
||||
number=1, type=12, cpp_type=9, label=1,
|
||||
has_default_value=False, default_value=_b(""),
|
||||
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=89,
|
||||
serialized_end=128,
|
||||
)
|
||||
|
||||
|
||||
_TEMPORALIRLSSMOOTHING = _descriptor.Descriptor(
|
||||
name='TemporalIRLSSmoothing',
|
||||
full_name='mediapipe.TemporalIRLSSmoothing',
|
||||
filename=None,
|
||||
file=DESCRIPTOR,
|
||||
containing_type=None,
|
||||
fields=[
|
||||
_descriptor.FieldDescriptor(
|
||||
name='weight_sum', full_name='mediapipe.TemporalIRLSSmoothing.weight_sum', index=0,
|
||||
number=1, 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='value_sum', full_name='mediapipe.TemporalIRLSSmoothing.value_sum', index=1,
|
||||
number=2, 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),
|
||||
],
|
||||
extensions=[
|
||||
],
|
||||
nested_types=[],
|
||||
enum_types=[
|
||||
],
|
||||
serialized_options=None,
|
||||
is_extendable=False,
|
||||
syntax='proto2',
|
||||
extension_ranges=[],
|
||||
oneofs=[
|
||||
],
|
||||
serialized_start=130,
|
||||
serialized_end=198,
|
||||
)
|
||||
|
||||
|
||||
_REGIONFLOWFEATURE = _descriptor.Descriptor(
|
||||
name='RegionFlowFeature',
|
||||
full_name='mediapipe.RegionFlowFeature',
|
||||
filename=None,
|
||||
file=DESCRIPTOR,
|
||||
containing_type=None,
|
||||
fields=[
|
||||
_descriptor.FieldDescriptor(
|
||||
name='x', full_name='mediapipe.RegionFlowFeature.x', index=0,
|
||||
number=1, 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', full_name='mediapipe.RegionFlowFeature.y', index=1,
|
||||
number=2, 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='dx', full_name='mediapipe.RegionFlowFeature.dx', index=2,
|
||||
number=3, 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='dy', full_name='mediapipe.RegionFlowFeature.dy', index=3,
|
||||
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='track_id', full_name='mediapipe.RegionFlowFeature.track_id', index=4,
|
||||
number=13, 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='tracking_error', full_name='mediapipe.RegionFlowFeature.tracking_error', index=5,
|
||||
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='irls_weight', full_name='mediapipe.RegionFlowFeature.irls_weight', index=6,
|
||||
number=6, 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='corner_response', full_name='mediapipe.RegionFlowFeature.corner_response', index=7,
|
||||
number=11, 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='feature_descriptor', full_name='mediapipe.RegionFlowFeature.feature_descriptor', index=8,
|
||||
number=7, 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='feature_match_descriptor', full_name='mediapipe.RegionFlowFeature.feature_match_descriptor', index=9,
|
||||
number=8, 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='internal_irls', full_name='mediapipe.RegionFlowFeature.internal_irls', index=10,
|
||||
number=10, 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='label', full_name='mediapipe.RegionFlowFeature.label', index=11,
|
||||
number=14, 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='flags', full_name='mediapipe.RegionFlowFeature.flags', index=12,
|
||||
number=15, 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='feature_id', full_name='mediapipe.RegionFlowFeature.feature_id', index=13,
|
||||
number=16, 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='octave', full_name='mediapipe.RegionFlowFeature.octave', index=14,
|
||||
number=17, 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='binary_feature_descriptor', full_name='mediapipe.RegionFlowFeature.binary_feature_descriptor', index=15,
|
||||
number=18, 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=[],
|
||||
enum_types=[
|
||||
_REGIONFLOWFEATURE_FLAGS,
|
||||
],
|
||||
serialized_options=None,
|
||||
is_extendable=True,
|
||||
syntax='proto2',
|
||||
extension_ranges=[(9, 10), (12, 13), ],
|
||||
oneofs=[
|
||||
],
|
||||
serialized_start=201,
|
||||
serialized_end=738,
|
||||
)
|
||||
|
||||
|
||||
_REGIONFLOWFRAME_REGIONFLOW = _descriptor.Descriptor(
|
||||
name='RegionFlow',
|
||||
full_name='mediapipe.RegionFlowFrame.RegionFlow',
|
||||
filename=None,
|
||||
file=DESCRIPTOR,
|
||||
containing_type=None,
|
||||
fields=[
|
||||
_descriptor.FieldDescriptor(
|
||||
name='region_id', full_name='mediapipe.RegionFlowFrame.RegionFlow.region_id', 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='centroid_x', full_name='mediapipe.RegionFlowFrame.RegionFlow.centroid_x', index=1,
|
||||
number=2, 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='centroid_y', full_name='mediapipe.RegionFlowFrame.RegionFlow.centroid_y', index=2,
|
||||
number=3, 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='flow_x', full_name='mediapipe.RegionFlowFrame.RegionFlow.flow_x', index=3,
|
||||
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='flow_y', full_name='mediapipe.RegionFlowFrame.RegionFlow.flow_y', index=4,
|
||||
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='feature', full_name='mediapipe.RegionFlowFrame.RegionFlow.feature', index=5,
|
||||
number=7, type=11, cpp_type=10, 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),
|
||||
],
|
||||
extensions=[
|
||||
],
|
||||
nested_types=[],
|
||||
enum_types=[
|
||||
],
|
||||
serialized_options=None,
|
||||
is_extendable=True,
|
||||
syntax='proto2',
|
||||
extension_ranges=[(6, 7), ],
|
||||
oneofs=[
|
||||
],
|
||||
serialized_start=1016,
|
||||
serialized_end=1184,
|
||||
)
|
||||
|
||||
_REGIONFLOWFRAME_BLOCKDESCRIPTOR = _descriptor.Descriptor(
|
||||
name='BlockDescriptor',
|
||||
full_name='mediapipe.RegionFlowFrame.BlockDescriptor',
|
||||
filename=None,
|
||||
file=DESCRIPTOR,
|
||||
containing_type=None,
|
||||
fields=[
|
||||
_descriptor.FieldDescriptor(
|
||||
name='block_width', full_name='mediapipe.RegionFlowFrame.BlockDescriptor.block_width', index=0,
|
||||
number=1, 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='block_height', full_name='mediapipe.RegionFlowFrame.BlockDescriptor.block_height', 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='num_blocks_x', full_name='mediapipe.RegionFlowFrame.BlockDescriptor.num_blocks_x', index=2,
|
||||
number=3, 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_blocks_y', full_name='mediapipe.RegionFlowFrame.BlockDescriptor.num_blocks_y', index=3,
|
||||
number=4, 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),
|
||||
],
|
||||
extensions=[
|
||||
],
|
||||
nested_types=[],
|
||||
enum_types=[
|
||||
],
|
||||
serialized_options=None,
|
||||
is_extendable=False,
|
||||
syntax='proto2',
|
||||
extension_ranges=[],
|
||||
oneofs=[
|
||||
],
|
||||
serialized_start=1186,
|
||||
serialized_end=1296,
|
||||
)
|
||||
|
||||
_REGIONFLOWFRAME = _descriptor.Descriptor(
|
||||
name='RegionFlowFrame',
|
||||
full_name='mediapipe.RegionFlowFrame',
|
||||
filename=None,
|
||||
file=DESCRIPTOR,
|
||||
containing_type=None,
|
||||
fields=[
|
||||
_descriptor.FieldDescriptor(
|
||||
name='region_flow', full_name='mediapipe.RegionFlowFrame.region_flow', index=0,
|
||||
number=1, type=11, cpp_type=10, 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='num_total_features', full_name='mediapipe.RegionFlowFrame.num_total_features', index=1,
|
||||
number=2, 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='unstable_frame', full_name='mediapipe.RegionFlowFrame.unstable_frame', index=2,
|
||||
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='blur_score', full_name='mediapipe.RegionFlowFrame.blur_score', index=3,
|
||||
number=7, 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='frame_width', full_name='mediapipe.RegionFlowFrame.frame_width', index=4,
|
||||
number=8, 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='frame_height', full_name='mediapipe.RegionFlowFrame.frame_height', index=5,
|
||||
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='block_descriptor', full_name='mediapipe.RegionFlowFrame.block_descriptor', index=6,
|
||||
number=10, 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=[_REGIONFLOWFRAME_REGIONFLOW, _REGIONFLOWFRAME_BLOCKDESCRIPTOR, ],
|
||||
enum_types=[
|
||||
],
|
||||
serialized_options=None,
|
||||
is_extendable=True,
|
||||
syntax='proto2',
|
||||
extension_ranges=[(3, 4), (5, 6), (6, 7), ],
|
||||
oneofs=[
|
||||
],
|
||||
serialized_start=741,
|
||||
serialized_end=1314,
|
||||
)
|
||||
|
||||
|
||||
_REGIONFLOWFEATURELIST = _descriptor.Descriptor(
|
||||
name='RegionFlowFeatureList',
|
||||
full_name='mediapipe.RegionFlowFeatureList',
|
||||
filename=None,
|
||||
file=DESCRIPTOR,
|
||||
containing_type=None,
|
||||
fields=[
|
||||
_descriptor.FieldDescriptor(
|
||||
name='feature', full_name='mediapipe.RegionFlowFeatureList.feature', index=0,
|
||||
number=1, type=11, cpp_type=10, 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='frame_width', full_name='mediapipe.RegionFlowFeatureList.frame_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='frame_height', full_name='mediapipe.RegionFlowFeatureList.frame_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='unstable', full_name='mediapipe.RegionFlowFeatureList.unstable', 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='distance_from_border', full_name='mediapipe.RegionFlowFeatureList.distance_from_border', index=4,
|
||||
number=5, 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='blur_score', full_name='mediapipe.RegionFlowFeatureList.blur_score', index=5,
|
||||
number=6, 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='long_tracks', full_name='mediapipe.RegionFlowFeatureList.long_tracks', index=6,
|
||||
number=7, 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='frac_long_features_rejected', full_name='mediapipe.RegionFlowFeatureList.frac_long_features_rejected', index=7,
|
||||
number=8, 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='visual_consistency', full_name='mediapipe.RegionFlowFeatureList.visual_consistency', index=8,
|
||||
number=9, 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='timestamp_usec', full_name='mediapipe.RegionFlowFeatureList.timestamp_usec', index=9,
|
||||
number=10, type=3, cpp_type=2, 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='match_frame', full_name='mediapipe.RegionFlowFeatureList.match_frame', index=10,
|
||||
number=11, 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='is_duplicated', full_name='mediapipe.RegionFlowFeatureList.is_duplicated', 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='actively_discarded_tracked_ids', full_name='mediapipe.RegionFlowFeatureList.actively_discarded_tracked_ids', index=12,
|
||||
number=13, 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),
|
||||
],
|
||||
extensions=[
|
||||
],
|
||||
nested_types=[],
|
||||
enum_types=[
|
||||
],
|
||||
serialized_options=None,
|
||||
is_extendable=False,
|
||||
syntax='proto2',
|
||||
extension_ranges=[],
|
||||
oneofs=[
|
||||
],
|
||||
serialized_start=1317,
|
||||
serialized_end=1729,
|
||||
)
|
||||
|
||||
|
||||
_SALIENTPOINT = _descriptor.Descriptor(
|
||||
name='SalientPoint',
|
||||
full_name='mediapipe.SalientPoint',
|
||||
filename=None,
|
||||
file=DESCRIPTOR,
|
||||
containing_type=None,
|
||||
fields=[
|
||||
_descriptor.FieldDescriptor(
|
||||
name='norm_point_x', full_name='mediapipe.SalientPoint.norm_point_x', index=0,
|
||||
number=1, 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='norm_point_y', full_name='mediapipe.SalientPoint.norm_point_y', index=1,
|
||||
number=2, 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='type', full_name='mediapipe.SalientPoint.type', index=2,
|
||||
number=11, type=14, cpp_type=8, 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='left', full_name='mediapipe.SalientPoint.left', index=3,
|
||||
number=3, type=2, cpp_type=6, label=1,
|
||||
has_default_value=True, default_value=float(0.3),
|
||||
message_type=None, enum_type=None, containing_type=None,
|
||||
is_extension=False, extension_scope=None,
|
||||
serialized_options=None, file=DESCRIPTOR),
|
||||
_descriptor.FieldDescriptor(
|
||||
name='bottom', full_name='mediapipe.SalientPoint.bottom', index=4,
|
||||
number=4, type=2, cpp_type=6, label=1,
|
||||
has_default_value=True, default_value=float(0.3),
|
||||
message_type=None, enum_type=None, containing_type=None,
|
||||
is_extension=False, extension_scope=None,
|
||||
serialized_options=None, file=DESCRIPTOR),
|
||||
_descriptor.FieldDescriptor(
|
||||
name='right', full_name='mediapipe.SalientPoint.right', index=5,
|
||||
number=9, type=2, cpp_type=6, label=1,
|
||||
has_default_value=True, default_value=float(0.3),
|
||||
message_type=None, enum_type=None, containing_type=None,
|
||||
is_extension=False, extension_scope=None,
|
||||
serialized_options=None, file=DESCRIPTOR),
|
||||
_descriptor.FieldDescriptor(
|
||||
name='top', full_name='mediapipe.SalientPoint.top', index=6,
|
||||
number=10, type=2, cpp_type=6, label=1,
|
||||
has_default_value=True, default_value=float(0.3),
|
||||
message_type=None, enum_type=None, containing_type=None,
|
||||
is_extension=False, extension_scope=None,
|
||||
serialized_options=None, file=DESCRIPTOR),
|
||||
_descriptor.FieldDescriptor(
|
||||
name='weight', full_name='mediapipe.SalientPoint.weight', index=7,
|
||||
number=5, type=2, cpp_type=6, label=1,
|
||||
has_default_value=True, default_value=float(15),
|
||||
message_type=None, enum_type=None, containing_type=None,
|
||||
is_extension=False, extension_scope=None,
|
||||
serialized_options=None, file=DESCRIPTOR),
|
||||
_descriptor.FieldDescriptor(
|
||||
name='norm_major', full_name='mediapipe.SalientPoint.norm_major', index=8,
|
||||
number=6, 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='norm_minor', full_name='mediapipe.SalientPoint.norm_minor', index=9,
|
||||
number=7, 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='angle', full_name='mediapipe.SalientPoint.angle', index=10,
|
||||
number=8, 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=[
|
||||
_SALIENTPOINT_SALIENTPOINTTYPE,
|
||||
],
|
||||
serialized_options=None,
|
||||
is_extendable=True,
|
||||
syntax='proto2',
|
||||
extension_ranges=[(20000, 536870912), ],
|
||||
oneofs=[
|
||||
],
|
||||
serialized_start=1732,
|
||||
serialized_end=2116,
|
||||
)
|
||||
|
||||
|
||||
_SALIENTPOINTFRAME = _descriptor.Descriptor(
|
||||
name='SalientPointFrame',
|
||||
full_name='mediapipe.SalientPointFrame',
|
||||
filename=None,
|
||||
file=DESCRIPTOR,
|
||||
containing_type=None,
|
||||
fields=[
|
||||
_descriptor.FieldDescriptor(
|
||||
name='point', full_name='mediapipe.SalientPointFrame.point', index=0,
|
||||
number=1, type=11, cpp_type=10, 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),
|
||||
],
|
||||
extensions=[
|
||||
],
|
||||
nested_types=[],
|
||||
enum_types=[
|
||||
],
|
||||
serialized_options=None,
|
||||
is_extendable=True,
|
||||
syntax='proto2',
|
||||
extension_ranges=[(20000, 536870912), ],
|
||||
oneofs=[
|
||||
],
|
||||
serialized_start=2118,
|
||||
serialized_end=2189,
|
||||
)
|
||||
|
||||
_REGIONFLOWFEATURE.fields_by_name['feature_descriptor'].message_type = _PATCHDESCRIPTOR
|
||||
_REGIONFLOWFEATURE.fields_by_name['feature_match_descriptor'].message_type = _PATCHDESCRIPTOR
|
||||
_REGIONFLOWFEATURE.fields_by_name['internal_irls'].message_type = _TEMPORALIRLSSMOOTHING
|
||||
_REGIONFLOWFEATURE.fields_by_name['binary_feature_descriptor'].message_type = _BINARYFEATUREDESCRIPTOR
|
||||
_REGIONFLOWFEATURE_FLAGS.containing_type = _REGIONFLOWFEATURE
|
||||
_REGIONFLOWFRAME_REGIONFLOW.fields_by_name['feature'].message_type = _REGIONFLOWFEATURE
|
||||
_REGIONFLOWFRAME_REGIONFLOW.containing_type = _REGIONFLOWFRAME
|
||||
_REGIONFLOWFRAME_BLOCKDESCRIPTOR.containing_type = _REGIONFLOWFRAME
|
||||
_REGIONFLOWFRAME.fields_by_name['region_flow'].message_type = _REGIONFLOWFRAME_REGIONFLOW
|
||||
_REGIONFLOWFRAME.fields_by_name['block_descriptor'].message_type = _REGIONFLOWFRAME_BLOCKDESCRIPTOR
|
||||
_REGIONFLOWFEATURELIST.fields_by_name['feature'].message_type = _REGIONFLOWFEATURE
|
||||
_SALIENTPOINT.fields_by_name['type'].enum_type = _SALIENTPOINT_SALIENTPOINTTYPE
|
||||
_SALIENTPOINT_SALIENTPOINTTYPE.containing_type = _SALIENTPOINT
|
||||
_SALIENTPOINTFRAME.fields_by_name['point'].message_type = _SALIENTPOINT
|
||||
DESCRIPTOR.message_types_by_name['PatchDescriptor'] = _PATCHDESCRIPTOR
|
||||
DESCRIPTOR.message_types_by_name['BinaryFeatureDescriptor'] = _BINARYFEATUREDESCRIPTOR
|
||||
DESCRIPTOR.message_types_by_name['TemporalIRLSSmoothing'] = _TEMPORALIRLSSMOOTHING
|
||||
DESCRIPTOR.message_types_by_name['RegionFlowFeature'] = _REGIONFLOWFEATURE
|
||||
DESCRIPTOR.message_types_by_name['RegionFlowFrame'] = _REGIONFLOWFRAME
|
||||
DESCRIPTOR.message_types_by_name['RegionFlowFeatureList'] = _REGIONFLOWFEATURELIST
|
||||
DESCRIPTOR.message_types_by_name['SalientPoint'] = _SALIENTPOINT
|
||||
DESCRIPTOR.message_types_by_name['SalientPointFrame'] = _SALIENTPOINTFRAME
|
||||
_sym_db.RegisterFileDescriptor(DESCRIPTOR)
|
||||
|
||||
PatchDescriptor = _reflection.GeneratedProtocolMessageType('PatchDescriptor', (_message.Message,), dict(
|
||||
DESCRIPTOR = _PATCHDESCRIPTOR,
|
||||
__module__ = 'mediapipe.util.tracking.region_flow_pb2'
|
||||
# @@protoc_insertion_point(class_scope:mediapipe.PatchDescriptor)
|
||||
))
|
||||
_sym_db.RegisterMessage(PatchDescriptor)
|
||||
|
||||
BinaryFeatureDescriptor = _reflection.GeneratedProtocolMessageType('BinaryFeatureDescriptor', (_message.Message,), dict(
|
||||
DESCRIPTOR = _BINARYFEATUREDESCRIPTOR,
|
||||
__module__ = 'mediapipe.util.tracking.region_flow_pb2'
|
||||
# @@protoc_insertion_point(class_scope:mediapipe.BinaryFeatureDescriptor)
|
||||
))
|
||||
_sym_db.RegisterMessage(BinaryFeatureDescriptor)
|
||||
|
||||
TemporalIRLSSmoothing = _reflection.GeneratedProtocolMessageType('TemporalIRLSSmoothing', (_message.Message,), dict(
|
||||
DESCRIPTOR = _TEMPORALIRLSSMOOTHING,
|
||||
__module__ = 'mediapipe.util.tracking.region_flow_pb2'
|
||||
# @@protoc_insertion_point(class_scope:mediapipe.TemporalIRLSSmoothing)
|
||||
))
|
||||
_sym_db.RegisterMessage(TemporalIRLSSmoothing)
|
||||
|
||||
RegionFlowFeature = _reflection.GeneratedProtocolMessageType('RegionFlowFeature', (_message.Message,), dict(
|
||||
DESCRIPTOR = _REGIONFLOWFEATURE,
|
||||
__module__ = 'mediapipe.util.tracking.region_flow_pb2'
|
||||
# @@protoc_insertion_point(class_scope:mediapipe.RegionFlowFeature)
|
||||
))
|
||||
_sym_db.RegisterMessage(RegionFlowFeature)
|
||||
|
||||
RegionFlowFrame = _reflection.GeneratedProtocolMessageType('RegionFlowFrame', (_message.Message,), dict(
|
||||
|
||||
RegionFlow = _reflection.GeneratedProtocolMessageType('RegionFlow', (_message.Message,), dict(
|
||||
DESCRIPTOR = _REGIONFLOWFRAME_REGIONFLOW,
|
||||
__module__ = 'mediapipe.util.tracking.region_flow_pb2'
|
||||
# @@protoc_insertion_point(class_scope:mediapipe.RegionFlowFrame.RegionFlow)
|
||||
))
|
||||
,
|
||||
|
||||
BlockDescriptor = _reflection.GeneratedProtocolMessageType('BlockDescriptor', (_message.Message,), dict(
|
||||
DESCRIPTOR = _REGIONFLOWFRAME_BLOCKDESCRIPTOR,
|
||||
__module__ = 'mediapipe.util.tracking.region_flow_pb2'
|
||||
# @@protoc_insertion_point(class_scope:mediapipe.RegionFlowFrame.BlockDescriptor)
|
||||
))
|
||||
,
|
||||
DESCRIPTOR = _REGIONFLOWFRAME,
|
||||
__module__ = 'mediapipe.util.tracking.region_flow_pb2'
|
||||
# @@protoc_insertion_point(class_scope:mediapipe.RegionFlowFrame)
|
||||
))
|
||||
_sym_db.RegisterMessage(RegionFlowFrame)
|
||||
_sym_db.RegisterMessage(RegionFlowFrame.RegionFlow)
|
||||
_sym_db.RegisterMessage(RegionFlowFrame.BlockDescriptor)
|
||||
|
||||
RegionFlowFeatureList = _reflection.GeneratedProtocolMessageType('RegionFlowFeatureList', (_message.Message,), dict(
|
||||
DESCRIPTOR = _REGIONFLOWFEATURELIST,
|
||||
__module__ = 'mediapipe.util.tracking.region_flow_pb2'
|
||||
# @@protoc_insertion_point(class_scope:mediapipe.RegionFlowFeatureList)
|
||||
))
|
||||
_sym_db.RegisterMessage(RegionFlowFeatureList)
|
||||
|
||||
SalientPoint = _reflection.GeneratedProtocolMessageType('SalientPoint', (_message.Message,), dict(
|
||||
DESCRIPTOR = _SALIENTPOINT,
|
||||
__module__ = 'mediapipe.util.tracking.region_flow_pb2'
|
||||
# @@protoc_insertion_point(class_scope:mediapipe.SalientPoint)
|
||||
))
|
||||
_sym_db.RegisterMessage(SalientPoint)
|
||||
|
||||
SalientPointFrame = _reflection.GeneratedProtocolMessageType('SalientPointFrame', (_message.Message,), dict(
|
||||
DESCRIPTOR = _SALIENTPOINTFRAME,
|
||||
__module__ = 'mediapipe.util.tracking.region_flow_pb2'
|
||||
# @@protoc_insertion_point(class_scope:mediapipe.SalientPointFrame)
|
||||
))
|
||||
_sym_db.RegisterMessage(SalientPointFrame)
|
||||
|
||||
|
||||
DESCRIPTOR._options = None
|
||||
# @@protoc_insertion_point(module_scope)
|
||||
@@ -0,0 +1,699 @@
|
||||
# -*- coding: utf-8 -*-
|
||||
# Generated by the protocol buffer compiler. DO NOT EDIT!
|
||||
# source: mediapipe/util/tracking/tone_estimation.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.util.tracking import tone_models_pb2 as mediapipe_dot_util_dot_tracking_dot_tone__models__pb2
|
||||
|
||||
|
||||
DESCRIPTOR = _descriptor.FileDescriptor(
|
||||
name='mediapipe/util/tracking/tone_estimation.proto',
|
||||
package='mediapipe',
|
||||
syntax='proto2',
|
||||
serialized_options=None,
|
||||
serialized_pb=_b('\n-mediapipe/util/tracking/tone_estimation.proto\x12\tmediapipe\x1a)mediapipe/util/tracking/tone_models.proto\"\x97\x05\n\nToneChange\x12+\n\tgain_bias\x18\x01 \x01(\x0b\x32\x18.mediapipe.GainBiasModel\x12*\n\x06\x61\x66\x66ine\x18\x02 \x01(\x0b\x32\x1a.mediapipe.AffineToneModel\x12:\n\x11mixture_gain_bias\x18\x03 \x01(\x0b\x32\x1f.mediapipe.MixtureGainBiasModel\x12\x39\n\x0emixture_affine\x18\x04 \x01(\x0b\x32!.mediapipe.MixtureAffineToneModel\x12\x1c\n\x14mixture_domain_sigma\x18\x05 \x01(\x02\x12\x17\n\x0c\x66rac_clipped\x18\x06 \x01(\x02:\x01\x30\x12\x16\n\x0elow_percentile\x18\x08 \x01(\x02\x12\x1a\n\x12low_mid_percentile\x18\t \x01(\x02\x12\x16\n\x0emid_percentile\x18\n \x01(\x02\x12\x1b\n\x13high_mid_percentile\x18\x0b \x01(\x02\x12\x17\n\x0fhigh_percentile\x18\x0c \x01(\x02\x12\x19\n\nlog_domain\x18\r \x01(\x08:\x05\x66\x61lse\x12/\n\x04type\x18\x0e \x01(\x0e\x32\x1a.mediapipe.ToneChange.Type:\x05VALID\x12=\n\x0fstability_stats\x18\x0f \x01(\x0b\x32$.mediapipe.ToneChange.StabilityStats\x1aU\n\x0eStabilityStats\x12\x13\n\x0bnum_inliers\x18\x01 \x01(\x05\x12\x17\n\x0finlier_fraction\x18\x02 \x01(\x02\x12\x15\n\rinlier_weight\x18\x03 \x01(\x01\"\x1e\n\x04Type\x12\t\n\x05VALID\x10\x00\x12\x0b\n\x07INVALID\x10\n\"\xd2\x01\n\x10ToneMatchOptions\x12\"\n\x14min_match_percentile\x18\x01 \x01(\x02:\x04\x30.01\x12\"\n\x14max_match_percentile\x18\x02 \x01(\x02:\x04\x30.99\x12\"\n\x16match_percentile_steps\x18\x03 \x01(\x05:\x02\x31\x30\x12\x18\n\x0cpatch_radius\x18\x04 \x01(\x05:\x02\x31\x38\x12\x1d\n\x10max_frac_clipped\x18\x05 \x01(\x02:\x03\x30.4\x12\x19\n\nlog_domain\x18\x08 \x01(\x08:\x05\x66\x61lse\"\x89\x01\n\x0f\x43lipMaskOptions\x12\x1a\n\x0cmin_exposure\x18\x01 \x01(\x02:\x04\x30.02\x12\x1a\n\x0cmax_exposure\x18\x02 \x01(\x02:\x04\x30.98\x12\x1f\n\x14max_clipped_channels\x18\x04 \x01(\x05:\x01\x31\x12\x1d\n\x12\x63lip_mask_diameter\x18\x05 \x01(\x05:\x01\x35\"\x81\x07\n\x15ToneEstimationOptions\x12\x37\n\x12tone_match_options\x18\x01 \x01(\x0b\x32\x1b.mediapipe.ToneMatchOptions\x12\x35\n\x11\x63lip_mask_options\x18\x02 \x01(\x0b\x32\x1a.mediapipe.ClipMaskOptions\x12\"\n\x14stats_low_percentile\x18\x03 \x01(\x02:\x04\x30.05\x12%\n\x18stats_low_mid_percentile\x18\x04 \x01(\x02:\x03\x30.2\x12!\n\x14stats_mid_percentile\x18\x05 \x01(\x02:\x03\x30.5\x12&\n\x19stats_high_mid_percentile\x18\x06 \x01(\x02:\x03\x30.8\x12#\n\x15stats_high_percentile\x18\x07 \x01(\x02:\x04\x30.95\x12\x1b\n\x0firls_iterations\x18\x08 \x01(\x05:\x02\x31\x30\x12P\n\x17stable_gain_bias_bounds\x18\t \x01(\x0b\x32/.mediapipe.ToneEstimationOptions.GainBiasBounds\x12Y\n\x0f\x64ownsample_mode\x18\n \x01(\x0e\x32/.mediapipe.ToneEstimationOptions.DownsampleMode:\x0f\x44OWNSAMPLE_NONE\x12\x1e\n\x11\x64ownsampling_size\x18\x0b \x01(\x05:\x03\x32\x35\x36\x12\x1c\n\x11\x64ownsample_factor\x18\x0c \x01(\x02:\x01\x32\x1a\xbb\x01\n\x0eGainBiasBounds\x12!\n\x13min_inlier_fraction\x18\x01 \x01(\x02:\x04\x30.75\x12\x1e\n\x11min_inlier_weight\x18\x02 \x01(\x02:\x03\x30.5\x12\x18\n\nlower_gain\x18\x03 \x01(\x02:\x04\x30.75\x12\x19\n\nupper_gain\x18\x04 \x01(\x02:\x05\x31.334\x12\x18\n\nlower_bias\x18\x05 \x01(\x02:\x04-0.2\x12\x17\n\nupper_bias\x18\x06 \x01(\x02:\x03\x30.2\"w\n\x0e\x44ownsampleMode\x12\x13\n\x0f\x44OWNSAMPLE_NONE\x10\x01\x12\x1a\n\x16\x44OWNSAMPLE_TO_MAX_SIZE\x10\x02\x12\x18\n\x14\x44OWNSAMPLE_BY_FACTOR\x10\x03\x12\x1a\n\x16\x44OWNSAMPLE_TO_MIN_SIZE\x10\x04\"/\n\tToneMatch\x12\x10\n\x08\x63urr_val\x18\x01 \x01(\x02\x12\x10\n\x08prev_val\x18\x02 \x01(\x02\"R\n\x0ePatchToneMatch\x12(\n\ntone_match\x18\x01 \x03(\x0b\x32\x14.mediapipe.ToneMatch\x12\x16\n\x0birls_weight\x18\x02 \x01(\x02:\x01\x31')
|
||||
,
|
||||
dependencies=[mediapipe_dot_util_dot_tracking_dot_tone__models__pb2.DESCRIPTOR,])
|
||||
|
||||
|
||||
|
||||
_TONECHANGE_TYPE = _descriptor.EnumDescriptor(
|
||||
name='Type',
|
||||
full_name='mediapipe.ToneChange.Type',
|
||||
filename=None,
|
||||
file=DESCRIPTOR,
|
||||
values=[
|
||||
_descriptor.EnumValueDescriptor(
|
||||
name='VALID', index=0, number=0,
|
||||
serialized_options=None,
|
||||
type=None),
|
||||
_descriptor.EnumValueDescriptor(
|
||||
name='INVALID', index=1, number=10,
|
||||
serialized_options=None,
|
||||
type=None),
|
||||
],
|
||||
containing_type=None,
|
||||
serialized_options=None,
|
||||
serialized_start=737,
|
||||
serialized_end=767,
|
||||
)
|
||||
_sym_db.RegisterEnumDescriptor(_TONECHANGE_TYPE)
|
||||
|
||||
_TONEESTIMATIONOPTIONS_DOWNSAMPLEMODE = _descriptor.EnumDescriptor(
|
||||
name='DownsampleMode',
|
||||
full_name='mediapipe.ToneEstimationOptions.DownsampleMode',
|
||||
filename=None,
|
||||
file=DESCRIPTOR,
|
||||
values=[
|
||||
_descriptor.EnumValueDescriptor(
|
||||
name='DOWNSAMPLE_NONE', index=0, number=1,
|
||||
serialized_options=None,
|
||||
type=None),
|
||||
_descriptor.EnumValueDescriptor(
|
||||
name='DOWNSAMPLE_TO_MAX_SIZE', index=1, number=2,
|
||||
serialized_options=None,
|
||||
type=None),
|
||||
_descriptor.EnumValueDescriptor(
|
||||
name='DOWNSAMPLE_BY_FACTOR', index=2, number=3,
|
||||
serialized_options=None,
|
||||
type=None),
|
||||
_descriptor.EnumValueDescriptor(
|
||||
name='DOWNSAMPLE_TO_MIN_SIZE', index=3, number=4,
|
||||
serialized_options=None,
|
||||
type=None),
|
||||
],
|
||||
containing_type=None,
|
||||
serialized_options=None,
|
||||
serialized_start=1901,
|
||||
serialized_end=2020,
|
||||
)
|
||||
_sym_db.RegisterEnumDescriptor(_TONEESTIMATIONOPTIONS_DOWNSAMPLEMODE)
|
||||
|
||||
|
||||
_TONECHANGE_STABILITYSTATS = _descriptor.Descriptor(
|
||||
name='StabilityStats',
|
||||
full_name='mediapipe.ToneChange.StabilityStats',
|
||||
filename=None,
|
||||
file=DESCRIPTOR,
|
||||
containing_type=None,
|
||||
fields=[
|
||||
_descriptor.FieldDescriptor(
|
||||
name='num_inliers', full_name='mediapipe.ToneChange.StabilityStats.num_inliers', index=0,
|
||||
number=1, 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='inlier_fraction', full_name='mediapipe.ToneChange.StabilityStats.inlier_fraction', 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),
|
||||
_descriptor.FieldDescriptor(
|
||||
name='inlier_weight', full_name='mediapipe.ToneChange.StabilityStats.inlier_weight', index=2,
|
||||
number=3, type=1, cpp_type=5, 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=650,
|
||||
serialized_end=735,
|
||||
)
|
||||
|
||||
_TONECHANGE = _descriptor.Descriptor(
|
||||
name='ToneChange',
|
||||
full_name='mediapipe.ToneChange',
|
||||
filename=None,
|
||||
file=DESCRIPTOR,
|
||||
containing_type=None,
|
||||
fields=[
|
||||
_descriptor.FieldDescriptor(
|
||||
name='gain_bias', full_name='mediapipe.ToneChange.gain_bias', 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='affine', full_name='mediapipe.ToneChange.affine', 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='mixture_gain_bias', full_name='mediapipe.ToneChange.mixture_gain_bias', 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='mixture_affine', full_name='mediapipe.ToneChange.mixture_affine', 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),
|
||||
_descriptor.FieldDescriptor(
|
||||
name='mixture_domain_sigma', full_name='mediapipe.ToneChange.mixture_domain_sigma', index=4,
|
||||
number=5, 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='frac_clipped', full_name='mediapipe.ToneChange.frac_clipped', index=5,
|
||||
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='low_percentile', full_name='mediapipe.ToneChange.low_percentile', index=6,
|
||||
number=8, 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='low_mid_percentile', full_name='mediapipe.ToneChange.low_mid_percentile', index=7,
|
||||
number=9, 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='mid_percentile', full_name='mediapipe.ToneChange.mid_percentile', index=8,
|
||||
number=10, 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='high_mid_percentile', full_name='mediapipe.ToneChange.high_mid_percentile', index=9,
|
||||
number=11, 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='high_percentile', full_name='mediapipe.ToneChange.high_percentile', index=10,
|
||||
number=12, 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='log_domain', full_name='mediapipe.ToneChange.log_domain', 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='type', full_name='mediapipe.ToneChange.type', index=12,
|
||||
number=14, 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='stability_stats', full_name='mediapipe.ToneChange.stability_stats', index=13,
|
||||
number=15, 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=[_TONECHANGE_STABILITYSTATS, ],
|
||||
enum_types=[
|
||||
_TONECHANGE_TYPE,
|
||||
],
|
||||
serialized_options=None,
|
||||
is_extendable=False,
|
||||
syntax='proto2',
|
||||
extension_ranges=[],
|
||||
oneofs=[
|
||||
],
|
||||
serialized_start=104,
|
||||
serialized_end=767,
|
||||
)
|
||||
|
||||
|
||||
_TONEMATCHOPTIONS = _descriptor.Descriptor(
|
||||
name='ToneMatchOptions',
|
||||
full_name='mediapipe.ToneMatchOptions',
|
||||
filename=None,
|
||||
file=DESCRIPTOR,
|
||||
containing_type=None,
|
||||
fields=[
|
||||
_descriptor.FieldDescriptor(
|
||||
name='min_match_percentile', full_name='mediapipe.ToneMatchOptions.min_match_percentile', index=0,
|
||||
number=1, type=2, cpp_type=6, label=1,
|
||||
has_default_value=True, default_value=float(0.01),
|
||||
message_type=None, enum_type=None, containing_type=None,
|
||||
is_extension=False, extension_scope=None,
|
||||
serialized_options=None, file=DESCRIPTOR),
|
||||
_descriptor.FieldDescriptor(
|
||||
name='max_match_percentile', full_name='mediapipe.ToneMatchOptions.max_match_percentile', index=1,
|
||||
number=2, type=2, cpp_type=6, label=1,
|
||||
has_default_value=True, default_value=float(0.99),
|
||||
message_type=None, enum_type=None, containing_type=None,
|
||||
is_extension=False, extension_scope=None,
|
||||
serialized_options=None, file=DESCRIPTOR),
|
||||
_descriptor.FieldDescriptor(
|
||||
name='match_percentile_steps', full_name='mediapipe.ToneMatchOptions.match_percentile_steps', index=2,
|
||||
number=3, type=5, cpp_type=1, label=1,
|
||||
has_default_value=True, default_value=10,
|
||||
message_type=None, enum_type=None, containing_type=None,
|
||||
is_extension=False, extension_scope=None,
|
||||
serialized_options=None, file=DESCRIPTOR),
|
||||
_descriptor.FieldDescriptor(
|
||||
name='patch_radius', full_name='mediapipe.ToneMatchOptions.patch_radius', index=3,
|
||||
number=4, type=5, cpp_type=1, label=1,
|
||||
has_default_value=True, default_value=18,
|
||||
message_type=None, enum_type=None, containing_type=None,
|
||||
is_extension=False, extension_scope=None,
|
||||
serialized_options=None, file=DESCRIPTOR),
|
||||
_descriptor.FieldDescriptor(
|
||||
name='max_frac_clipped', full_name='mediapipe.ToneMatchOptions.max_frac_clipped', index=4,
|
||||
number=5, type=2, cpp_type=6, label=1,
|
||||
has_default_value=True, default_value=float(0.4),
|
||||
message_type=None, enum_type=None, containing_type=None,
|
||||
is_extension=False, extension_scope=None,
|
||||
serialized_options=None, file=DESCRIPTOR),
|
||||
_descriptor.FieldDescriptor(
|
||||
name='log_domain', full_name='mediapipe.ToneMatchOptions.log_domain', index=5,
|
||||
number=8, 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=[
|
||||
],
|
||||
nested_types=[],
|
||||
enum_types=[
|
||||
],
|
||||
serialized_options=None,
|
||||
is_extendable=False,
|
||||
syntax='proto2',
|
||||
extension_ranges=[],
|
||||
oneofs=[
|
||||
],
|
||||
serialized_start=770,
|
||||
serialized_end=980,
|
||||
)
|
||||
|
||||
|
||||
_CLIPMASKOPTIONS = _descriptor.Descriptor(
|
||||
name='ClipMaskOptions',
|
||||
full_name='mediapipe.ClipMaskOptions',
|
||||
filename=None,
|
||||
file=DESCRIPTOR,
|
||||
containing_type=None,
|
||||
fields=[
|
||||
_descriptor.FieldDescriptor(
|
||||
name='min_exposure', full_name='mediapipe.ClipMaskOptions.min_exposure', index=0,
|
||||
number=1, type=2, cpp_type=6, label=1,
|
||||
has_default_value=True, default_value=float(0.02),
|
||||
message_type=None, enum_type=None, containing_type=None,
|
||||
is_extension=False, extension_scope=None,
|
||||
serialized_options=None, file=DESCRIPTOR),
|
||||
_descriptor.FieldDescriptor(
|
||||
name='max_exposure', full_name='mediapipe.ClipMaskOptions.max_exposure', index=1,
|
||||
number=2, type=2, cpp_type=6, label=1,
|
||||
has_default_value=True, default_value=float(0.98),
|
||||
message_type=None, enum_type=None, containing_type=None,
|
||||
is_extension=False, extension_scope=None,
|
||||
serialized_options=None, file=DESCRIPTOR),
|
||||
_descriptor.FieldDescriptor(
|
||||
name='max_clipped_channels', full_name='mediapipe.ClipMaskOptions.max_clipped_channels', index=2,
|
||||
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='clip_mask_diameter', full_name='mediapipe.ClipMaskOptions.clip_mask_diameter', index=3,
|
||||
number=5, type=5, cpp_type=1, label=1,
|
||||
has_default_value=True, default_value=5,
|
||||
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=983,
|
||||
serialized_end=1120,
|
||||
)
|
||||
|
||||
|
||||
_TONEESTIMATIONOPTIONS_GAINBIASBOUNDS = _descriptor.Descriptor(
|
||||
name='GainBiasBounds',
|
||||
full_name='mediapipe.ToneEstimationOptions.GainBiasBounds',
|
||||
filename=None,
|
||||
file=DESCRIPTOR,
|
||||
containing_type=None,
|
||||
fields=[
|
||||
_descriptor.FieldDescriptor(
|
||||
name='min_inlier_fraction', full_name='mediapipe.ToneEstimationOptions.GainBiasBounds.min_inlier_fraction', index=0,
|
||||
number=1, type=2, cpp_type=6, label=1,
|
||||
has_default_value=True, default_value=float(0.75),
|
||||
message_type=None, enum_type=None, containing_type=None,
|
||||
is_extension=False, extension_scope=None,
|
||||
serialized_options=None, file=DESCRIPTOR),
|
||||
_descriptor.FieldDescriptor(
|
||||
name='min_inlier_weight', full_name='mediapipe.ToneEstimationOptions.GainBiasBounds.min_inlier_weight', index=1,
|
||||
number=2, type=2, cpp_type=6, label=1,
|
||||
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='lower_gain', full_name='mediapipe.ToneEstimationOptions.GainBiasBounds.lower_gain', index=2,
|
||||
number=3, type=2, cpp_type=6, label=1,
|
||||
has_default_value=True, default_value=float(0.75),
|
||||
message_type=None, enum_type=None, containing_type=None,
|
||||
is_extension=False, extension_scope=None,
|
||||
serialized_options=None, file=DESCRIPTOR),
|
||||
_descriptor.FieldDescriptor(
|
||||
name='upper_gain', full_name='mediapipe.ToneEstimationOptions.GainBiasBounds.upper_gain', index=3,
|
||||
number=4, type=2, cpp_type=6, label=1,
|
||||
has_default_value=True, default_value=float(1.334),
|
||||
message_type=None, enum_type=None, containing_type=None,
|
||||
is_extension=False, extension_scope=None,
|
||||
serialized_options=None, file=DESCRIPTOR),
|
||||
_descriptor.FieldDescriptor(
|
||||
name='lower_bias', full_name='mediapipe.ToneEstimationOptions.GainBiasBounds.lower_bias', index=4,
|
||||
number=5, type=2, cpp_type=6, label=1,
|
||||
has_default_value=True, default_value=float(-0.2),
|
||||
message_type=None, enum_type=None, containing_type=None,
|
||||
is_extension=False, extension_scope=None,
|
||||
serialized_options=None, file=DESCRIPTOR),
|
||||
_descriptor.FieldDescriptor(
|
||||
name='upper_bias', full_name='mediapipe.ToneEstimationOptions.GainBiasBounds.upper_bias', index=5,
|
||||
number=6, type=2, cpp_type=6, label=1,
|
||||
has_default_value=True, default_value=float(0.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=[
|
||||
],
|
||||
serialized_options=None,
|
||||
is_extendable=False,
|
||||
syntax='proto2',
|
||||
extension_ranges=[],
|
||||
oneofs=[
|
||||
],
|
||||
serialized_start=1712,
|
||||
serialized_end=1899,
|
||||
)
|
||||
|
||||
_TONEESTIMATIONOPTIONS = _descriptor.Descriptor(
|
||||
name='ToneEstimationOptions',
|
||||
full_name='mediapipe.ToneEstimationOptions',
|
||||
filename=None,
|
||||
file=DESCRIPTOR,
|
||||
containing_type=None,
|
||||
fields=[
|
||||
_descriptor.FieldDescriptor(
|
||||
name='tone_match_options', full_name='mediapipe.ToneEstimationOptions.tone_match_options', 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='clip_mask_options', full_name='mediapipe.ToneEstimationOptions.clip_mask_options', 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='stats_low_percentile', full_name='mediapipe.ToneEstimationOptions.stats_low_percentile', index=2,
|
||||
number=3, type=2, cpp_type=6, label=1,
|
||||
has_default_value=True, default_value=float(0.05),
|
||||
message_type=None, enum_type=None, containing_type=None,
|
||||
is_extension=False, extension_scope=None,
|
||||
serialized_options=None, file=DESCRIPTOR),
|
||||
_descriptor.FieldDescriptor(
|
||||
name='stats_low_mid_percentile', full_name='mediapipe.ToneEstimationOptions.stats_low_mid_percentile', index=3,
|
||||
number=4, type=2, cpp_type=6, label=1,
|
||||
has_default_value=True, default_value=float(0.2),
|
||||
message_type=None, enum_type=None, containing_type=None,
|
||||
is_extension=False, extension_scope=None,
|
||||
serialized_options=None, file=DESCRIPTOR),
|
||||
_descriptor.FieldDescriptor(
|
||||
name='stats_mid_percentile', full_name='mediapipe.ToneEstimationOptions.stats_mid_percentile', index=4,
|
||||
number=5, type=2, cpp_type=6, label=1,
|
||||
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='stats_high_mid_percentile', full_name='mediapipe.ToneEstimationOptions.stats_high_mid_percentile', index=5,
|
||||
number=6, type=2, cpp_type=6, label=1,
|
||||
has_default_value=True, default_value=float(0.8),
|
||||
message_type=None, enum_type=None, containing_type=None,
|
||||
is_extension=False, extension_scope=None,
|
||||
serialized_options=None, file=DESCRIPTOR),
|
||||
_descriptor.FieldDescriptor(
|
||||
name='stats_high_percentile', full_name='mediapipe.ToneEstimationOptions.stats_high_percentile', index=6,
|
||||
number=7, type=2, cpp_type=6, label=1,
|
||||
has_default_value=True, default_value=float(0.95),
|
||||
message_type=None, enum_type=None, containing_type=None,
|
||||
is_extension=False, extension_scope=None,
|
||||
serialized_options=None, file=DESCRIPTOR),
|
||||
_descriptor.FieldDescriptor(
|
||||
name='irls_iterations', full_name='mediapipe.ToneEstimationOptions.irls_iterations', index=7,
|
||||
number=8, type=5, cpp_type=1, label=1,
|
||||
has_default_value=True, default_value=10,
|
||||
message_type=None, enum_type=None, containing_type=None,
|
||||
is_extension=False, extension_scope=None,
|
||||
serialized_options=None, file=DESCRIPTOR),
|
||||
_descriptor.FieldDescriptor(
|
||||
name='stable_gain_bias_bounds', full_name='mediapipe.ToneEstimationOptions.stable_gain_bias_bounds', 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),
|
||||
_descriptor.FieldDescriptor(
|
||||
name='downsample_mode', full_name='mediapipe.ToneEstimationOptions.downsample_mode', index=9,
|
||||
number=10, type=14, cpp_type=8, 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='downsampling_size', full_name='mediapipe.ToneEstimationOptions.downsampling_size', index=10,
|
||||
number=11, type=5, cpp_type=1, label=1,
|
||||
has_default_value=True, default_value=256,
|
||||
message_type=None, enum_type=None, containing_type=None,
|
||||
is_extension=False, extension_scope=None,
|
||||
serialized_options=None, file=DESCRIPTOR),
|
||||
_descriptor.FieldDescriptor(
|
||||
name='downsample_factor', full_name='mediapipe.ToneEstimationOptions.downsample_factor', index=11,
|
||||
number=12, type=2, cpp_type=6, label=1,
|
||||
has_default_value=True, default_value=float(2),
|
||||
message_type=None, enum_type=None, containing_type=None,
|
||||
is_extension=False, extension_scope=None,
|
||||
serialized_options=None, file=DESCRIPTOR),
|
||||
],
|
||||
extensions=[
|
||||
],
|
||||
nested_types=[_TONEESTIMATIONOPTIONS_GAINBIASBOUNDS, ],
|
||||
enum_types=[
|
||||
_TONEESTIMATIONOPTIONS_DOWNSAMPLEMODE,
|
||||
],
|
||||
serialized_options=None,
|
||||
is_extendable=False,
|
||||
syntax='proto2',
|
||||
extension_ranges=[],
|
||||
oneofs=[
|
||||
],
|
||||
serialized_start=1123,
|
||||
serialized_end=2020,
|
||||
)
|
||||
|
||||
|
||||
_TONEMATCH = _descriptor.Descriptor(
|
||||
name='ToneMatch',
|
||||
full_name='mediapipe.ToneMatch',
|
||||
filename=None,
|
||||
file=DESCRIPTOR,
|
||||
containing_type=None,
|
||||
fields=[
|
||||
_descriptor.FieldDescriptor(
|
||||
name='curr_val', full_name='mediapipe.ToneMatch.curr_val', 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='prev_val', full_name='mediapipe.ToneMatch.prev_val', 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=2022,
|
||||
serialized_end=2069,
|
||||
)
|
||||
|
||||
|
||||
_PATCHTONEMATCH = _descriptor.Descriptor(
|
||||
name='PatchToneMatch',
|
||||
full_name='mediapipe.PatchToneMatch',
|
||||
filename=None,
|
||||
file=DESCRIPTOR,
|
||||
containing_type=None,
|
||||
fields=[
|
||||
_descriptor.FieldDescriptor(
|
||||
name='tone_match', full_name='mediapipe.PatchToneMatch.tone_match', index=0,
|
||||
number=1, type=11, cpp_type=10, 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='irls_weight', full_name='mediapipe.PatchToneMatch.irls_weight', index=1,
|
||||
number=2, 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),
|
||||
],
|
||||
extensions=[
|
||||
],
|
||||
nested_types=[],
|
||||
enum_types=[
|
||||
],
|
||||
serialized_options=None,
|
||||
is_extendable=False,
|
||||
syntax='proto2',
|
||||
extension_ranges=[],
|
||||
oneofs=[
|
||||
],
|
||||
serialized_start=2071,
|
||||
serialized_end=2153,
|
||||
)
|
||||
|
||||
_TONECHANGE_STABILITYSTATS.containing_type = _TONECHANGE
|
||||
_TONECHANGE.fields_by_name['gain_bias'].message_type = mediapipe_dot_util_dot_tracking_dot_tone__models__pb2._GAINBIASMODEL
|
||||
_TONECHANGE.fields_by_name['affine'].message_type = mediapipe_dot_util_dot_tracking_dot_tone__models__pb2._AFFINETONEMODEL
|
||||
_TONECHANGE.fields_by_name['mixture_gain_bias'].message_type = mediapipe_dot_util_dot_tracking_dot_tone__models__pb2._MIXTUREGAINBIASMODEL
|
||||
_TONECHANGE.fields_by_name['mixture_affine'].message_type = mediapipe_dot_util_dot_tracking_dot_tone__models__pb2._MIXTUREAFFINETONEMODEL
|
||||
_TONECHANGE.fields_by_name['type'].enum_type = _TONECHANGE_TYPE
|
||||
_TONECHANGE.fields_by_name['stability_stats'].message_type = _TONECHANGE_STABILITYSTATS
|
||||
_TONECHANGE_TYPE.containing_type = _TONECHANGE
|
||||
_TONEESTIMATIONOPTIONS_GAINBIASBOUNDS.containing_type = _TONEESTIMATIONOPTIONS
|
||||
_TONEESTIMATIONOPTIONS.fields_by_name['tone_match_options'].message_type = _TONEMATCHOPTIONS
|
||||
_TONEESTIMATIONOPTIONS.fields_by_name['clip_mask_options'].message_type = _CLIPMASKOPTIONS
|
||||
_TONEESTIMATIONOPTIONS.fields_by_name['stable_gain_bias_bounds'].message_type = _TONEESTIMATIONOPTIONS_GAINBIASBOUNDS
|
||||
_TONEESTIMATIONOPTIONS.fields_by_name['downsample_mode'].enum_type = _TONEESTIMATIONOPTIONS_DOWNSAMPLEMODE
|
||||
_TONEESTIMATIONOPTIONS_DOWNSAMPLEMODE.containing_type = _TONEESTIMATIONOPTIONS
|
||||
_PATCHTONEMATCH.fields_by_name['tone_match'].message_type = _TONEMATCH
|
||||
DESCRIPTOR.message_types_by_name['ToneChange'] = _TONECHANGE
|
||||
DESCRIPTOR.message_types_by_name['ToneMatchOptions'] = _TONEMATCHOPTIONS
|
||||
DESCRIPTOR.message_types_by_name['ClipMaskOptions'] = _CLIPMASKOPTIONS
|
||||
DESCRIPTOR.message_types_by_name['ToneEstimationOptions'] = _TONEESTIMATIONOPTIONS
|
||||
DESCRIPTOR.message_types_by_name['ToneMatch'] = _TONEMATCH
|
||||
DESCRIPTOR.message_types_by_name['PatchToneMatch'] = _PATCHTONEMATCH
|
||||
_sym_db.RegisterFileDescriptor(DESCRIPTOR)
|
||||
|
||||
ToneChange = _reflection.GeneratedProtocolMessageType('ToneChange', (_message.Message,), dict(
|
||||
|
||||
StabilityStats = _reflection.GeneratedProtocolMessageType('StabilityStats', (_message.Message,), dict(
|
||||
DESCRIPTOR = _TONECHANGE_STABILITYSTATS,
|
||||
__module__ = 'mediapipe.util.tracking.tone_estimation_pb2'
|
||||
# @@protoc_insertion_point(class_scope:mediapipe.ToneChange.StabilityStats)
|
||||
))
|
||||
,
|
||||
DESCRIPTOR = _TONECHANGE,
|
||||
__module__ = 'mediapipe.util.tracking.tone_estimation_pb2'
|
||||
# @@protoc_insertion_point(class_scope:mediapipe.ToneChange)
|
||||
))
|
||||
_sym_db.RegisterMessage(ToneChange)
|
||||
_sym_db.RegisterMessage(ToneChange.StabilityStats)
|
||||
|
||||
ToneMatchOptions = _reflection.GeneratedProtocolMessageType('ToneMatchOptions', (_message.Message,), dict(
|
||||
DESCRIPTOR = _TONEMATCHOPTIONS,
|
||||
__module__ = 'mediapipe.util.tracking.tone_estimation_pb2'
|
||||
# @@protoc_insertion_point(class_scope:mediapipe.ToneMatchOptions)
|
||||
))
|
||||
_sym_db.RegisterMessage(ToneMatchOptions)
|
||||
|
||||
ClipMaskOptions = _reflection.GeneratedProtocolMessageType('ClipMaskOptions', (_message.Message,), dict(
|
||||
DESCRIPTOR = _CLIPMASKOPTIONS,
|
||||
__module__ = 'mediapipe.util.tracking.tone_estimation_pb2'
|
||||
# @@protoc_insertion_point(class_scope:mediapipe.ClipMaskOptions)
|
||||
))
|
||||
_sym_db.RegisterMessage(ClipMaskOptions)
|
||||
|
||||
ToneEstimationOptions = _reflection.GeneratedProtocolMessageType('ToneEstimationOptions', (_message.Message,), dict(
|
||||
|
||||
GainBiasBounds = _reflection.GeneratedProtocolMessageType('GainBiasBounds', (_message.Message,), dict(
|
||||
DESCRIPTOR = _TONEESTIMATIONOPTIONS_GAINBIASBOUNDS,
|
||||
__module__ = 'mediapipe.util.tracking.tone_estimation_pb2'
|
||||
# @@protoc_insertion_point(class_scope:mediapipe.ToneEstimationOptions.GainBiasBounds)
|
||||
))
|
||||
,
|
||||
DESCRIPTOR = _TONEESTIMATIONOPTIONS,
|
||||
__module__ = 'mediapipe.util.tracking.tone_estimation_pb2'
|
||||
# @@protoc_insertion_point(class_scope:mediapipe.ToneEstimationOptions)
|
||||
))
|
||||
_sym_db.RegisterMessage(ToneEstimationOptions)
|
||||
_sym_db.RegisterMessage(ToneEstimationOptions.GainBiasBounds)
|
||||
|
||||
ToneMatch = _reflection.GeneratedProtocolMessageType('ToneMatch', (_message.Message,), dict(
|
||||
DESCRIPTOR = _TONEMATCH,
|
||||
__module__ = 'mediapipe.util.tracking.tone_estimation_pb2'
|
||||
# @@protoc_insertion_point(class_scope:mediapipe.ToneMatch)
|
||||
))
|
||||
_sym_db.RegisterMessage(ToneMatch)
|
||||
|
||||
PatchToneMatch = _reflection.GeneratedProtocolMessageType('PatchToneMatch', (_message.Message,), dict(
|
||||
DESCRIPTOR = _PATCHTONEMATCH,
|
||||
__module__ = 'mediapipe.util.tracking.tone_estimation_pb2'
|
||||
# @@protoc_insertion_point(class_scope:mediapipe.PatchToneMatch)
|
||||
))
|
||||
_sym_db.RegisterMessage(PatchToneMatch)
|
||||
|
||||
|
||||
# @@protoc_insertion_point(module_scope)
|
||||
@@ -0,0 +1,301 @@
|
||||
# -*- coding: utf-8 -*-
|
||||
# Generated by the protocol buffer compiler. DO NOT EDIT!
|
||||
# source: mediapipe/util/tracking/tone_models.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()
|
||||
|
||||
|
||||
|
||||
|
||||
DESCRIPTOR = _descriptor.FileDescriptor(
|
||||
name='mediapipe/util/tracking/tone_models.proto',
|
||||
package='mediapipe',
|
||||
syntax='proto2',
|
||||
serialized_options=None,
|
||||
serialized_pb=_b('\n)mediapipe/util/tracking/tone_models.proto\x12\tmediapipe\"\x87\x01\n\rGainBiasModel\x12\x12\n\x07gain_c1\x18\x01 \x01(\x02:\x01\x31\x12\x12\n\x07\x62ias_c1\x18\x02 \x01(\x02:\x01\x30\x12\x12\n\x07gain_c2\x18\x03 \x01(\x02:\x01\x31\x12\x12\n\x07\x62ias_c2\x18\x04 \x01(\x02:\x01\x30\x12\x12\n\x07gain_c3\x18\x05 \x01(\x02:\x01\x31\x12\x12\n\x07\x62ias_c3\x18\x06 \x01(\x02:\x01\x30\"?\n\x14MixtureGainBiasModel\x12\'\n\x05model\x18\x01 \x03(\x0b\x32\x18.mediapipe.GainBiasModel\"\xdd\x01\n\x0f\x41\x66\x66ineToneModel\x12\x0f\n\x04g_00\x18\x01 \x01(\x02:\x01\x31\x12\x0f\n\x04g_01\x18\x02 \x01(\x02:\x01\x30\x12\x0f\n\x04g_02\x18\x03 \x01(\x02:\x01\x30\x12\x0f\n\x04g_03\x18\x04 \x01(\x02:\x01\x30\x12\x0f\n\x04g_10\x18\x05 \x01(\x02:\x01\x30\x12\x0f\n\x04g_11\x18\x06 \x01(\x02:\x01\x31\x12\x0f\n\x04g_12\x18\x07 \x01(\x02:\x01\x30\x12\x0f\n\x04g_13\x18\x08 \x01(\x02:\x01\x30\x12\x0f\n\x04g_20\x18\t \x01(\x02:\x01\x30\x12\x0f\n\x04g_21\x18\n \x01(\x02:\x01\x30\x12\x0f\n\x04g_22\x18\x0b \x01(\x02:\x01\x31\x12\x0f\n\x04g_23\x18\x0c \x01(\x02:\x01\x30\"C\n\x16MixtureAffineToneModel\x12)\n\x05model\x18\x01 \x03(\x0b\x32\x1a.mediapipe.AffineToneModel')
|
||||
)
|
||||
|
||||
|
||||
|
||||
|
||||
_GAINBIASMODEL = _descriptor.Descriptor(
|
||||
name='GainBiasModel',
|
||||
full_name='mediapipe.GainBiasModel',
|
||||
filename=None,
|
||||
file=DESCRIPTOR,
|
||||
containing_type=None,
|
||||
fields=[
|
||||
_descriptor.FieldDescriptor(
|
||||
name='gain_c1', full_name='mediapipe.GainBiasModel.gain_c1', index=0,
|
||||
number=1, 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='bias_c1', full_name='mediapipe.GainBiasModel.bias_c1', index=1,
|
||||
number=2, 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='gain_c2', full_name='mediapipe.GainBiasModel.gain_c2', index=2,
|
||||
number=3, 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='bias_c2', full_name='mediapipe.GainBiasModel.bias_c2', index=3,
|
||||
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='gain_c3', full_name='mediapipe.GainBiasModel.gain_c3', index=4,
|
||||
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='bias_c3', full_name='mediapipe.GainBiasModel.bias_c3', index=5,
|
||||
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),
|
||||
],
|
||||
extensions=[
|
||||
],
|
||||
nested_types=[],
|
||||
enum_types=[
|
||||
],
|
||||
serialized_options=None,
|
||||
is_extendable=False,
|
||||
syntax='proto2',
|
||||
extension_ranges=[],
|
||||
oneofs=[
|
||||
],
|
||||
serialized_start=57,
|
||||
serialized_end=192,
|
||||
)
|
||||
|
||||
|
||||
_MIXTUREGAINBIASMODEL = _descriptor.Descriptor(
|
||||
name='MixtureGainBiasModel',
|
||||
full_name='mediapipe.MixtureGainBiasModel',
|
||||
filename=None,
|
||||
file=DESCRIPTOR,
|
||||
containing_type=None,
|
||||
fields=[
|
||||
_descriptor.FieldDescriptor(
|
||||
name='model', full_name='mediapipe.MixtureGainBiasModel.model', index=0,
|
||||
number=1, type=11, cpp_type=10, 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),
|
||||
],
|
||||
extensions=[
|
||||
],
|
||||
nested_types=[],
|
||||
enum_types=[
|
||||
],
|
||||
serialized_options=None,
|
||||
is_extendable=False,
|
||||
syntax='proto2',
|
||||
extension_ranges=[],
|
||||
oneofs=[
|
||||
],
|
||||
serialized_start=194,
|
||||
serialized_end=257,
|
||||
)
|
||||
|
||||
|
||||
_AFFINETONEMODEL = _descriptor.Descriptor(
|
||||
name='AffineToneModel',
|
||||
full_name='mediapipe.AffineToneModel',
|
||||
filename=None,
|
||||
file=DESCRIPTOR,
|
||||
containing_type=None,
|
||||
fields=[
|
||||
_descriptor.FieldDescriptor(
|
||||
name='g_00', full_name='mediapipe.AffineToneModel.g_00', index=0,
|
||||
number=1, 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='g_01', full_name='mediapipe.AffineToneModel.g_01', index=1,
|
||||
number=2, 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='g_02', full_name='mediapipe.AffineToneModel.g_02', index=2,
|
||||
number=3, 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='g_03', full_name='mediapipe.AffineToneModel.g_03', index=3,
|
||||
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='g_10', full_name='mediapipe.AffineToneModel.g_10', index=4,
|
||||
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='g_11', full_name='mediapipe.AffineToneModel.g_11', index=5,
|
||||
number=6, 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='g_12', full_name='mediapipe.AffineToneModel.g_12', index=6,
|
||||
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='g_13', full_name='mediapipe.AffineToneModel.g_13', index=7,
|
||||
number=8, 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='g_20', full_name='mediapipe.AffineToneModel.g_20', index=8,
|
||||
number=9, 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='g_21', full_name='mediapipe.AffineToneModel.g_21', index=9,
|
||||
number=10, 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='g_22', full_name='mediapipe.AffineToneModel.g_22', index=10,
|
||||
number=11, 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='g_23', full_name='mediapipe.AffineToneModel.g_23', index=11,
|
||||
number=12, 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),
|
||||
],
|
||||
extensions=[
|
||||
],
|
||||
nested_types=[],
|
||||
enum_types=[
|
||||
],
|
||||
serialized_options=None,
|
||||
is_extendable=False,
|
||||
syntax='proto2',
|
||||
extension_ranges=[],
|
||||
oneofs=[
|
||||
],
|
||||
serialized_start=260,
|
||||
serialized_end=481,
|
||||
)
|
||||
|
||||
|
||||
_MIXTUREAFFINETONEMODEL = _descriptor.Descriptor(
|
||||
name='MixtureAffineToneModel',
|
||||
full_name='mediapipe.MixtureAffineToneModel',
|
||||
filename=None,
|
||||
file=DESCRIPTOR,
|
||||
containing_type=None,
|
||||
fields=[
|
||||
_descriptor.FieldDescriptor(
|
||||
name='model', full_name='mediapipe.MixtureAffineToneModel.model', index=0,
|
||||
number=1, type=11, cpp_type=10, 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),
|
||||
],
|
||||
extensions=[
|
||||
],
|
||||
nested_types=[],
|
||||
enum_types=[
|
||||
],
|
||||
serialized_options=None,
|
||||
is_extendable=False,
|
||||
syntax='proto2',
|
||||
extension_ranges=[],
|
||||
oneofs=[
|
||||
],
|
||||
serialized_start=483,
|
||||
serialized_end=550,
|
||||
)
|
||||
|
||||
_MIXTUREGAINBIASMODEL.fields_by_name['model'].message_type = _GAINBIASMODEL
|
||||
_MIXTUREAFFINETONEMODEL.fields_by_name['model'].message_type = _AFFINETONEMODEL
|
||||
DESCRIPTOR.message_types_by_name['GainBiasModel'] = _GAINBIASMODEL
|
||||
DESCRIPTOR.message_types_by_name['MixtureGainBiasModel'] = _MIXTUREGAINBIASMODEL
|
||||
DESCRIPTOR.message_types_by_name['AffineToneModel'] = _AFFINETONEMODEL
|
||||
DESCRIPTOR.message_types_by_name['MixtureAffineToneModel'] = _MIXTUREAFFINETONEMODEL
|
||||
_sym_db.RegisterFileDescriptor(DESCRIPTOR)
|
||||
|
||||
GainBiasModel = _reflection.GeneratedProtocolMessageType('GainBiasModel', (_message.Message,), dict(
|
||||
DESCRIPTOR = _GAINBIASMODEL,
|
||||
__module__ = 'mediapipe.util.tracking.tone_models_pb2'
|
||||
# @@protoc_insertion_point(class_scope:mediapipe.GainBiasModel)
|
||||
))
|
||||
_sym_db.RegisterMessage(GainBiasModel)
|
||||
|
||||
MixtureGainBiasModel = _reflection.GeneratedProtocolMessageType('MixtureGainBiasModel', (_message.Message,), dict(
|
||||
DESCRIPTOR = _MIXTUREGAINBIASMODEL,
|
||||
__module__ = 'mediapipe.util.tracking.tone_models_pb2'
|
||||
# @@protoc_insertion_point(class_scope:mediapipe.MixtureGainBiasModel)
|
||||
))
|
||||
_sym_db.RegisterMessage(MixtureGainBiasModel)
|
||||
|
||||
AffineToneModel = _reflection.GeneratedProtocolMessageType('AffineToneModel', (_message.Message,), dict(
|
||||
DESCRIPTOR = _AFFINETONEMODEL,
|
||||
__module__ = 'mediapipe.util.tracking.tone_models_pb2'
|
||||
# @@protoc_insertion_point(class_scope:mediapipe.AffineToneModel)
|
||||
))
|
||||
_sym_db.RegisterMessage(AffineToneModel)
|
||||
|
||||
MixtureAffineToneModel = _reflection.GeneratedProtocolMessageType('MixtureAffineToneModel', (_message.Message,), dict(
|
||||
DESCRIPTOR = _MIXTUREAFFINETONEMODEL,
|
||||
__module__ = 'mediapipe.util.tracking.tone_models_pb2'
|
||||
# @@protoc_insertion_point(class_scope:mediapipe.MixtureAffineToneModel)
|
||||
))
|
||||
_sym_db.RegisterMessage(MixtureAffineToneModel)
|
||||
|
||||
|
||||
# @@protoc_insertion_point(module_scope)
|
||||
@@ -0,0 +1,77 @@
|
||||
# -*- coding: utf-8 -*-
|
||||
# Generated by the protocol buffer compiler. DO NOT EDIT!
|
||||
# source: mediapipe/util/tracking/tracked_detection_manager_config.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()
|
||||
|
||||
|
||||
|
||||
|
||||
DESCRIPTOR = _descriptor.FileDescriptor(
|
||||
name='mediapipe/util/tracking/tracked_detection_manager_config.proto',
|
||||
package='mediapipe',
|
||||
syntax='proto2',
|
||||
serialized_options=None,
|
||||
serialized_pb=_b('\n>mediapipe/util/tracking/tracked_detection_manager_config.proto\x12\tmediapipe\"~\n\x1dTrackedDetectionManagerConfig\x12+\n is_same_detection_max_area_ratio\x18\x01 \x01(\x02:\x01\x33\x12\x30\n#is_same_detection_min_overlap_ratio\x18\x02 \x01(\x02:\x03\x30.5')
|
||||
)
|
||||
|
||||
|
||||
|
||||
|
||||
_TRACKEDDETECTIONMANAGERCONFIG = _descriptor.Descriptor(
|
||||
name='TrackedDetectionManagerConfig',
|
||||
full_name='mediapipe.TrackedDetectionManagerConfig',
|
||||
filename=None,
|
||||
file=DESCRIPTOR,
|
||||
containing_type=None,
|
||||
fields=[
|
||||
_descriptor.FieldDescriptor(
|
||||
name='is_same_detection_max_area_ratio', full_name='mediapipe.TrackedDetectionManagerConfig.is_same_detection_max_area_ratio', index=0,
|
||||
number=1, type=2, cpp_type=6, label=1,
|
||||
has_default_value=True, default_value=float(3),
|
||||
message_type=None, enum_type=None, containing_type=None,
|
||||
is_extension=False, extension_scope=None,
|
||||
serialized_options=None, file=DESCRIPTOR),
|
||||
_descriptor.FieldDescriptor(
|
||||
name='is_same_detection_min_overlap_ratio', full_name='mediapipe.TrackedDetectionManagerConfig.is_same_detection_min_overlap_ratio', index=1,
|
||||
number=2, type=2, cpp_type=6, label=1,
|
||||
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),
|
||||
],
|
||||
extensions=[
|
||||
],
|
||||
nested_types=[],
|
||||
enum_types=[
|
||||
],
|
||||
serialized_options=None,
|
||||
is_extendable=False,
|
||||
syntax='proto2',
|
||||
extension_ranges=[],
|
||||
oneofs=[
|
||||
],
|
||||
serialized_start=77,
|
||||
serialized_end=203,
|
||||
)
|
||||
|
||||
DESCRIPTOR.message_types_by_name['TrackedDetectionManagerConfig'] = _TRACKEDDETECTIONMANAGERCONFIG
|
||||
_sym_db.RegisterFileDescriptor(DESCRIPTOR)
|
||||
|
||||
TrackedDetectionManagerConfig = _reflection.GeneratedProtocolMessageType('TrackedDetectionManagerConfig', (_message.Message,), dict(
|
||||
DESCRIPTOR = _TRACKEDDETECTIONMANAGERCONFIG,
|
||||
__module__ = 'mediapipe.util.tracking.tracked_detection_manager_config_pb2'
|
||||
# @@protoc_insertion_point(class_scope:mediapipe.TrackedDetectionManagerConfig)
|
||||
))
|
||||
_sym_db.RegisterMessage(TrackedDetectionManagerConfig)
|
||||
|
||||
|
||||
# @@protoc_insertion_point(module_scope)
|
||||
File diff suppressed because one or more lines are too long
Reference in New Issue
Block a user