mediapipe-nightly 0.10.21.post20250114__cp312-cp312-manylinux_2_28_x86_64.whl
Sign up to get free protection for your applications and to get access to all the features.
- mediapipe/__init__.py +26 -0
- mediapipe/calculators/__init__.py +0 -0
- mediapipe/calculators/audio/__init__.py +0 -0
- mediapipe/calculators/audio/mfcc_mel_calculators_pb2.py +33 -0
- mediapipe/calculators/audio/rational_factor_resample_calculator_pb2.py +33 -0
- mediapipe/calculators/audio/spectrogram_calculator_pb2.py +37 -0
- mediapipe/calculators/audio/stabilized_log_calculator_pb2.py +31 -0
- mediapipe/calculators/audio/time_series_framer_calculator_pb2.py +33 -0
- mediapipe/calculators/core/__init__.py +0 -0
- mediapipe/calculators/core/bypass_calculator_pb2.py +31 -0
- mediapipe/calculators/core/clip_vector_size_calculator_pb2.py +31 -0
- mediapipe/calculators/core/concatenate_vector_calculator_pb2.py +31 -0
- mediapipe/calculators/core/constant_side_packet_calculator_pb2.py +39 -0
- mediapipe/calculators/core/dequantize_byte_array_calculator_pb2.py +31 -0
- mediapipe/calculators/core/flow_limiter_calculator_pb2.py +32 -0
- mediapipe/calculators/core/gate_calculator_pb2.py +33 -0
- mediapipe/calculators/core/get_vector_item_calculator_pb2.py +31 -0
- mediapipe/calculators/core/graph_profile_calculator_pb2.py +31 -0
- mediapipe/calculators/core/packet_cloner_calculator_pb2.py +31 -0
- mediapipe/calculators/core/packet_resampler_calculator_pb2.py +33 -0
- mediapipe/calculators/core/packet_thinner_calculator_pb2.py +33 -0
- mediapipe/calculators/core/quantize_float_vector_calculator_pb2.py +31 -0
- mediapipe/calculators/core/sequence_shift_calculator_pb2.py +31 -0
- mediapipe/calculators/core/split_vector_calculator_pb2.py +33 -0
- mediapipe/calculators/image/__init__.py +0 -0
- mediapipe/calculators/image/bilateral_filter_calculator_pb2.py +31 -0
- mediapipe/calculators/image/feature_detector_calculator_pb2.py +31 -0
- mediapipe/calculators/image/image_clone_calculator_pb2.py +31 -0
- mediapipe/calculators/image/image_cropping_calculator_pb2.py +33 -0
- mediapipe/calculators/image/image_transformation_calculator_pb2.py +38 -0
- mediapipe/calculators/image/mask_overlay_calculator_pb2.py +33 -0
- mediapipe/calculators/image/opencv_encoded_image_to_image_frame_calculator_pb2.py +31 -0
- mediapipe/calculators/image/opencv_image_encoder_calculator_pb2.py +35 -0
- mediapipe/calculators/image/recolor_calculator_pb2.py +34 -0
- mediapipe/calculators/image/rotation_mode_pb2.py +29 -0
- mediapipe/calculators/image/scale_image_calculator_pb2.py +34 -0
- mediapipe/calculators/image/segmentation_smoothing_calculator_pb2.py +31 -0
- mediapipe/calculators/image/set_alpha_calculator_pb2.py +31 -0
- mediapipe/calculators/image/warp_affine_calculator_pb2.py +36 -0
- mediapipe/calculators/internal/__init__.py +0 -0
- mediapipe/calculators/internal/callback_packet_calculator_pb2.py +33 -0
- mediapipe/calculators/tensor/__init__.py +0 -0
- mediapipe/calculators/tensor/audio_to_tensor_calculator_pb2.py +35 -0
- mediapipe/calculators/tensor/bert_preprocessor_calculator_pb2.py +31 -0
- mediapipe/calculators/tensor/feedback_tensors_calculator_pb2.py +37 -0
- mediapipe/calculators/tensor/image_to_tensor_calculator_pb2.py +40 -0
- mediapipe/calculators/tensor/inference_calculator_pb2.py +63 -0
- mediapipe/calculators/tensor/landmarks_to_tensor_calculator_pb2.py +33 -0
- mediapipe/calculators/tensor/regex_preprocessor_calculator_pb2.py +31 -0
- mediapipe/calculators/tensor/tensor_converter_calculator_pb2.py +34 -0
- mediapipe/calculators/tensor/tensor_to_joints_calculator_pb2.py +31 -0
- mediapipe/calculators/tensor/tensors_readback_calculator_pb2.py +35 -0
- mediapipe/calculators/tensor/tensors_to_audio_calculator_pb2.py +33 -0
- mediapipe/calculators/tensor/tensors_to_classification_calculator_pb2.py +44 -0
- mediapipe/calculators/tensor/tensors_to_detections_calculator_pb2.py +39 -0
- mediapipe/calculators/tensor/tensors_to_floats_calculator_pb2.py +33 -0
- mediapipe/calculators/tensor/tensors_to_landmarks_calculator_pb2.py +33 -0
- mediapipe/calculators/tensor/tensors_to_segmentation_calculator_pb2.py +34 -0
- mediapipe/calculators/tensor/vector_to_tensor_calculator_pb2.py +27 -0
- mediapipe/calculators/tflite/__init__.py +0 -0
- mediapipe/calculators/tflite/ssd_anchors_calculator_pb2.py +32 -0
- mediapipe/calculators/tflite/tflite_converter_calculator_pb2.py +33 -0
- mediapipe/calculators/tflite/tflite_custom_op_resolver_calculator_pb2.py +31 -0
- mediapipe/calculators/tflite/tflite_inference_calculator_pb2.py +49 -0
- mediapipe/calculators/tflite/tflite_tensors_to_classification_calculator_pb2.py +31 -0
- mediapipe/calculators/tflite/tflite_tensors_to_detections_calculator_pb2.py +31 -0
- mediapipe/calculators/tflite/tflite_tensors_to_landmarks_calculator_pb2.py +33 -0
- mediapipe/calculators/tflite/tflite_tensors_to_segmentation_calculator_pb2.py +31 -0
- mediapipe/calculators/util/__init__.py +0 -0
- mediapipe/calculators/util/align_hand_to_pose_in_world_calculator_pb2.py +31 -0
- mediapipe/calculators/util/annotation_overlay_calculator_pb2.py +32 -0
- mediapipe/calculators/util/association_calculator_pb2.py +31 -0
- mediapipe/calculators/util/collection_has_min_size_calculator_pb2.py +31 -0
- mediapipe/calculators/util/combine_joints_calculator_pb2.py +36 -0
- mediapipe/calculators/util/detection_label_id_to_text_calculator_pb2.py +36 -0
- mediapipe/calculators/util/detections_to_rects_calculator_pb2.py +33 -0
- mediapipe/calculators/util/detections_to_render_data_calculator_pb2.py +33 -0
- mediapipe/calculators/util/face_to_rect_calculator_pb2.py +26 -0
- mediapipe/calculators/util/filter_detections_calculator_pb2.py +31 -0
- mediapipe/calculators/util/flat_color_image_calculator_pb2.py +32 -0
- mediapipe/calculators/util/labels_to_render_data_calculator_pb2.py +34 -0
- mediapipe/calculators/util/landmark_projection_calculator_pb2.py +31 -0
- mediapipe/calculators/util/landmarks_refinement_calculator_pb2.py +41 -0
- mediapipe/calculators/util/landmarks_smoothing_calculator_pb2.py +33 -0
- mediapipe/calculators/util/landmarks_to_detection_calculator_pb2.py +31 -0
- mediapipe/calculators/util/landmarks_to_floats_calculator_pb2.py +31 -0
- mediapipe/calculators/util/landmarks_to_render_data_calculator_pb2.py +32 -0
- mediapipe/calculators/util/landmarks_transformation_calculator_pb2.py +37 -0
- mediapipe/calculators/util/latency_pb2.py +26 -0
- mediapipe/calculators/util/local_file_contents_calculator_pb2.py +31 -0
- mediapipe/calculators/util/logic_calculator_pb2.py +34 -0
- mediapipe/calculators/util/non_max_suppression_calculator_pb2.py +35 -0
- mediapipe/calculators/util/packet_frequency_calculator_pb2.py +31 -0
- mediapipe/calculators/util/packet_frequency_pb2.py +26 -0
- mediapipe/calculators/util/packet_latency_calculator_pb2.py +31 -0
- mediapipe/calculators/util/rect_to_render_data_calculator_pb2.py +32 -0
- mediapipe/calculators/util/rect_to_render_scale_calculator_pb2.py +31 -0
- mediapipe/calculators/util/rect_transformation_calculator_pb2.py +31 -0
- mediapipe/calculators/util/refine_landmarks_from_heatmap_calculator_pb2.py +31 -0
- mediapipe/calculators/util/resource_provider_calculator_pb2.py +28 -0
- mediapipe/calculators/util/set_joints_visibility_calculator_pb2.py +41 -0
- mediapipe/calculators/util/thresholding_calculator_pb2.py +31 -0
- mediapipe/calculators/util/timed_box_list_id_to_label_calculator_pb2.py +31 -0
- mediapipe/calculators/util/timed_box_list_to_render_data_calculator_pb2.py +32 -0
- mediapipe/calculators/util/top_k_scores_calculator_pb2.py +31 -0
- mediapipe/calculators/util/visibility_copy_calculator_pb2.py +27 -0
- mediapipe/calculators/util/visibility_smoothing_calculator_pb2.py +31 -0
- mediapipe/calculators/video/__init__.py +0 -0
- mediapipe/calculators/video/box_detector_calculator_pb2.py +32 -0
- mediapipe/calculators/video/box_tracker_calculator_pb2.py +32 -0
- mediapipe/calculators/video/flow_packager_calculator_pb2.py +32 -0
- mediapipe/calculators/video/flow_to_image_calculator_pb2.py +31 -0
- mediapipe/calculators/video/motion_analysis_calculator_pb2.py +42 -0
- mediapipe/calculators/video/opencv_video_encoder_calculator_pb2.py +31 -0
- mediapipe/calculators/video/tool/__init__.py +0 -0
- mediapipe/calculators/video/tool/flow_quantizer_model_pb2.py +26 -0
- mediapipe/calculators/video/tracked_detection_manager_calculator_pb2.py +32 -0
- mediapipe/calculators/video/video_pre_stream_calculator_pb2.py +35 -0
- mediapipe/examples/__init__.py +14 -0
- mediapipe/examples/desktop/__init__.py +14 -0
- mediapipe/framework/__init__.py +0 -0
- mediapipe/framework/calculator_options_pb2.py +29 -0
- mediapipe/framework/calculator_pb2.py +59 -0
- mediapipe/framework/calculator_profile_pb2.py +48 -0
- mediapipe/framework/deps/__init__.py +0 -0
- mediapipe/framework/deps/proto_descriptor_pb2.py +29 -0
- mediapipe/framework/formats/__init__.py +0 -0
- mediapipe/framework/formats/affine_transform_data_pb2.py +28 -0
- mediapipe/framework/formats/annotation/__init__.py +0 -0
- mediapipe/framework/formats/annotation/locus_pb2.py +32 -0
- mediapipe/framework/formats/annotation/rasterization_pb2.py +29 -0
- mediapipe/framework/formats/body_rig_pb2.py +28 -0
- mediapipe/framework/formats/classification_pb2.py +31 -0
- mediapipe/framework/formats/detection_pb2.py +36 -0
- mediapipe/framework/formats/image_file_properties_pb2.py +26 -0
- mediapipe/framework/formats/image_format_pb2.py +29 -0
- mediapipe/framework/formats/landmark_pb2.py +37 -0
- mediapipe/framework/formats/location_data_pb2.py +38 -0
- mediapipe/framework/formats/matrix_data_pb2.py +31 -0
- mediapipe/framework/formats/motion/__init__.py +0 -0
- mediapipe/framework/formats/motion/optical_flow_field_data_pb2.py +30 -0
- mediapipe/framework/formats/object_detection/__init__.py +0 -0
- mediapipe/framework/formats/object_detection/anchor_pb2.py +26 -0
- mediapipe/framework/formats/rect_pb2.py +29 -0
- mediapipe/framework/formats/time_series_header_pb2.py +28 -0
- mediapipe/framework/graph_runtime_info_pb2.py +31 -0
- mediapipe/framework/mediapipe_options_pb2.py +27 -0
- mediapipe/framework/packet_factory_pb2.py +31 -0
- mediapipe/framework/packet_generator_pb2.py +33 -0
- mediapipe/framework/status_handler_pb2.py +28 -0
- mediapipe/framework/stream_handler/__init__.py +0 -0
- mediapipe/framework/stream_handler/default_input_stream_handler_pb2.py +27 -0
- mediapipe/framework/stream_handler/fixed_size_input_stream_handler_pb2.py +27 -0
- mediapipe/framework/stream_handler/sync_set_input_stream_handler_pb2.py +29 -0
- mediapipe/framework/stream_handler/timestamp_align_input_stream_handler_pb2.py +27 -0
- mediapipe/framework/stream_handler_pb2.py +30 -0
- mediapipe/framework/test_calculators_pb2.py +31 -0
- mediapipe/framework/thread_pool_executor_pb2.py +29 -0
- mediapipe/framework/tool/__init__.py +0 -0
- mediapipe/framework/tool/calculator_graph_template_pb2.py +44 -0
- mediapipe/framework/tool/field_data_pb2.py +28 -0
- mediapipe/framework/tool/node_chain_subgraph_pb2.py +31 -0
- mediapipe/framework/tool/packet_generator_wrapper_calculator_pb2.py +28 -0
- mediapipe/framework/tool/source_pb2.py +33 -0
- mediapipe/framework/tool/switch_container_pb2.py +32 -0
- mediapipe/gpu/__init__.py +0 -0
- mediapipe/gpu/copy_calculator_pb2.py +33 -0
- mediapipe/gpu/gl_animation_overlay_calculator_pb2.py +31 -0
- mediapipe/gpu/gl_context_options_pb2.py +31 -0
- mediapipe/gpu/gl_scaler_calculator_pb2.py +32 -0
- mediapipe/gpu/gl_surface_sink_calculator_pb2.py +32 -0
- mediapipe/gpu/gpu_origin_pb2.py +29 -0
- mediapipe/gpu/scale_mode_pb2.py +28 -0
- mediapipe/model_maker/__init__.py +27 -0
- mediapipe/model_maker/setup.py +107 -0
- mediapipe/modules/__init__.py +0 -0
- mediapipe/modules/face_detection/__init__.py +0 -0
- mediapipe/modules/face_detection/face_detection_full_range_cpu.binarypb +0 -0
- mediapipe/modules/face_detection/face_detection_full_range_sparse.tflite +0 -0
- mediapipe/modules/face_detection/face_detection_pb2.py +30 -0
- mediapipe/modules/face_detection/face_detection_short_range.tflite +0 -0
- mediapipe/modules/face_detection/face_detection_short_range_cpu.binarypb +0 -0
- mediapipe/modules/face_geometry/__init__.py +0 -0
- mediapipe/modules/face_geometry/data/__init__.py +0 -0
- mediapipe/modules/face_geometry/effect_renderer_calculator_pb2.py +27 -0
- mediapipe/modules/face_geometry/env_generator_calculator_pb2.py +28 -0
- mediapipe/modules/face_geometry/geometry_pipeline_calculator_pb2.py +27 -0
- mediapipe/modules/face_geometry/libs/__init__.py +0 -0
- mediapipe/modules/face_geometry/protos/__init__.py +0 -0
- mediapipe/modules/face_geometry/protos/environment_pb2.py +31 -0
- mediapipe/modules/face_geometry/protos/face_geometry_pb2.py +29 -0
- mediapipe/modules/face_geometry/protos/geometry_pipeline_metadata_pb2.py +32 -0
- mediapipe/modules/face_geometry/protos/mesh_3d_pb2.py +31 -0
- mediapipe/modules/face_landmark/__init__.py +0 -0
- mediapipe/modules/face_landmark/face_landmark.tflite +0 -0
- mediapipe/modules/face_landmark/face_landmark_front_cpu.binarypb +0 -0
- mediapipe/modules/face_landmark/face_landmark_with_attention.tflite +0 -0
- mediapipe/modules/hand_landmark/__init__.py +0 -0
- mediapipe/modules/hand_landmark/calculators/__init__.py +0 -0
- mediapipe/modules/hand_landmark/hand_landmark_full.tflite +0 -0
- mediapipe/modules/hand_landmark/hand_landmark_lite.tflite +0 -0
- mediapipe/modules/hand_landmark/hand_landmark_tracking_cpu.binarypb +0 -0
- mediapipe/modules/hand_landmark/handedness.txt +2 -0
- mediapipe/modules/holistic_landmark/__init__.py +0 -0
- mediapipe/modules/holistic_landmark/calculators/__init__.py +0 -0
- mediapipe/modules/holistic_landmark/calculators/roi_tracking_calculator_pb2.py +37 -0
- mediapipe/modules/holistic_landmark/hand_recrop.tflite +0 -0
- mediapipe/modules/holistic_landmark/holistic_landmark_cpu.binarypb +0 -0
- mediapipe/modules/iris_landmark/__init__.py +0 -0
- mediapipe/modules/iris_landmark/iris_landmark.tflite +0 -0
- mediapipe/modules/objectron/__init__.py +0 -0
- mediapipe/modules/objectron/calculators/__init__.py +0 -0
- mediapipe/modules/objectron/calculators/a_r_capture_metadata_pb2.py +102 -0
- mediapipe/modules/objectron/calculators/annotation_data_pb2.py +38 -0
- mediapipe/modules/objectron/calculators/belief_decoder_config_pb2.py +28 -0
- mediapipe/modules/objectron/calculators/camera_parameters_pb2.py +30 -0
- mediapipe/modules/objectron/calculators/filter_detection_calculator_pb2.py +35 -0
- mediapipe/modules/objectron/calculators/frame_annotation_to_rect_calculator_pb2.py +31 -0
- mediapipe/modules/objectron/calculators/frame_annotation_tracker_calculator_pb2.py +31 -0
- mediapipe/modules/objectron/calculators/lift_2d_frame_annotation_to_3d_calculator_pb2.py +32 -0
- mediapipe/modules/objectron/calculators/object_pb2.py +38 -0
- mediapipe/modules/objectron/calculators/tensors_to_objects_calculator_pb2.py +32 -0
- mediapipe/modules/objectron/calculators/tflite_tensors_to_objects_calculator_pb2.py +32 -0
- mediapipe/modules/objectron/object_detection_oidv4_labelmap.txt +24 -0
- mediapipe/modules/objectron/objectron_cpu.binarypb +0 -0
- mediapipe/modules/palm_detection/__init__.py +0 -0
- mediapipe/modules/palm_detection/palm_detection_full.tflite +0 -0
- mediapipe/modules/palm_detection/palm_detection_lite.tflite +0 -0
- mediapipe/modules/pose_detection/__init__.py +0 -0
- mediapipe/modules/pose_detection/pose_detection.tflite +0 -0
- mediapipe/modules/pose_landmark/__init__.py +0 -0
- mediapipe/modules/pose_landmark/pose_landmark_cpu.binarypb +0 -0
- mediapipe/modules/pose_landmark/pose_landmark_full.tflite +0 -0
- mediapipe/modules/selfie_segmentation/__init__.py +0 -0
- mediapipe/modules/selfie_segmentation/selfie_segmentation.tflite +0 -0
- mediapipe/modules/selfie_segmentation/selfie_segmentation_cpu.binarypb +0 -0
- mediapipe/modules/selfie_segmentation/selfie_segmentation_landscape.tflite +0 -0
- mediapipe/python/__init__.py +29 -0
- mediapipe/python/_framework_bindings.cpython-312-x86_64-linux-gnu.so +0 -0
- mediapipe/python/calculator_graph_test.py +251 -0
- mediapipe/python/image_frame_test.py +194 -0
- mediapipe/python/image_test.py +218 -0
- mediapipe/python/packet_creator.py +275 -0
- mediapipe/python/packet_getter.py +120 -0
- mediapipe/python/packet_test.py +533 -0
- mediapipe/python/solution_base.py +604 -0
- mediapipe/python/solution_base_test.py +396 -0
- mediapipe/python/solutions/__init__.py +27 -0
- mediapipe/python/solutions/download_utils.py +37 -0
- mediapipe/python/solutions/drawing_styles.py +249 -0
- mediapipe/python/solutions/drawing_utils.py +320 -0
- mediapipe/python/solutions/drawing_utils_test.py +258 -0
- mediapipe/python/solutions/face_detection.py +105 -0
- mediapipe/python/solutions/face_detection_test.py +92 -0
- mediapipe/python/solutions/face_mesh.py +125 -0
- mediapipe/python/solutions/face_mesh_connections.py +500 -0
- mediapipe/python/solutions/face_mesh_test.py +170 -0
- mediapipe/python/solutions/hands.py +153 -0
- mediapipe/python/solutions/hands_connections.py +32 -0
- mediapipe/python/solutions/hands_test.py +219 -0
- mediapipe/python/solutions/holistic.py +167 -0
- mediapipe/python/solutions/holistic_test.py +142 -0
- mediapipe/python/solutions/objectron.py +288 -0
- mediapipe/python/solutions/objectron_test.py +81 -0
- mediapipe/python/solutions/pose.py +192 -0
- mediapipe/python/solutions/pose_connections.py +22 -0
- mediapipe/python/solutions/pose_test.py +262 -0
- mediapipe/python/solutions/selfie_segmentation.py +76 -0
- mediapipe/python/solutions/selfie_segmentation_test.py +68 -0
- mediapipe/python/timestamp_test.py +78 -0
- mediapipe/tasks/__init__.py +14 -0
- mediapipe/tasks/cc/__init__.py +0 -0
- mediapipe/tasks/cc/audio/__init__.py +0 -0
- mediapipe/tasks/cc/audio/audio_classifier/__init__.py +0 -0
- mediapipe/tasks/cc/audio/audio_classifier/proto/__init__.py +0 -0
- mediapipe/tasks/cc/audio/audio_classifier/proto/audio_classifier_graph_options_pb2.py +35 -0
- mediapipe/tasks/cc/audio/audio_embedder/__init__.py +0 -0
- mediapipe/tasks/cc/audio/audio_embedder/proto/__init__.py +0 -0
- mediapipe/tasks/cc/audio/audio_embedder/proto/audio_embedder_graph_options_pb2.py +35 -0
- mediapipe/tasks/cc/audio/core/__init__.py +0 -0
- mediapipe/tasks/cc/audio/utils/__init__.py +0 -0
- mediapipe/tasks/cc/components/__init__.py +0 -0
- mediapipe/tasks/cc/components/calculators/__init__.py +0 -0
- mediapipe/tasks/cc/components/calculators/classification_aggregation_calculator_pb2.py +31 -0
- mediapipe/tasks/cc/components/calculators/score_calibration_calculator_pb2.py +35 -0
- mediapipe/tasks/cc/components/calculators/tensors_to_embeddings_calculator_pb2.py +32 -0
- mediapipe/tasks/cc/components/containers/__init__.py +0 -0
- mediapipe/tasks/cc/components/containers/proto/__init__.py +0 -0
- mediapipe/tasks/cc/components/containers/proto/classifications_pb2.py +30 -0
- mediapipe/tasks/cc/components/containers/proto/embeddings_pb2.py +35 -0
- mediapipe/tasks/cc/components/containers/proto/landmarks_detection_result_pb2.py +32 -0
- mediapipe/tasks/cc/components/processors/__init__.py +0 -0
- mediapipe/tasks/cc/components/processors/proto/__init__.py +0 -0
- mediapipe/tasks/cc/components/processors/proto/classification_postprocessing_graph_options_pb2.py +38 -0
- mediapipe/tasks/cc/components/processors/proto/classifier_options_pb2.py +27 -0
- mediapipe/tasks/cc/components/processors/proto/detection_postprocessing_graph_options_pb2.py +36 -0
- mediapipe/tasks/cc/components/processors/proto/detector_options_pb2.py +27 -0
- mediapipe/tasks/cc/components/processors/proto/embedder_options_pb2.py +27 -0
- mediapipe/tasks/cc/components/processors/proto/embedding_postprocessing_graph_options_pb2.py +32 -0
- mediapipe/tasks/cc/components/processors/proto/image_preprocessing_graph_options_pb2.py +34 -0
- mediapipe/tasks/cc/components/processors/proto/text_model_type_pb2.py +28 -0
- mediapipe/tasks/cc/components/processors/proto/text_preprocessing_graph_options_pb2.py +32 -0
- mediapipe/tasks/cc/components/utils/__init__.py +0 -0
- mediapipe/tasks/cc/core/__init__.py +0 -0
- mediapipe/tasks/cc/core/proto/__init__.py +0 -0
- mediapipe/tasks/cc/core/proto/acceleration_pb2.py +28 -0
- mediapipe/tasks/cc/core/proto/base_options_pb2.py +30 -0
- mediapipe/tasks/cc/core/proto/external_file_pb2.py +31 -0
- mediapipe/tasks/cc/core/proto/inference_subgraph_pb2.py +32 -0
- mediapipe/tasks/cc/core/proto/model_resources_calculator_pb2.py +32 -0
- mediapipe/tasks/cc/genai/__init__.py +0 -0
- mediapipe/tasks/cc/genai/inference/__init__.py +0 -0
- mediapipe/tasks/cc/genai/inference/c/__init__.py +0 -0
- mediapipe/tasks/cc/genai/inference/calculators/__init__.py +0 -0
- mediapipe/tasks/cc/genai/inference/calculators/detokenizer_calculator_pb2.py +27 -0
- mediapipe/tasks/cc/genai/inference/calculators/llm_gpu_calculator_pb2.py +32 -0
- mediapipe/tasks/cc/genai/inference/calculators/model_data_calculator_pb2.py +27 -0
- mediapipe/tasks/cc/genai/inference/calculators/tokenizer_calculator_pb2.py +29 -0
- mediapipe/tasks/cc/genai/inference/common/__init__.py +0 -0
- mediapipe/tasks/cc/genai/inference/proto/__init__.py +0 -0
- mediapipe/tasks/cc/genai/inference/proto/llm_file_metadata_pb2.py +32 -0
- mediapipe/tasks/cc/genai/inference/proto/llm_params_pb2.py +33 -0
- mediapipe/tasks/cc/genai/inference/proto/prompt_template_pb2.py +27 -0
- mediapipe/tasks/cc/genai/inference/proto/sampler_params_pb2.py +29 -0
- mediapipe/tasks/cc/genai/inference/proto/transformer_params_pb2.py +45 -0
- mediapipe/tasks/cc/genai/inference/utils/__init__.py +0 -0
- mediapipe/tasks/cc/genai/inference/utils/llm_utils/__init__.py +0 -0
- mediapipe/tasks/cc/genai/inference/utils/xnn_utils/__init__.py +0 -0
- mediapipe/tasks/cc/metadata/__init__.py +0 -0
- mediapipe/tasks/cc/metadata/python/__init__.py +0 -0
- mediapipe/tasks/cc/metadata/python/_pywrap_metadata_version.cpython-312-x86_64-linux-gnu.so +0 -0
- mediapipe/tasks/cc/metadata/tests/__init__.py +0 -0
- mediapipe/tasks/cc/metadata/utils/__init__.py +0 -0
- mediapipe/tasks/cc/text/__init__.py +0 -0
- mediapipe/tasks/cc/text/custom_ops/__init__.py +0 -0
- mediapipe/tasks/cc/text/custom_ops/ragged/__init__.py +0 -0
- mediapipe/tasks/cc/text/custom_ops/sentencepiece/__init__.py +0 -0
- mediapipe/tasks/cc/text/custom_ops/sentencepiece/testdata/__init__.py +0 -0
- mediapipe/tasks/cc/text/language_detector/__init__.py +0 -0
- mediapipe/tasks/cc/text/language_detector/custom_ops/__init__.py +0 -0
- mediapipe/tasks/cc/text/language_detector/custom_ops/utils/__init__.py +0 -0
- mediapipe/tasks/cc/text/language_detector/custom_ops/utils/hash/__init__.py +0 -0
- mediapipe/tasks/cc/text/language_detector/custom_ops/utils/utf/__init__.py +0 -0
- mediapipe/tasks/cc/text/text_classifier/__init__.py +0 -0
- mediapipe/tasks/cc/text/text_classifier/proto/__init__.py +0 -0
- mediapipe/tasks/cc/text/text_classifier/proto/text_classifier_graph_options_pb2.py +35 -0
- mediapipe/tasks/cc/text/text_embedder/__init__.py +0 -0
- mediapipe/tasks/cc/text/text_embedder/proto/__init__.py +0 -0
- mediapipe/tasks/cc/text/text_embedder/proto/text_embedder_graph_options_pb2.py +35 -0
- mediapipe/tasks/cc/text/tokenizers/__init__.py +0 -0
- mediapipe/tasks/cc/text/utils/__init__.py +0 -0
- mediapipe/tasks/cc/vision/__init__.py +0 -0
- mediapipe/tasks/cc/vision/core/__init__.py +0 -0
- mediapipe/tasks/cc/vision/custom_ops/__init__.py +0 -0
- mediapipe/tasks/cc/vision/face_detector/__init__.py +0 -0
- mediapipe/tasks/cc/vision/face_detector/proto/__init__.py +0 -0
- mediapipe/tasks/cc/vision/face_detector/proto/face_detector_graph_options_pb2.py +34 -0
- mediapipe/tasks/cc/vision/face_geometry/__init__.py +0 -0
- mediapipe/tasks/cc/vision/face_geometry/calculators/__init__.py +0 -0
- mediapipe/tasks/cc/vision/face_geometry/calculators/env_generator_calculator_pb2.py +28 -0
- mediapipe/tasks/cc/vision/face_geometry/calculators/geometry_pipeline_calculator_pb2.py +29 -0
- mediapipe/tasks/cc/vision/face_geometry/data/__init__.py +0 -0
- mediapipe/tasks/cc/vision/face_geometry/libs/__init__.py +0 -0
- mediapipe/tasks/cc/vision/face_geometry/proto/__init__.py +0 -0
- mediapipe/tasks/cc/vision/face_geometry/proto/environment_pb2.py +31 -0
- mediapipe/tasks/cc/vision/face_geometry/proto/face_geometry_graph_options_pb2.py +29 -0
- mediapipe/tasks/cc/vision/face_geometry/proto/face_geometry_pb2.py +29 -0
- mediapipe/tasks/cc/vision/face_geometry/proto/geometry_pipeline_metadata_pb2.py +32 -0
- mediapipe/tasks/cc/vision/face_geometry/proto/mesh_3d_pb2.py +31 -0
- mediapipe/tasks/cc/vision/face_landmarker/__init__.py +0 -0
- mediapipe/tasks/cc/vision/face_landmarker/proto/__init__.py +0 -0
- mediapipe/tasks/cc/vision/face_landmarker/proto/face_blendshapes_graph_options_pb2.py +34 -0
- mediapipe/tasks/cc/vision/face_landmarker/proto/face_landmarker_graph_options_pb2.py +37 -0
- mediapipe/tasks/cc/vision/face_landmarker/proto/face_landmarks_detector_graph_options_pb2.py +35 -0
- mediapipe/tasks/cc/vision/face_landmarker/proto/tensors_to_face_landmarks_graph_options_pb2.py +32 -0
- mediapipe/tasks/cc/vision/face_stylizer/__init__.py +0 -0
- mediapipe/tasks/cc/vision/face_stylizer/calculators/__init__.py +0 -0
- mediapipe/tasks/cc/vision/face_stylizer/calculators/tensors_to_image_calculator_pb2.py +36 -0
- mediapipe/tasks/cc/vision/face_stylizer/proto/__init__.py +0 -0
- mediapipe/tasks/cc/vision/face_stylizer/proto/face_stylizer_graph_options_pb2.py +35 -0
- mediapipe/tasks/cc/vision/gesture_recognizer/__init__.py +0 -0
- mediapipe/tasks/cc/vision/gesture_recognizer/calculators/__init__.py +0 -0
- mediapipe/tasks/cc/vision/gesture_recognizer/calculators/combined_prediction_calculator_pb2.py +33 -0
- mediapipe/tasks/cc/vision/gesture_recognizer/calculators/landmarks_to_matrix_calculator_pb2.py +31 -0
- mediapipe/tasks/cc/vision/gesture_recognizer/proto/__init__.py +0 -0
- mediapipe/tasks/cc/vision/gesture_recognizer/proto/gesture_classifier_graph_options_pb2.py +35 -0
- mediapipe/tasks/cc/vision/gesture_recognizer/proto/gesture_embedder_graph_options_pb2.py +34 -0
- mediapipe/tasks/cc/vision/gesture_recognizer/proto/gesture_recognizer_graph_options_pb2.py +36 -0
- mediapipe/tasks/cc/vision/gesture_recognizer/proto/hand_gesture_recognizer_graph_options_pb2.py +36 -0
- mediapipe/tasks/cc/vision/hand_detector/__init__.py +0 -0
- mediapipe/tasks/cc/vision/hand_detector/proto/__init__.py +0 -0
- mediapipe/tasks/cc/vision/hand_detector/proto/hand_detector_graph_options_pb2.py +34 -0
- mediapipe/tasks/cc/vision/hand_detector/proto/hand_detector_result_pb2.py +30 -0
- mediapipe/tasks/cc/vision/hand_landmarker/__init__.py +0 -0
- mediapipe/tasks/cc/vision/hand_landmarker/calculators/__init__.py +0 -0
- mediapipe/tasks/cc/vision/hand_landmarker/calculators/hand_association_calculator_pb2.py +31 -0
- mediapipe/tasks/cc/vision/hand_landmarker/proto/__init__.py +0 -0
- mediapipe/tasks/cc/vision/hand_landmarker/proto/hand_landmarker_graph_options_pb2.py +36 -0
- mediapipe/tasks/cc/vision/hand_landmarker/proto/hand_landmarks_detector_graph_options_pb2.py +34 -0
- mediapipe/tasks/cc/vision/hand_landmarker/proto/hand_roi_refinement_graph_options_pb2.py +28 -0
- mediapipe/tasks/cc/vision/holistic_landmarker/__init__.py +0 -0
- mediapipe/tasks/cc/vision/holistic_landmarker/proto/__init__.py +0 -0
- mediapipe/tasks/cc/vision/holistic_landmarker/proto/holistic_landmarker_graph_options_pb2.py +34 -0
- mediapipe/tasks/cc/vision/holistic_landmarker/proto/holistic_result_pb2.py +29 -0
- mediapipe/tasks/cc/vision/image_classifier/__init__.py +0 -0
- mediapipe/tasks/cc/vision/image_classifier/proto/__init__.py +0 -0
- mediapipe/tasks/cc/vision/image_classifier/proto/image_classifier_graph_options_pb2.py +35 -0
- mediapipe/tasks/cc/vision/image_embedder/__init__.py +0 -0
- mediapipe/tasks/cc/vision/image_embedder/proto/__init__.py +0 -0
- mediapipe/tasks/cc/vision/image_embedder/proto/image_embedder_graph_options_pb2.py +35 -0
- mediapipe/tasks/cc/vision/image_generator/__init__.py +0 -0
- mediapipe/tasks/cc/vision/image_generator/diffuser/__init__.py +0 -0
- mediapipe/tasks/cc/vision/image_generator/diffuser/stable_diffusion_iterate_calculator_pb2.py +40 -0
- mediapipe/tasks/cc/vision/image_generator/proto/__init__.py +0 -0
- mediapipe/tasks/cc/vision/image_generator/proto/conditioned_image_graph_options_pb2.py +40 -0
- mediapipe/tasks/cc/vision/image_generator/proto/control_plugin_graph_options_pb2.py +34 -0
- mediapipe/tasks/cc/vision/image_generator/proto/image_generator_graph_options_pb2.py +30 -0
- mediapipe/tasks/cc/vision/image_segmenter/__init__.py +0 -0
- mediapipe/tasks/cc/vision/image_segmenter/calculators/__init__.py +0 -0
- mediapipe/tasks/cc/vision/image_segmenter/calculators/tensors_to_segmentation_calculator_pb2.py +34 -0
- mediapipe/tasks/cc/vision/image_segmenter/proto/__init__.py +0 -0
- mediapipe/tasks/cc/vision/image_segmenter/proto/image_segmenter_graph_options_pb2.py +35 -0
- mediapipe/tasks/cc/vision/image_segmenter/proto/segmenter_options_pb2.py +33 -0
- mediapipe/tasks/cc/vision/interactive_segmenter/__init__.py +0 -0
- mediapipe/tasks/cc/vision/object_detector/__init__.py +0 -0
- mediapipe/tasks/cc/vision/object_detector/proto/__init__.py +0 -0
- mediapipe/tasks/cc/vision/object_detector/proto/object_detector_options_pb2.py +34 -0
- mediapipe/tasks/cc/vision/pose_detector/__init__.py +0 -0
- mediapipe/tasks/cc/vision/pose_detector/proto/__init__.py +0 -0
- mediapipe/tasks/cc/vision/pose_detector/proto/pose_detector_graph_options_pb2.py +34 -0
- mediapipe/tasks/cc/vision/pose_landmarker/__init__.py +0 -0
- mediapipe/tasks/cc/vision/pose_landmarker/proto/__init__.py +0 -0
- mediapipe/tasks/cc/vision/pose_landmarker/proto/pose_landmarker_graph_options_pb2.py +36 -0
- mediapipe/tasks/cc/vision/pose_landmarker/proto/pose_landmarks_detector_graph_options_pb2.py +34 -0
- mediapipe/tasks/cc/vision/utils/__init__.py +0 -0
- mediapipe/tasks/cc/vision/utils/ghum/__init__.py +0 -0
- mediapipe/tasks/metadata/image_segmenter_metadata_schema.fbs +59 -0
- mediapipe/tasks/metadata/image_segmenter_metadata_schema_py_generated.py +108 -0
- mediapipe/tasks/metadata/metadata_schema.fbs +732 -0
- mediapipe/tasks/metadata/metadata_schema_py_generated.py +3251 -0
- mediapipe/tasks/metadata/object_detector_metadata_schema.fbs +98 -0
- mediapipe/tasks/metadata/object_detector_metadata_schema_py_generated.py +674 -0
- mediapipe/tasks/metadata/schema_py_generated.py +18438 -0
- mediapipe/tasks/python/__init__.py +27 -0
- mediapipe/tasks/python/audio/__init__.py +33 -0
- mediapipe/tasks/python/audio/audio_classifier.py +324 -0
- mediapipe/tasks/python/audio/audio_embedder.py +285 -0
- mediapipe/tasks/python/audio/core/__init__.py +16 -0
- mediapipe/tasks/python/audio/core/audio_record.py +125 -0
- mediapipe/tasks/python/audio/core/audio_task_running_mode.py +29 -0
- mediapipe/tasks/python/audio/core/base_audio_task_api.py +181 -0
- mediapipe/tasks/python/benchmark/__init__.py +13 -0
- mediapipe/tasks/python/benchmark/benchmark_utils.py +70 -0
- mediapipe/tasks/python/benchmark/vision/__init__.py +13 -0
- mediapipe/tasks/python/benchmark/vision/benchmark.py +99 -0
- mediapipe/tasks/python/benchmark/vision/core/__init__.py +14 -0
- mediapipe/tasks/python/benchmark/vision/core/base_vision_benchmark_api.py +40 -0
- mediapipe/tasks/python/components/__init__.py +13 -0
- mediapipe/tasks/python/components/containers/__init__.py +53 -0
- mediapipe/tasks/python/components/containers/audio_data.py +137 -0
- mediapipe/tasks/python/components/containers/bounding_box.py +73 -0
- mediapipe/tasks/python/components/containers/category.py +78 -0
- mediapipe/tasks/python/components/containers/classification_result.py +111 -0
- mediapipe/tasks/python/components/containers/detections.py +181 -0
- mediapipe/tasks/python/components/containers/embedding_result.py +89 -0
- mediapipe/tasks/python/components/containers/keypoint.py +77 -0
- mediapipe/tasks/python/components/containers/landmark.py +122 -0
- mediapipe/tasks/python/components/containers/landmark_detection_result.py +106 -0
- mediapipe/tasks/python/components/containers/rect.py +109 -0
- mediapipe/tasks/python/components/processors/__init__.py +23 -0
- mediapipe/tasks/python/components/processors/classifier_options.py +86 -0
- mediapipe/tasks/python/components/utils/__init__.py +13 -0
- mediapipe/tasks/python/components/utils/cosine_similarity.py +68 -0
- mediapipe/tasks/python/core/__init__.py +13 -0
- mediapipe/tasks/python/core/base_options.py +121 -0
- mediapipe/tasks/python/core/optional_dependencies.py +25 -0
- mediapipe/tasks/python/core/task_info.py +139 -0
- mediapipe/tasks/python/genai/__init__.py +14 -0
- mediapipe/tasks/python/genai/bundler/__init__.py +23 -0
- mediapipe/tasks/python/genai/bundler/llm_bundler.py +130 -0
- mediapipe/tasks/python/genai/bundler/llm_bundler_test.py +168 -0
- mediapipe/tasks/python/genai/converter/__init__.py +24 -0
- mediapipe/tasks/python/genai/converter/converter_base.py +179 -0
- mediapipe/tasks/python/genai/converter/converter_factory.py +79 -0
- mediapipe/tasks/python/genai/converter/llm_converter.py +374 -0
- mediapipe/tasks/python/genai/converter/llm_converter_test.py +63 -0
- mediapipe/tasks/python/genai/converter/pytorch_converter.py +318 -0
- mediapipe/tasks/python/genai/converter/pytorch_converter_test.py +86 -0
- mediapipe/tasks/python/genai/converter/quantization_util.py +516 -0
- mediapipe/tasks/python/genai/converter/quantization_util_test.py +259 -0
- mediapipe/tasks/python/genai/converter/safetensors_converter.py +580 -0
- mediapipe/tasks/python/genai/converter/safetensors_converter_test.py +83 -0
- mediapipe/tasks/python/genai/converter/weight_bins_writer.py +120 -0
- mediapipe/tasks/python/genai/converter/weight_bins_writer_test.py +95 -0
- mediapipe/tasks/python/metadata/__init__.py +13 -0
- mediapipe/tasks/python/metadata/flatbuffers_lib/_pywrap_flatbuffers.cpython-312-x86_64-linux-gnu.so +0 -0
- mediapipe/tasks/python/metadata/metadata.py +928 -0
- mediapipe/tasks/python/metadata/metadata_displayer_cli.py +34 -0
- mediapipe/tasks/python/metadata/metadata_writers/__init__.py +13 -0
- mediapipe/tasks/python/metadata/metadata_writers/face_stylizer.py +138 -0
- mediapipe/tasks/python/metadata/metadata_writers/image_classifier.py +71 -0
- mediapipe/tasks/python/metadata/metadata_writers/image_segmenter.py +170 -0
- mediapipe/tasks/python/metadata/metadata_writers/metadata_info.py +1166 -0
- mediapipe/tasks/python/metadata/metadata_writers/metadata_writer.py +845 -0
- mediapipe/tasks/python/metadata/metadata_writers/model_asset_bundle_utils.py +71 -0
- mediapipe/tasks/python/metadata/metadata_writers/object_detector.py +331 -0
- mediapipe/tasks/python/metadata/metadata_writers/text_classifier.py +119 -0
- mediapipe/tasks/python/metadata/metadata_writers/writer_utils.py +91 -0
- mediapipe/tasks/python/test/__init__.py +13 -0
- mediapipe/tasks/python/test/audio/__init__.py +13 -0
- mediapipe/tasks/python/test/audio/audio_classifier_test.py +387 -0
- mediapipe/tasks/python/test/audio/audio_embedder_test.py +297 -0
- mediapipe/tasks/python/test/test_utils.py +196 -0
- mediapipe/tasks/python/test/text/__init__.py +13 -0
- mediapipe/tasks/python/test/text/language_detector_test.py +228 -0
- mediapipe/tasks/python/test/text/text_classifier_test.py +235 -0
- mediapipe/tasks/python/test/text/text_embedder_test.py +326 -0
- mediapipe/tasks/python/test/vision/__init__.py +13 -0
- mediapipe/tasks/python/test/vision/face_aligner_test.py +190 -0
- mediapipe/tasks/python/test/vision/face_detector_test.py +523 -0
- mediapipe/tasks/python/test/vision/face_landmarker_test.py +565 -0
- mediapipe/tasks/python/test/vision/face_stylizer_test.py +191 -0
- mediapipe/tasks/python/test/vision/hand_landmarker_test.py +437 -0
- mediapipe/tasks/python/test/vision/holistic_landmarker_test.py +544 -0
- mediapipe/tasks/python/test/vision/image_classifier_test.py +657 -0
- mediapipe/tasks/python/test/vision/image_embedder_test.py +423 -0
- mediapipe/tasks/python/test/vision/image_segmenter_test.py +512 -0
- mediapipe/tasks/python/test/vision/interactive_segmenter_test.py +341 -0
- mediapipe/tasks/python/test/vision/object_detector_test.py +493 -0
- mediapipe/tasks/python/test/vision/pose_landmarker_test.py +518 -0
- mediapipe/tasks/python/text/__init__.py +35 -0
- mediapipe/tasks/python/text/core/__init__.py +16 -0
- mediapipe/tasks/python/text/core/base_text_task_api.py +54 -0
- mediapipe/tasks/python/text/language_detector.py +220 -0
- mediapipe/tasks/python/text/text_classifier.py +187 -0
- mediapipe/tasks/python/text/text_embedder.py +188 -0
- mediapipe/tasks/python/vision/__init__.py +90 -0
- mediapipe/tasks/python/vision/core/__init__.py +14 -0
- mediapipe/tasks/python/vision/core/base_vision_task_api.py +226 -0
- mediapipe/tasks/python/vision/core/image_processing_options.py +39 -0
- mediapipe/tasks/python/vision/core/vision_task_running_mode.py +31 -0
- mediapipe/tasks/python/vision/face_aligner.py +158 -0
- mediapipe/tasks/python/vision/face_detector.py +332 -0
- mediapipe/tasks/python/vision/face_landmarker.py +3244 -0
- mediapipe/tasks/python/vision/face_stylizer.py +158 -0
- mediapipe/tasks/python/vision/gesture_recognizer.py +480 -0
- mediapipe/tasks/python/vision/hand_landmarker.py +504 -0
- mediapipe/tasks/python/vision/holistic_landmarker.py +576 -0
- mediapipe/tasks/python/vision/image_classifier.py +358 -0
- mediapipe/tasks/python/vision/image_embedder.py +362 -0
- mediapipe/tasks/python/vision/image_segmenter.py +433 -0
- mediapipe/tasks/python/vision/interactive_segmenter.py +285 -0
- mediapipe/tasks/python/vision/object_detector.py +389 -0
- mediapipe/tasks/python/vision/pose_landmarker.py +455 -0
- mediapipe/util/__init__.py +0 -0
- mediapipe/util/analytics/__init__.py +0 -0
- mediapipe/util/analytics/mediapipe_log_extension_pb2.py +44 -0
- mediapipe/util/analytics/mediapipe_logging_enums_pb2.py +37 -0
- mediapipe/util/audio_decoder_pb2.py +33 -0
- mediapipe/util/color_pb2.py +33 -0
- mediapipe/util/label_map_pb2.py +27 -0
- mediapipe/util/render_data_pb2.py +58 -0
- mediapipe/util/sequence/__init__.py +14 -0
- mediapipe/util/sequence/media_sequence.py +716 -0
- mediapipe/util/sequence/media_sequence_test.py +290 -0
- mediapipe/util/sequence/media_sequence_util.py +800 -0
- mediapipe/util/sequence/media_sequence_util_test.py +389 -0
- mediapipe/util/tracking/__init__.py +0 -0
- mediapipe/util/tracking/box_detector_pb2.py +39 -0
- mediapipe/util/tracking/box_tracker_pb2.py +32 -0
- mediapipe/util/tracking/camera_motion_pb2.py +31 -0
- mediapipe/util/tracking/flow_packager_pb2.py +60 -0
- mediapipe/util/tracking/frame_selection_pb2.py +35 -0
- mediapipe/util/tracking/frame_selection_solution_evaluator_pb2.py +28 -0
- mediapipe/util/tracking/motion_analysis_pb2.py +35 -0
- mediapipe/util/tracking/motion_estimation_pb2.py +66 -0
- mediapipe/util/tracking/motion_models_pb2.py +42 -0
- mediapipe/util/tracking/motion_saliency_pb2.py +26 -0
- mediapipe/util/tracking/push_pull_filtering_pb2.py +26 -0
- mediapipe/util/tracking/region_flow_computation_pb2.py +59 -0
- mediapipe/util/tracking/region_flow_pb2.py +49 -0
- mediapipe/util/tracking/tone_estimation_pb2.py +45 -0
- mediapipe/util/tracking/tone_models_pb2.py +32 -0
- mediapipe/util/tracking/tracked_detection_manager_config_pb2.py +26 -0
- mediapipe/util/tracking/tracking_pb2.py +73 -0
- mediapipe_nightly-0.10.21.post20250114.dist-info/LICENSE +218 -0
- mediapipe_nightly-0.10.21.post20250114.dist-info/METADATA +199 -0
- mediapipe_nightly-0.10.21.post20250114.dist-info/RECORD +593 -0
- mediapipe_nightly-0.10.21.post20250114.dist-info/WHEEL +5 -0
- mediapipe_nightly-0.10.21.post20250114.dist-info/top_level.txt +4 -0
- mediapipe_nightly.libs/libEGL-48f73270.so.1.1.0 +0 -0
- mediapipe_nightly.libs/libGLESv2-ed5eda4f.so.2.1.0 +0 -0
- mediapipe_nightly.libs/libGLdispatch-64b28464.so.0.0.0 +0 -0
@@ -0,0 +1,389 @@
|
|
1
|
+
"""Copyright 2019 The MediaPipe Authors.
|
2
|
+
|
3
|
+
Licensed under the Apache License, Version 2.0 (the "License");
|
4
|
+
you may not use this file except in compliance with the License.
|
5
|
+
You may obtain a copy of the License at
|
6
|
+
|
7
|
+
http://www.apache.org/licenses/LICENSE-2.0
|
8
|
+
|
9
|
+
Unless required by applicable law or agreed to in writing, software
|
10
|
+
distributed under the License is distributed on an "AS IS" BASIS,
|
11
|
+
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
12
|
+
See the License for the specific language governing permissions and
|
13
|
+
limitations under the License.
|
14
|
+
|
15
|
+
Tests for media_sequence_util.py.
|
16
|
+
"""
|
17
|
+
|
18
|
+
from __future__ import absolute_import
|
19
|
+
from __future__ import division
|
20
|
+
from __future__ import print_function
|
21
|
+
|
22
|
+
import tensorflow.compat.v1 as tf
|
23
|
+
|
24
|
+
from mediapipe.util.sequence import media_sequence_util as msu
|
25
|
+
|
26
|
+
# run this code at the module level to procedurally define functions for test
|
27
|
+
# cases.
|
28
|
+
msu.create_bytes_context_feature(
|
29
|
+
"string_context", "string_feature", module_dict=msu.__dict__)
|
30
|
+
msu.create_float_context_feature(
|
31
|
+
"float_context", "float_feature", module_dict=msu.__dict__)
|
32
|
+
msu.create_int_context_feature(
|
33
|
+
"int64_context", "int64_feature", module_dict=msu.__dict__)
|
34
|
+
msu.create_bytes_list_context_feature(
|
35
|
+
"string_list_context", "string_vector_feature", module_dict=msu.__dict__)
|
36
|
+
msu.create_float_list_context_feature(
|
37
|
+
"float_list_context", "float_vector_feature", module_dict=msu.__dict__)
|
38
|
+
msu.create_int_list_context_feature(
|
39
|
+
"int64_list_context", "int64_vector_feature", module_dict=msu.__dict__)
|
40
|
+
msu.create_bytes_feature_list(
|
41
|
+
"string_feature_list", "string_feature_list", module_dict=msu.__dict__)
|
42
|
+
msu.create_float_feature_list(
|
43
|
+
"float_feature_list", "float_feature_list", module_dict=msu.__dict__)
|
44
|
+
msu.create_int_feature_list(
|
45
|
+
"int64_feature_list", "int64_feature_list", module_dict=msu.__dict__)
|
46
|
+
msu.create_bytes_list_feature_list(
|
47
|
+
"string_list_feature_list", "string_list_feature_list",
|
48
|
+
module_dict=msu.__dict__)
|
49
|
+
msu.create_float_list_feature_list(
|
50
|
+
"float_list_feature_list", "float_list_feature_list",
|
51
|
+
module_dict=msu.__dict__)
|
52
|
+
msu.create_int_list_feature_list(
|
53
|
+
"int64_list_feature_list", "int64_list_feature_list",
|
54
|
+
module_dict=msu.__dict__)
|
55
|
+
|
56
|
+
|
57
|
+
class MediaSequenceTest(tf.test.TestCase):
|
58
|
+
|
59
|
+
def test_set_context_float(self):
|
60
|
+
example = tf.train.SequenceExample()
|
61
|
+
key = "test_float"
|
62
|
+
msu.set_context_float(key, 0.1, example)
|
63
|
+
self.assertAlmostEqual(0.1,
|
64
|
+
example.context.feature[key].float_list.value[0])
|
65
|
+
|
66
|
+
def test_set_context_bytes(self):
|
67
|
+
example = tf.train.SequenceExample()
|
68
|
+
key = "test_string"
|
69
|
+
msu.set_context_bytes(key, b"test", example)
|
70
|
+
self.assertEqual(b"test",
|
71
|
+
example.context.feature[key].bytes_list.value[0])
|
72
|
+
|
73
|
+
def test_set_context_int(self):
|
74
|
+
example = tf.train.SequenceExample()
|
75
|
+
key = "test_int"
|
76
|
+
msu.set_context_int(key, 47, example)
|
77
|
+
self.assertEqual(47,
|
78
|
+
example.context.feature[key].int64_list.value[0])
|
79
|
+
|
80
|
+
def test_set_context_float_list(self):
|
81
|
+
example = tf.train.SequenceExample()
|
82
|
+
key = "test_float_list"
|
83
|
+
msu.set_context_float_list(key, [0.0, 0.1], example)
|
84
|
+
self.assertSequenceAlmostEqual(
|
85
|
+
[0.0, 0.1], example.context.feature[key].float_list.value)
|
86
|
+
|
87
|
+
def test_set_context_bytes_list(self):
|
88
|
+
example = tf.train.SequenceExample()
|
89
|
+
key = "test_string_list"
|
90
|
+
msu.set_context_bytes_list(key, [b"test", b"test"], example)
|
91
|
+
self.assertSequenceAlmostEqual(
|
92
|
+
[b"test", b"test"], example.context.feature[key].bytes_list.value)
|
93
|
+
|
94
|
+
def test_set_context_int_list(self):
|
95
|
+
example = tf.train.SequenceExample()
|
96
|
+
key = "test_int_list"
|
97
|
+
msu.set_context_int_list(key, [0, 1], example)
|
98
|
+
self.assertSequenceAlmostEqual(
|
99
|
+
[0, 1], example.context.feature[key].int64_list.value)
|
100
|
+
|
101
|
+
def test_round_trip_float_list_feature(self):
|
102
|
+
example = tf.train.SequenceExample()
|
103
|
+
key = "test_float_features"
|
104
|
+
msu.add_float_list(key, [0.0, 0.1], example)
|
105
|
+
msu.add_float_list(key, [0.1, 0.2], example)
|
106
|
+
self.assertEqual(2, msu.get_feature_list_size(key, example))
|
107
|
+
self.assertTrue(msu.has_feature_list(key, example))
|
108
|
+
self.assertAllClose([0.0, 0.1], msu.get_float_list_at(key, 0, example))
|
109
|
+
self.assertAllClose([0.1, 0.2], msu.get_float_list_at(key, 1, example))
|
110
|
+
msu.clear_feature_list(key, example)
|
111
|
+
self.assertEqual(0, msu.get_feature_list_size(key, example))
|
112
|
+
self.assertFalse(msu.has_feature_list(key, example))
|
113
|
+
|
114
|
+
def test_round_trip_bytes_list_feature(self):
|
115
|
+
example = tf.train.SequenceExample()
|
116
|
+
key = "test_bytes_features"
|
117
|
+
msu.add_bytes_list(key, [b"test0", b"test1"], example)
|
118
|
+
msu.add_bytes_list(key, [b"test1", b"test2"], example)
|
119
|
+
self.assertEqual(2, msu.get_feature_list_size(key, example))
|
120
|
+
self.assertTrue(msu.has_feature_list(key, example))
|
121
|
+
self.assertAllEqual([b"test0", b"test1"],
|
122
|
+
msu.get_bytes_list_at(key, 0, example))
|
123
|
+
self.assertAllEqual([b"test1", b"test2"],
|
124
|
+
msu.get_bytes_list_at(key, 1, example))
|
125
|
+
msu.clear_feature_list(key, example)
|
126
|
+
self.assertEqual(0, msu.get_feature_list_size(key, example))
|
127
|
+
self.assertFalse(msu.has_feature_list(key, example))
|
128
|
+
|
129
|
+
def test_round_trip_int_list_feature(self):
|
130
|
+
example = tf.train.SequenceExample()
|
131
|
+
key = "test_ints_features"
|
132
|
+
msu.add_int_list(key, [0, 1], example)
|
133
|
+
msu.add_int_list(key, [1, 2], example)
|
134
|
+
self.assertEqual(2, msu.get_feature_list_size(key, example))
|
135
|
+
self.assertTrue(msu.has_feature_list(key, example))
|
136
|
+
self.assertAllClose([0, 1], msu.get_int_list_at(key, 0, example))
|
137
|
+
self.assertAllClose([1, 2], msu.get_int_list_at(key, 1, example))
|
138
|
+
msu.clear_feature_list(key, example)
|
139
|
+
self.assertEqual(0, msu.get_feature_list_size(key, example))
|
140
|
+
self.assertFalse(msu.has_feature_list(key, example))
|
141
|
+
|
142
|
+
def test_round_trip_string_context(self):
|
143
|
+
example = tf.train.SequenceExample()
|
144
|
+
key = "string_feature"
|
145
|
+
msu.set_string_context(b"test", example)
|
146
|
+
self.assertEqual(b"test", msu.get_string_context(example))
|
147
|
+
self.assertTrue(msu.has_string_context(example))
|
148
|
+
self.assertEqual(b"test",
|
149
|
+
example.context.feature[key].bytes_list.value[0])
|
150
|
+
msu.clear_string_context(example)
|
151
|
+
self.assertFalse(msu.has_string_context(example))
|
152
|
+
self.assertEqual("string_feature", msu.get_string_context_key())
|
153
|
+
|
154
|
+
def test_round_trip_float_context(self):
|
155
|
+
example = tf.train.SequenceExample()
|
156
|
+
key = "float_feature"
|
157
|
+
msu.set_float_context(0.47, example)
|
158
|
+
self.assertAlmostEqual(0.47, msu.get_float_context(example))
|
159
|
+
self.assertTrue(msu.has_float_context(example))
|
160
|
+
self.assertAlmostEqual(0.47,
|
161
|
+
example.context.feature[key].float_list.value[0])
|
162
|
+
msu.clear_float_context(example)
|
163
|
+
self.assertFalse(msu.has_float_context(example))
|
164
|
+
self.assertEqual("float_feature", msu.get_float_context_key())
|
165
|
+
|
166
|
+
def test_round_trip_int_context(self):
|
167
|
+
example = tf.train.SequenceExample()
|
168
|
+
key = "int64_feature"
|
169
|
+
msu.set_int64_context(47, example)
|
170
|
+
self.assertEqual(47, msu.get_int64_context(example))
|
171
|
+
self.assertTrue(msu.has_int64_context(example))
|
172
|
+
self.assertEqual(47,
|
173
|
+
example.context.feature[key].int64_list.value[0])
|
174
|
+
msu.clear_int64_context(example)
|
175
|
+
self.assertFalse(msu.has_int64_context(example))
|
176
|
+
self.assertEqual("int64_feature", msu.get_int64_context_key())
|
177
|
+
|
178
|
+
def test_round_trip_string_list_context(self):
|
179
|
+
example = tf.train.SequenceExample()
|
180
|
+
msu.set_string_list_context([b"test0", b"test1"], example)
|
181
|
+
self.assertSequenceEqual([b"test0", b"test1"],
|
182
|
+
msu.get_string_list_context(example))
|
183
|
+
self.assertTrue(msu.has_string_list_context(example))
|
184
|
+
msu.clear_string_list_context(example)
|
185
|
+
self.assertFalse(msu.has_string_list_context(example))
|
186
|
+
self.assertEqual("string_vector_feature",
|
187
|
+
msu.get_string_list_context_key())
|
188
|
+
|
189
|
+
def test_round_trip_float_list_context(self):
|
190
|
+
example = tf.train.SequenceExample()
|
191
|
+
msu.set_float_list_context([0.47, 0.49], example)
|
192
|
+
self.assertSequenceAlmostEqual([0.47, 0.49],
|
193
|
+
msu.get_float_list_context(example))
|
194
|
+
self.assertTrue(msu.has_float_list_context(example))
|
195
|
+
msu.clear_float_list_context(example)
|
196
|
+
self.assertFalse(msu.has_float_list_context(example))
|
197
|
+
self.assertEqual("float_vector_feature", msu.get_float_list_context_key())
|
198
|
+
|
199
|
+
def test_round_trip_int_list_context(self):
|
200
|
+
example = tf.train.SequenceExample()
|
201
|
+
msu.set_int64_list_context([47, 49], example)
|
202
|
+
self.assertSequenceEqual([47, 49], msu.get_int64_list_context(example))
|
203
|
+
self.assertTrue(msu.has_int64_list_context(example))
|
204
|
+
msu.clear_int64_list_context(example)
|
205
|
+
self.assertFalse(msu.has_int64_list_context(example))
|
206
|
+
self.assertEqual("int64_vector_feature", msu.get_int64_list_context_key())
|
207
|
+
|
208
|
+
def test_round_trip_string_feature_list(self):
|
209
|
+
example = tf.train.SequenceExample()
|
210
|
+
msu.add_string_feature_list(b"test0", example)
|
211
|
+
msu.add_string_feature_list(b"test1", example)
|
212
|
+
self.assertEqual(b"test0", msu.get_string_feature_list_at(0, example))
|
213
|
+
self.assertEqual(b"test1", msu.get_string_feature_list_at(1, example))
|
214
|
+
self.assertTrue(msu.has_string_feature_list(example))
|
215
|
+
self.assertEqual(2, msu.get_string_feature_list_size(example))
|
216
|
+
msu.clear_string_feature_list(example)
|
217
|
+
self.assertFalse(msu.has_string_feature_list(example))
|
218
|
+
self.assertEqual(0, msu.get_string_feature_list_size(example))
|
219
|
+
self.assertEqual("string_feature_list", msu.get_string_feature_list_key())
|
220
|
+
|
221
|
+
def test_round_trip_float_feature_list(self):
|
222
|
+
example = tf.train.SequenceExample()
|
223
|
+
msu.add_float_feature_list(0.47, example)
|
224
|
+
msu.add_float_feature_list(0.49, example)
|
225
|
+
self.assertAlmostEqual(0.47, msu.get_float_feature_list_at(0, example))
|
226
|
+
self.assertAlmostEqual(0.49, msu.get_float_feature_list_at(1, example))
|
227
|
+
self.assertTrue(msu.has_float_feature_list(example))
|
228
|
+
self.assertEqual(2, msu.get_float_feature_list_size(example))
|
229
|
+
msu.clear_float_feature_list(example)
|
230
|
+
self.assertFalse(msu.has_float_feature_list(example))
|
231
|
+
self.assertEqual(0, msu.get_float_feature_list_size(example))
|
232
|
+
self.assertEqual("float_feature_list", msu.get_float_feature_list_key())
|
233
|
+
|
234
|
+
def test_round_trip_int_feature_list(self):
|
235
|
+
example = tf.train.SequenceExample()
|
236
|
+
msu.add_int64_feature_list(47, example)
|
237
|
+
msu.add_int64_feature_list(49, example)
|
238
|
+
self.assertEqual(47, msu.get_int64_feature_list_at(0, example))
|
239
|
+
self.assertEqual(49, msu.get_int64_feature_list_at(1, example))
|
240
|
+
self.assertTrue(msu.has_int64_feature_list(example))
|
241
|
+
self.assertEqual(2, msu.get_int64_feature_list_size(example))
|
242
|
+
msu.clear_int64_feature_list(example)
|
243
|
+
self.assertFalse(msu.has_int64_feature_list(example))
|
244
|
+
self.assertEqual(0, msu.get_int64_feature_list_size(example))
|
245
|
+
self.assertEqual("int64_feature_list", msu.get_int64_feature_list_key())
|
246
|
+
|
247
|
+
def test_round_trip_string_list_feature_list(self):
|
248
|
+
example = tf.train.SequenceExample()
|
249
|
+
msu.add_string_list_feature_list([b"test0", b"test1"], example)
|
250
|
+
msu.add_string_list_feature_list([b"test1", b"test2"], example)
|
251
|
+
self.assertSequenceEqual([b"test0", b"test1"],
|
252
|
+
msu.get_string_list_feature_list_at(0, example))
|
253
|
+
self.assertSequenceEqual([b"test1", b"test2"],
|
254
|
+
msu.get_string_list_feature_list_at(1, example))
|
255
|
+
self.assertTrue(msu.has_string_list_feature_list(example))
|
256
|
+
self.assertEqual(2, msu.get_string_list_feature_list_size(example))
|
257
|
+
msu.clear_string_list_feature_list(example)
|
258
|
+
self.assertFalse(msu.has_string_list_feature_list(example))
|
259
|
+
self.assertEqual(0, msu.get_string_list_feature_list_size(example))
|
260
|
+
self.assertEqual("string_list_feature_list",
|
261
|
+
msu.get_string_list_feature_list_key())
|
262
|
+
|
263
|
+
def test_round_trip_float_list_feature_list(self):
|
264
|
+
example = tf.train.SequenceExample()
|
265
|
+
msu.add_float_list_feature_list([0.47, 0.49], example)
|
266
|
+
msu.add_float_list_feature_list([0.49, 0.50], example)
|
267
|
+
self.assertSequenceAlmostEqual(
|
268
|
+
[0.47, 0.49], msu.get_float_list_feature_list_at(0, example))
|
269
|
+
self.assertSequenceAlmostEqual(
|
270
|
+
[0.49, 0.50], msu.get_float_list_feature_list_at(1, example))
|
271
|
+
self.assertTrue(msu.has_float_list_feature_list(example))
|
272
|
+
self.assertEqual(2, msu.get_float_list_feature_list_size(example))
|
273
|
+
msu.clear_float_list_feature_list(example)
|
274
|
+
self.assertFalse(msu.has_float_list_feature_list(example))
|
275
|
+
self.assertEqual(0, msu.get_float_list_feature_list_size(example))
|
276
|
+
self.assertEqual("float_list_feature_list",
|
277
|
+
msu.get_float_list_feature_list_key())
|
278
|
+
|
279
|
+
def test_round_trip_int_list_feature_list(self):
|
280
|
+
example = tf.train.SequenceExample()
|
281
|
+
msu.add_int64_list_feature_list([47, 49], example)
|
282
|
+
msu.add_int64_list_feature_list([49, 50], example)
|
283
|
+
self.assertSequenceEqual(
|
284
|
+
[47, 49], msu.get_int64_list_feature_list_at(0, example))
|
285
|
+
self.assertSequenceEqual(
|
286
|
+
[49, 50], msu.get_int64_list_feature_list_at(1, example))
|
287
|
+
self.assertTrue(msu.has_int64_list_feature_list(example))
|
288
|
+
self.assertEqual(2, msu.get_int64_list_feature_list_size(example))
|
289
|
+
msu.clear_int64_list_feature_list(example)
|
290
|
+
self.assertFalse(msu.has_int64_list_feature_list(example))
|
291
|
+
self.assertEqual(0, msu.get_int64_list_feature_list_size(example))
|
292
|
+
self.assertEqual("int64_list_feature_list",
|
293
|
+
msu.get_int64_list_feature_list_key())
|
294
|
+
|
295
|
+
def test_prefix_int64_context(self):
|
296
|
+
example = tf.train.SequenceExample()
|
297
|
+
msu.set_int64_context(47, example, prefix="magic")
|
298
|
+
self.assertFalse(msu.has_int64_context(example))
|
299
|
+
self.assertTrue(msu.has_int64_context(example, prefix="magic"))
|
300
|
+
self.assertEqual(47, msu.get_int64_context(example, prefix="magic"))
|
301
|
+
|
302
|
+
def test_prefix_float_context(self):
|
303
|
+
example = tf.train.SequenceExample()
|
304
|
+
msu.set_float_context(47., example, prefix="magic")
|
305
|
+
self.assertFalse(msu.has_float_context(example))
|
306
|
+
self.assertTrue(msu.has_float_context(example, prefix="magic"))
|
307
|
+
self.assertAlmostEqual(47., msu.get_float_context(example, prefix="magic"))
|
308
|
+
|
309
|
+
def test_prefix_string_context(self):
|
310
|
+
example = tf.train.SequenceExample()
|
311
|
+
msu.set_string_context(b"47", example, prefix="magic")
|
312
|
+
self.assertFalse(msu.has_string_context(example))
|
313
|
+
self.assertTrue(msu.has_string_context(example, prefix="magic"))
|
314
|
+
self.assertEqual(b"47", msu.get_string_context(example, prefix="magic"))
|
315
|
+
|
316
|
+
def test_prefix_int64_list_context(self):
|
317
|
+
example = tf.train.SequenceExample()
|
318
|
+
msu.set_int64_list_context((47,), example, prefix="magic")
|
319
|
+
self.assertFalse(msu.has_int64_list_context(example))
|
320
|
+
self.assertTrue(msu.has_int64_list_context(example, prefix="magic"))
|
321
|
+
self.assertEqual([47,], msu.get_int64_list_context(example, prefix="magic"))
|
322
|
+
|
323
|
+
def test_prefix_float_list_context(self):
|
324
|
+
example = tf.train.SequenceExample()
|
325
|
+
msu.set_float_list_context((47.,), example, prefix="magic")
|
326
|
+
self.assertFalse(msu.has_float_list_context(example))
|
327
|
+
self.assertTrue(msu.has_float_list_context(example, prefix="magic"))
|
328
|
+
self.assertAlmostEqual([47.,],
|
329
|
+
msu.get_float_list_context(example, prefix="magic"))
|
330
|
+
|
331
|
+
def test_prefix_string_list_context(self):
|
332
|
+
example = tf.train.SequenceExample()
|
333
|
+
msu.set_string_list_context((b"47",), example, prefix="magic")
|
334
|
+
self.assertFalse(msu.has_string_list_context(example))
|
335
|
+
self.assertTrue(msu.has_string_list_context(example, prefix="magic"))
|
336
|
+
self.assertEqual([b"47",],
|
337
|
+
msu.get_string_list_context(example, prefix="magic"))
|
338
|
+
|
339
|
+
def test_prefix_int64_feature_list(self):
|
340
|
+
example = tf.train.SequenceExample()
|
341
|
+
msu.add_int64_feature_list(47, example, prefix="magic")
|
342
|
+
self.assertFalse(msu.has_int64_feature_list(example))
|
343
|
+
self.assertTrue(msu.has_int64_feature_list(example, prefix="magic"))
|
344
|
+
self.assertEqual(47,
|
345
|
+
msu.get_int64_feature_list_at(0, example, prefix="magic"))
|
346
|
+
|
347
|
+
def test_prefix_float_feature_list(self):
|
348
|
+
example = tf.train.SequenceExample()
|
349
|
+
msu.add_float_feature_list(47., example, prefix="magic")
|
350
|
+
self.assertFalse(msu.has_float_feature_list(example))
|
351
|
+
self.assertTrue(msu.has_float_feature_list(example, prefix="magic"))
|
352
|
+
self.assertAlmostEqual(
|
353
|
+
47., msu.get_float_feature_list_at(0, example, prefix="magic"))
|
354
|
+
|
355
|
+
def test_prefix_string_feature_list(self):
|
356
|
+
example = tf.train.SequenceExample()
|
357
|
+
msu.add_string_feature_list(b"47", example, prefix="magic")
|
358
|
+
self.assertFalse(msu.has_string_feature_list(example))
|
359
|
+
self.assertTrue(msu.has_string_feature_list(example, prefix="magic"))
|
360
|
+
self.assertEqual(
|
361
|
+
b"47", msu.get_string_feature_list_at(0, example, prefix="magic"))
|
362
|
+
|
363
|
+
def test_prefix_int64_list_feature_list(self):
|
364
|
+
example = tf.train.SequenceExample()
|
365
|
+
msu.add_int64_list_feature_list((47,), example, prefix="magic")
|
366
|
+
self.assertFalse(msu.has_int64_list_feature_list(example))
|
367
|
+
self.assertTrue(msu.has_int64_list_feature_list(example, prefix="magic"))
|
368
|
+
self.assertEqual(
|
369
|
+
[47,], msu.get_int64_list_feature_list_at(0, example, prefix="magic"))
|
370
|
+
|
371
|
+
def test_prefix_float_list_feature_list(self):
|
372
|
+
example = tf.train.SequenceExample()
|
373
|
+
msu.add_float_list_feature_list((47.,), example, prefix="magic")
|
374
|
+
self.assertFalse(msu.has_float_list_feature_list(example))
|
375
|
+
self.assertTrue(msu.has_float_list_feature_list(example, prefix="magic"))
|
376
|
+
self.assertAlmostEqual(
|
377
|
+
[47.,], msu.get_float_list_feature_list_at(0, example, prefix="magic"))
|
378
|
+
|
379
|
+
def test_prefix_string_list_feature_list(self):
|
380
|
+
example = tf.train.SequenceExample()
|
381
|
+
msu.add_string_list_feature_list((b"47",), example, prefix="magic")
|
382
|
+
self.assertFalse(msu.has_string_list_feature_list(example))
|
383
|
+
self.assertTrue(msu.has_string_list_feature_list(example, prefix="magic"))
|
384
|
+
self.assertEqual(
|
385
|
+
[b"47",],
|
386
|
+
msu.get_string_list_feature_list_at(0, example, prefix="magic"))
|
387
|
+
|
388
|
+
if __name__ == "__main__":
|
389
|
+
tf.test.main()
|
File without changes
|
@@ -0,0 +1,39 @@
|
|
1
|
+
# -*- coding: utf-8 -*-
|
2
|
+
# Generated by the protocol buffer compiler. DO NOT EDIT!
|
3
|
+
# source: mediapipe/util/tracking/box_detector.proto
|
4
|
+
# Protobuf Python Version: 4.25.1
|
5
|
+
"""Generated protocol buffer code."""
|
6
|
+
from google.protobuf import descriptor as _descriptor
|
7
|
+
from google.protobuf import descriptor_pool as _descriptor_pool
|
8
|
+
from google.protobuf import symbol_database as _symbol_database
|
9
|
+
from google.protobuf.internal import builder as _builder
|
10
|
+
# @@protoc_insertion_point(imports)
|
11
|
+
|
12
|
+
_sym_db = _symbol_database.Default()
|
13
|
+
|
14
|
+
|
15
|
+
from mediapipe.util.tracking import box_tracker_pb2 as mediapipe_dot_util_dot_tracking_dot_box__tracker__pb2
|
16
|
+
from mediapipe.util.tracking import region_flow_pb2 as mediapipe_dot_util_dot_tracking_dot_region__flow__pb2
|
17
|
+
|
18
|
+
|
19
|
+
DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(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')
|
20
|
+
|
21
|
+
_globals = globals()
|
22
|
+
_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
|
23
|
+
_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'mediapipe.util.tracking.box_detector_pb2', _globals)
|
24
|
+
if _descriptor._USE_C_DESCRIPTORS == False:
|
25
|
+
_globals['DESCRIPTOR']._options = None
|
26
|
+
_globals['DESCRIPTOR']._serialized_options = b'\n\035com.google.mediapipe.trackingB\020BoxDetectorProto'
|
27
|
+
_globals['_BOXDETECTOROPTIONS']._serialized_start=144
|
28
|
+
_globals['_BOXDETECTOROPTIONS']._serialized_end=764
|
29
|
+
_globals['_BOXDETECTOROPTIONS_IMAGEQUERYSETTINGS']._serialized_start=566
|
30
|
+
_globals['_BOXDETECTOROPTIONS_IMAGEQUERYSETTINGS']._serialized_end=713
|
31
|
+
_globals['_BOXDETECTOROPTIONS_INDEXTYPE']._serialized_start=715
|
32
|
+
_globals['_BOXDETECTOROPTIONS_INDEXTYPE']._serialized_end=764
|
33
|
+
_globals['_BOXDETECTORINDEX']._serialized_start=767
|
34
|
+
_globals['_BOXDETECTORINDEX']._serialized_end=1054
|
35
|
+
_globals['_BOXDETECTORINDEX_BOXENTRY']._serialized_start=845
|
36
|
+
_globals['_BOXDETECTORINDEX_BOXENTRY']._serialized_end=1054
|
37
|
+
_globals['_BOXDETECTORINDEX_BOXENTRY_FRAMEENTRY']._serialized_start=927
|
38
|
+
_globals['_BOXDETECTORINDEX_BOXENTRY_FRAMEENTRY']._serialized_end=1054
|
39
|
+
# @@protoc_insertion_point(module_scope)
|
@@ -0,0 +1,32 @@
|
|
1
|
+
# -*- coding: utf-8 -*-
|
2
|
+
# Generated by the protocol buffer compiler. DO NOT EDIT!
|
3
|
+
# source: mediapipe/util/tracking/box_tracker.proto
|
4
|
+
# Protobuf Python Version: 4.25.1
|
5
|
+
"""Generated protocol buffer code."""
|
6
|
+
from google.protobuf import descriptor as _descriptor
|
7
|
+
from google.protobuf import descriptor_pool as _descriptor_pool
|
8
|
+
from google.protobuf import symbol_database as _symbol_database
|
9
|
+
from google.protobuf.internal import builder as _builder
|
10
|
+
# @@protoc_insertion_point(imports)
|
11
|
+
|
12
|
+
_sym_db = _symbol_database.Default()
|
13
|
+
|
14
|
+
|
15
|
+
from mediapipe.util.tracking import tracking_pb2 as mediapipe_dot_util_dot_tracking_dot_tracking__pb2
|
16
|
+
|
17
|
+
|
18
|
+
DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(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')
|
19
|
+
|
20
|
+
_globals = globals()
|
21
|
+
_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
|
22
|
+
_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'mediapipe.util.tracking.box_tracker_pb2', _globals)
|
23
|
+
if _descriptor._USE_C_DESCRIPTORS == False:
|
24
|
+
_globals['DESCRIPTOR']._options = None
|
25
|
+
_globals['DESCRIPTOR']._serialized_options = b'\n\035com.google.mediapipe.trackingB\017BoxTrackerProto'
|
26
|
+
_globals['_BOXTRACKEROPTIONS']._serialized_start=97
|
27
|
+
_globals['_BOXTRACKEROPTIONS']._serialized_end=359
|
28
|
+
_globals['_TIMEDBOXPROTO']._serialized_start=362
|
29
|
+
_globals['_TIMEDBOXPROTO']._serialized_end=657
|
30
|
+
_globals['_TIMEDBOXPROTOLIST']._serialized_start=659
|
31
|
+
_globals['_TIMEDBOXPROTOLIST']._serialized_end=717
|
32
|
+
# @@protoc_insertion_point(module_scope)
|
@@ -0,0 +1,31 @@
|
|
1
|
+
# -*- coding: utf-8 -*-
|
2
|
+
# Generated by the protocol buffer compiler. DO NOT EDIT!
|
3
|
+
# source: mediapipe/util/tracking/camera_motion.proto
|
4
|
+
# Protobuf Python Version: 4.25.1
|
5
|
+
"""Generated protocol buffer code."""
|
6
|
+
from google.protobuf import descriptor as _descriptor
|
7
|
+
from google.protobuf import descriptor_pool as _descriptor_pool
|
8
|
+
from google.protobuf import symbol_database as _symbol_database
|
9
|
+
from google.protobuf.internal import builder as _builder
|
10
|
+
# @@protoc_insertion_point(imports)
|
11
|
+
|
12
|
+
_sym_db = _symbol_database.Default()
|
13
|
+
|
14
|
+
|
15
|
+
from mediapipe.util.tracking import motion_models_pb2 as mediapipe_dot_util_dot_tracking_dot_motion__models__pb2
|
16
|
+
|
17
|
+
|
18
|
+
DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(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')
|
19
|
+
|
20
|
+
_globals = globals()
|
21
|
+
_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
|
22
|
+
_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'mediapipe.util.tracking.camera_motion_pb2', _globals)
|
23
|
+
if _descriptor._USE_C_DESCRIPTORS == False:
|
24
|
+
DESCRIPTOR._options = None
|
25
|
+
_globals['_CAMERAMOTION']._serialized_start=104
|
26
|
+
_globals['_CAMERAMOTION']._serialized_end=1518
|
27
|
+
_globals['_CAMERAMOTION_TYPE']._serialized_start=1232
|
28
|
+
_globals['_CAMERAMOTION_TYPE']._serialized_end=1314
|
29
|
+
_globals['_CAMERAMOTION_FLAGS']._serialized_start=1317
|
30
|
+
_globals['_CAMERAMOTION_FLAGS']._serialized_end=1512
|
31
|
+
# @@protoc_insertion_point(module_scope)
|
@@ -0,0 +1,60 @@
|
|
1
|
+
# -*- coding: utf-8 -*-
|
2
|
+
# Generated by the protocol buffer compiler. DO NOT EDIT!
|
3
|
+
# source: mediapipe/util/tracking/flow_packager.proto
|
4
|
+
# Protobuf Python Version: 4.25.1
|
5
|
+
"""Generated protocol buffer code."""
|
6
|
+
from google.protobuf import descriptor as _descriptor
|
7
|
+
from google.protobuf import descriptor_pool as _descriptor_pool
|
8
|
+
from google.protobuf import symbol_database as _symbol_database
|
9
|
+
from google.protobuf.internal import builder as _builder
|
10
|
+
# @@protoc_insertion_point(imports)
|
11
|
+
|
12
|
+
_sym_db = _symbol_database.Default()
|
13
|
+
|
14
|
+
|
15
|
+
from mediapipe.util.tracking import motion_models_pb2 as mediapipe_dot_util_dot_tracking_dot_motion__models__pb2
|
16
|
+
from mediapipe.util.tracking import region_flow_pb2 as mediapipe_dot_util_dot_tracking_dot_region__flow__pb2
|
17
|
+
|
18
|
+
|
19
|
+
DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(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?')
|
20
|
+
|
21
|
+
_globals = globals()
|
22
|
+
_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
|
23
|
+
_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'mediapipe.util.tracking.flow_packager_pb2', _globals)
|
24
|
+
if _descriptor._USE_C_DESCRIPTORS == False:
|
25
|
+
DESCRIPTOR._options = None
|
26
|
+
_globals['_TRACKINGDATA_MOTIONDATA'].fields_by_name['vector_data']._options = None
|
27
|
+
_globals['_TRACKINGDATA_MOTIONDATA'].fields_by_name['vector_data']._serialized_options = b'\020\001'
|
28
|
+
_globals['_TRACKINGDATA_MOTIONDATA'].fields_by_name['track_id']._options = None
|
29
|
+
_globals['_TRACKINGDATA_MOTIONDATA'].fields_by_name['track_id']._serialized_options = b'\020\001'
|
30
|
+
_globals['_TRACKINGDATA_MOTIONDATA'].fields_by_name['row_indices']._options = None
|
31
|
+
_globals['_TRACKINGDATA_MOTIONDATA'].fields_by_name['row_indices']._serialized_options = b'\020\001'
|
32
|
+
_globals['_TRACKINGDATA_MOTIONDATA'].fields_by_name['col_starts']._options = None
|
33
|
+
_globals['_TRACKINGDATA_MOTIONDATA'].fields_by_name['col_starts']._serialized_options = b'\020\001'
|
34
|
+
_globals['_TRACKINGDATA']._serialized_start=147
|
35
|
+
_globals['_TRACKINGDATA']._serialized_end=836
|
36
|
+
_globals['_TRACKINGDATA_MOTIONDATA']._serialized_start=428
|
37
|
+
_globals['_TRACKINGDATA_MOTIONDATA']._serialized_end=663
|
38
|
+
_globals['_TRACKINGDATA_FRAMEFLAGS']._serialized_start=666
|
39
|
+
_globals['_TRACKINGDATA_FRAMEFLAGS']._serialized_end=836
|
40
|
+
_globals['_TRACKINGDATACHUNK']._serialized_start=839
|
41
|
+
_globals['_TRACKINGDATACHUNK']._serialized_end=1090
|
42
|
+
_globals['_TRACKINGDATACHUNK_ITEM']._serialized_start=964
|
43
|
+
_globals['_TRACKINGDATACHUNK_ITEM']._serialized_end=1090
|
44
|
+
_globals['_BINARYTRACKINGDATA']._serialized_start=1092
|
45
|
+
_globals['_BINARYTRACKINGDATA']._serialized_end=1126
|
46
|
+
_globals['_METADATA']._serialized_start=1129
|
47
|
+
_globals['_METADATA']._serialized_end=1267
|
48
|
+
_globals['_METADATA_TRACKOFFSET']._serialized_start=1217
|
49
|
+
_globals['_METADATA_TRACKOFFSET']._serialized_end=1267
|
50
|
+
_globals['_TRACKINGCONTAINER']._serialized_start=1269
|
51
|
+
_globals['_TRACKINGCONTAINER']._serialized_end=1352
|
52
|
+
_globals['_TRACKINGCONTAINERFORMAT']._serialized_start=1355
|
53
|
+
_globals['_TRACKINGCONTAINERFORMAT']._serialized_end=1528
|
54
|
+
_globals['_TRACKINGCONTAINERPROTO']._serialized_start=1530
|
55
|
+
_globals['_TRACKINGCONTAINERPROTO']._serialized_end=1645
|
56
|
+
_globals['_FLOWPACKAGEROPTIONS']._serialized_start=1648
|
57
|
+
_globals['_FLOWPACKAGEROPTIONS']._serialized_end=1969
|
58
|
+
_globals['_FLOWPACKAGEROPTIONS_HIGHPROFILEENCODING']._serialized_start=1888
|
59
|
+
_globals['_FLOWPACKAGEROPTIONS_HIGHPROFILEENCODING']._serialized_end=1969
|
60
|
+
# @@protoc_insertion_point(module_scope)
|
@@ -0,0 +1,35 @@
|
|
1
|
+
# -*- coding: utf-8 -*-
|
2
|
+
# Generated by the protocol buffer compiler. DO NOT EDIT!
|
3
|
+
# source: mediapipe/util/tracking/frame_selection.proto
|
4
|
+
# Protobuf Python Version: 4.25.1
|
5
|
+
"""Generated protocol buffer code."""
|
6
|
+
from google.protobuf import descriptor as _descriptor
|
7
|
+
from google.protobuf import descriptor_pool as _descriptor_pool
|
8
|
+
from google.protobuf import symbol_database as _symbol_database
|
9
|
+
from google.protobuf.internal import builder as _builder
|
10
|
+
# @@protoc_insertion_point(imports)
|
11
|
+
|
12
|
+
_sym_db = _symbol_database.Default()
|
13
|
+
|
14
|
+
|
15
|
+
from mediapipe.util.tracking import camera_motion_pb2 as mediapipe_dot_util_dot_tracking_dot_camera__motion__pb2
|
16
|
+
from mediapipe.util.tracking import frame_selection_solution_evaluator_pb2 as mediapipe_dot_util_dot_tracking_dot_frame__selection__solution__evaluator__pb2
|
17
|
+
from mediapipe.util.tracking import region_flow_pb2 as mediapipe_dot_util_dot_tracking_dot_region__flow__pb2
|
18
|
+
|
19
|
+
|
20
|
+
DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(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')
|
21
|
+
|
22
|
+
_globals = globals()
|
23
|
+
_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
|
24
|
+
_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'mediapipe.util.tracking.frame_selection_pb2', _globals)
|
25
|
+
if _descriptor._USE_C_DESCRIPTORS == False:
|
26
|
+
DESCRIPTOR._options = None
|
27
|
+
_globals['_FRAMESELECTIONTIMESTAMP']._serialized_start=214
|
28
|
+
_globals['_FRAMESELECTIONTIMESTAMP']._serialized_end=315
|
29
|
+
_globals['_FRAMESELECTIONRESULT']._serialized_start=318
|
30
|
+
_globals['_FRAMESELECTIONRESULT']._serialized_end=516
|
31
|
+
_globals['_FRAMESELECTIONCRITERION']._serialized_start=519
|
32
|
+
_globals['_FRAMESELECTIONCRITERION']._serialized_end=739
|
33
|
+
_globals['_FRAMESELECTIONOPTIONS']._serialized_start=741
|
34
|
+
_globals['_FRAMESELECTIONOPTIONS']._serialized_end=844
|
35
|
+
# @@protoc_insertion_point(module_scope)
|
@@ -0,0 +1,28 @@
|
|
1
|
+
# -*- coding: utf-8 -*-
|
2
|
+
# Generated by the protocol buffer compiler. DO NOT EDIT!
|
3
|
+
# source: mediapipe/util/tracking/frame_selection_solution_evaluator.proto
|
4
|
+
# Protobuf Python Version: 4.25.1
|
5
|
+
"""Generated protocol buffer code."""
|
6
|
+
from google.protobuf import descriptor as _descriptor
|
7
|
+
from google.protobuf import descriptor_pool as _descriptor_pool
|
8
|
+
from google.protobuf import symbol_database as _symbol_database
|
9
|
+
from google.protobuf.internal import builder as _builder
|
10
|
+
# @@protoc_insertion_point(imports)
|
11
|
+
|
12
|
+
_sym_db = _symbol_database.Default()
|
13
|
+
|
14
|
+
|
15
|
+
|
16
|
+
|
17
|
+
DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(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')
|
18
|
+
|
19
|
+
_globals = globals()
|
20
|
+
_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
|
21
|
+
_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'mediapipe.util.tracking.frame_selection_solution_evaluator_pb2', _globals)
|
22
|
+
if _descriptor._USE_C_DESCRIPTORS == False:
|
23
|
+
DESCRIPTOR._options = None
|
24
|
+
_globals['_FRAMESELECTIONSOLUTIONEVALUATOROPTIONS']._serialized_start=79
|
25
|
+
_globals['_FRAMESELECTIONSOLUTIONEVALUATOROPTIONS']._serialized_end=131
|
26
|
+
_globals['_FRAMESELECTIONSOLUTIONEVALUATORTYPE']._serialized_start=134
|
27
|
+
_globals['_FRAMESELECTIONSOLUTIONEVALUATORTYPE']._serialized_end=292
|
28
|
+
# @@protoc_insertion_point(module_scope)
|