Wpfb:dZddlZddlmZmZmZmZddlmZddlm Z ddl m Z ddl m Z ddlmZdd lmZdd lmZdd lmZdd lmZdd lmZddlmZddlmZddlm Z ddlm!Z"ddlm#Z#ej$Z%ej&Z'ej(Z)ej*Z+ej,Z-e#j.Z/e"j0Z1ej2Z3dZ4dZ5dZ6dZ7dZ8dZ9dZ:dZ;dZGdde j?Z@dS) z MediaPipe image classifier task.N)CallableMappingOptionalList)packet_creator) packet_getter)image)packet)classifications_pb2)classifier_options_pb2)"image_classifier_graph_options_pb2)classification_result)rect) base_options) task_info) doc_controls)base_vision_task_api)image_processing_options)vision_task_running_modeclassifications_outCLASSIFICATIONSimage_in image_outIMAGE norm_rect_in NORM_RECTz.packets_callbacks . / 8 8 : :$7$L$N$N!!**  !.1M"N O O%n5K&LMMe !78Bi   / /0K L L  /; ;r2:) task_graph input_streamsoutput_streams task_options)enable_flow_limitingN)rr9r Packet _TaskInfo_TASK_GRAPH_NAMEjoin _IMAGE_TAG_IMAGE_IN_STREAM_NAME_NORM_RECT_TAG_NORM_RECT_STREAM_NAME_CLASSIFICATIONS_TAGrSrNgenerate_graph_configrr* LIVE_STREAMr%)rGrHrZrs ` r1rFz#ImageClassifier.create_from_optionss $fm1C)D # HHj"78 9 9 HHn&<= > > HH*,HI J J HHj"89 : :    I 3''!(!5'"( (   #3=   r2Nr rc |||}|ttj|t tj|i}tj }| tj |tt|S)aPerforms image classification on the provided MediaPipe Image. Args: image: MediaPipe Image. image_processing_options: Options for image processing. Returns: A classification result object that contains a list of classifications. Raises: ValueError: If any of the input arguments is invalid. RuntimeError: If image classification failed to run. )convert_to_normalized_rect_process_image_datarfr create_imagerh create_protor)r rPrQrrRrSr<rV)r.r rnormalized_rectrKrYs r1classifyzImageClassifier.classifys$55 %O--~:5AA ;  " " $ $! ! /N#6"J"L"L((/K LMM ! 0 01L M MMr2 timestamp_msc  |||}|ttj||t zttj| |t zi}tj }| tj|tt |S)aPerforms image classification on the provided video frames. Only use this method when the ImageClassifier is created with the video running mode. It's required to provide the video frame's timestamp (in milliseconds) along with the video frame. The input timestamps should be monotonically increasing for adjacent calls of this method. Args: image: MediaPipe Image. timestamp_ms: The timestamp of the input video frame in milliseconds. image_processing_options: Options for image processing. Returns: A classification result object that contains a list of classifications. Raises: ValueError: If any of the input arguments is invalid. RuntimeError: If image classification failed to run. )rm_process_video_datarfrroatrXrhrpr)r rPrQrrRrSr<rV)r.r rsrrqrKrYs r1classify_for_videoz"ImageClassifier.classify_for_videos255 %O--~:5AADD 9 9    ;  " " $ $! ! "\: : ; ; /N#6"J"L"L((/K LMM ! 0 01L M MMr2c P|||}|ttj||t zttj| |t zidS)aSends live image data (an Image with a unique timestamp) to perform image classification. Only use this method when the ImageClassifier is created with the live stream running mode. The input timestamps should be monotonically increasing for adjacent calls of this method. This method will return immediately after the input image is accepted. The results will be available via the `result_callback` provided in the `ImageClassifierOptions`. The `classify_async` method is designed to process live stream data such as camera input. To lower the overall latency, image classifier may drop the input images if needed. In other words, it's not guaranteed to have output per input image. The `result_callback` provides: - A classification result object that contains a list of classifications. - The input image that the image classifier runs on. - The input timestamp in milliseconds. Args: image: MediaPipe Image. timestamp_ms: The timestamp of the input image in milliseconds. image_processing_options: Options for image processing. Raises: ValueError: If the current input timestamp is smaller than what the image classifier has already processed. N) rm_send_live_stream_datarfrrorvrXrhrpr))r.r rsrrqs r1classify_asynczImageClassifier.classify_async<s@55 %O ~:5AADD 9 9    ;  " " $ $! ! "\: : ; ; !r2rM)r3r4r5r6 classmethodr9rIrrFr=r>r_ImageProcessingOptionsr<rrr:rwrzr@r2r1rBrBysu!!F,c,6G,,,;,04*4444;4rEI!N!N  !N!))@ A!N !N!N!N!NNEI *N*N  *N*N!))@ A *N  *N*N*N*N`EI **  **!))@ A *  ******r2rB)Ar6 dataclassestypingrrrrmediapipe.pythonrr$mediapipe.python._framework_bindingsr r=r .mediapipe.tasks.cc.components.containers.protor .mediapipe.tasks.cc.components.processors.protor 0mediapipe.tasks.cc.vision.image_classifier.protor ,mediapipe.tasks.python.components.containersrclassification_result_modulermediapipe.tasks.python.corerbase_options_modulertask_info_module1mediapipe.tasks.python.core.optional_dependenciesr"mediapipe.tasks.python.vision.corerrimage_processing_options_modulerrPr<NormalizedRect_NormalizedRect BaseOptionsr7ImageClassifierGraphOptionsr-ClassifierOptionsr,VisionTaskRunningModer*ImageProcessingOptionsr|TaskInforbrSrirfrNrerhrgrcrX dataclassrBaseVisionTaskApirBr@r2r1rsF'&444444444444++++++******FFFFFF777777NNNNNNQQQQQQ______nnnnnn======KKKKKKEEEEEEJJJJJJCCCCCCjjjjjjGGGGGG4I%". &B"1B'= 9P  % 4("$  'B"& ::::::::zmmmmm*<mmmmmr2