Wpf, dZddlmZmZddlZddlZddlmZddl m Z ddl m Z ddl m Z ddl m Z e jZe jZe jZe jZe jZe jZe jZe jZe jZe jZe jZe jZe jZe jZe jZe jZe jZe jZe j Z e j!Z!e j"Z"e j#Z#e j$Z$ddd d ee j%ej&fd e j'd e(d e j)fdZ*ddd d ee j+ej&fd e j'd e(d e j)fdZ,dej-d e j)fdZ.deej-d e j)fdZ/dS)z&The public facing packet creator APIs.)ListUnionN)message)_packet_creator)image) image_frame)packet) image_formatcopydatar r returnct|tjrC||j|krt d||st dt j|S|t d||jjrdnd}|sB|jj st d|jjrtj dtd t j |||S) a# Create a MediaPipe ImageFrame packet. A MediaPipe ImageFrame packet can be created from an existing MediaPipe ImageFrame object and the data will be realigned and copied into a new ImageFrame object inside of the packet. A MediaPipe ImageFrame packet can also be created from the raw pixel data represented as a numpy array with one of the uint8, uint16, and float data types. There are three data ownership modes depending on how the 'copy' arg is set. i) Default mode If copy is not set, mutable data is always copied while the immutable data is by reference. ii) Copy mode (safe) If copy is set to True, the data will be realigned and copied into an ImageFrame object inside of the packet regardless the immutablity of the original data. iii) Reference mode (dangerous) If copy is set to False, the data will be forced to be shared. If the data is mutable (data.flags.writeable is True), a warning will be raised. Args: data: A MediaPipe ImageFrame object or the raw pixel data that is represnted as a numpy ndarray. image_format: One of the image_frame.ImageFormat enum types. copy: Indicate if the packet should copy the data from the numpy nparray. Returns: A MediaPipe ImageFrame Packet. Raises: ValueError: i) When "data" is a numpy ndarray, "image_format" is not provided or the "data" array is not c_contiguous in the reference mode. ii) When "data" is an ImageFrame object, the "image_format" arg doesn't match the image format of the "data" ImageFrame object or "copy" is explicitly set to False. TypeError: If "image format" doesn't match "data" array's data type. Examples: np_array = np.random.randint(255, size=(321, 123, 3), dtype=np.uint8) # Copy mode by default if the data array is writable. image_frame_packet = mp.packet_creator.create_image_frame( image_format=mp.ImageFormat.SRGB, data=np_array) # Make the array unwriteable to trigger the reference mode. np_array.flags.writeable = False image_frame_packet = mp.packet_creator.create_image_frame( image_format=mp.ImageFormat.SRGB, data=np_array) image_frame = mp.ImageFrame(image_format=mp.ImageFormat.SRGB, data=np_array) image_frame_packet = mp.packet_creator.create_image_frame(image_frame) NBThe provided image_format doesn't match the one from the data arg.zcCreating ImageFrame packet by taking a reference of another ImageFrame object is not supported yet.*Please provide 'image_format' with 'data'.TFdD " ""ct|tjrC||j|krt d||st dt j|S|t d||jjrdnd}|sB|jj st d|jjrtj dtd t j |||S) aCreate a MediaPipe Image packet. A MediaPipe Image packet can be created from an existing MediaPipe Image object and the data will be realigned and copied into a new Image object inside of the packet. A MediaPipe Image packet can also be created from the raw pixel data represented as a numpy array with one of the uint8, uint16, and float data types. There are three data ownership modes depending on how the 'copy' arg is set. i) Default mode If copy is not set, mutable data is always copied while the immutable data is by reference. ii) Copy mode (safe) If copy is set to True, the data will be realigned and copied into an Image object inside of the packet regardless the immutablity of the original data. iii) Reference mode (dangerous) If copy is set to False, the data will be forced to be shared. If the data is mutable (data.flags.writeable is True), a warning will be raised. Args: data: A MediaPipe Image object or the raw pixel data that is represnted as a numpy ndarray. image_format: One of the mp.ImageFormat enum types. copy: Indicate if the packet should copy the data from the numpy nparray. Returns: A MediaPipe Image Packet. Raises: ValueError: i) When "data" is a numpy ndarray, "image_format" is not provided or the "data" array is not c_contiguous in the reference mode. ii) When "data" is an Image object, the "image_format" arg doesn't match the image format of the "data" Image object or "copy" is explicitly set to False. TypeError: If "image format" doesn't match "data" array's data type. Examples: np_array = np.random.randint(255, size=(321, 123, 3), dtype=np.uint8) # Copy mode by default if the data array is writable. image_packet = mp.packet_creator.create_image( image_format=mp.ImageFormat.SRGB, data=np_array) # Make the array unwriteable to trigger the reference mode. np_array.flags.writeable = False image_packet = mp.packet_creator.create_image( image_format=mp.ImageFormat.SRGB, data=np_array) image = mp.Image(image_format=mp.ImageFormat.SRGB, data=np_array) image_packet = mp.packet_creator.create_image(image) NrzYCreating Image packet by taking a reference of another Image object is not supported yet.rTFrz^'data' is still writeable. Taking a reference of the data to create Image packet is dangerous.r)rrImager rr_create_image_from_imagerrrrrr_create_image_from_pixel_datars r create_imager&s!zek""!"D$5$E$E  O Q QQ   e    3D 9 99 G H HH |Z) 4TTud  Z $N LNN N   n A     8dD " ""r! proto_messagecdtj|jj|S)aCreate a MediaPipe protobuf message packet. Args: proto_message: A Python protobuf message. Returns: A MediaPipe protobuf message Packet. Raises: RuntimeError: If the protobuf message type is not registered in MediaPipe. Examples: detection = detection_pb2.Detection() text_format.Parse('score: 0.5', detection) packet = mp.packet_creator.create_proto(detection) output_detection = mp.packet_getter.get_proto(packet) )r _create_proto DESCRIPTOR full_nameSerializeToString)r's r create_protor-s3&  &}'?'I'4'F'F'H'H J JJr! message_listc td)Nz'create_proto_vector is not implemented.)NotImplementedError)r.s rcreate_proto_vectorr1sEFFFr!)0__doc__typingrrrnumpynpgoogle.protobufr$mediapipe.python._framework_bindingsrrrr create_string create_bool create_int create_int8 create_int16 create_int32 create_int64 create_uint8 create_uint16 create_uint32 create_uint64 create_float create_doublecreate_int_arraycreate_float_arraycreate_int_vectorcreate_bool_vectorcreate_float_vectorcreate_string_vectorcreate_image_vectorcreate_packet_vectorcreate_string_to_packet_map create_matrixrndarray ImageFormatboolPacketr r#r&Messager-r1r!rrUs-,######@@@@@@666666<<<<<<777777 - )  ' ) + + + + - - - + - "3$7#5$7%9&;%9&;-I- @D$(^"^"^"U;#92:#EF^"%0%<^""^".4]^"^"^"^"H:>"^"^"^"uU["*45^"*6^"^"(.}^"^"^"^"DJJFMJJJJ0Gd7?&;G GGGGGGr!