Wpf dZddlmZddlZddlmZddlmZddlm Z ddlm Z ddl m Z dd l mZdd lmZd ZGd d eZdS)zMediaPipe Selfie Segmentation.) NamedTupleN)#constant_side_packet_calculator_pb2)image_to_tensor_calculator_pb2)inference_calculator_pb2)&tensors_to_segmentation_calculator_pb2)"local_file_contents_calculator_pb2)switch_container_pb2) SolutionBasezFmediapipe/modules/selfie_segmentation/selfie_segmentation_cpu.binarypbcBeZdZdZdfd Zdejdeffd ZxZ S)SelfieSegmentationzMediaPipe Selfie Segmentation. MediaPipe Selfie Segmentation processes an RGB image and returns a segmentation mask. Please refer to https://solutions.mediapipe.dev/selfie_segmentation#python-solution-api for usage examples. rc`ttd|idgdS)a(Initializes a MediaPipe Selfie Segmentation object. Args: model_selection: 0 or 1. 0 to select a general-purpose model, and 1 to select a model more optimized for landscape images. See details in https://solutions.mediapipe.dev/selfie_segmentation#model_selection. model_selectionsegmentation_mask)binary_graph_path side_inputsoutputsN)super__init___BINARYPB_FILE_PATH)selfr __class__s n/var/www/html/nettyfy-visnx/env/lib/python3.11/site-packages/mediapipe/python/solutions/selfie_segmentation.pyrzSelfieSegmentation.__init__.sE GG-  %% '''''imagereturncLtd|iS)aProcesses an RGB image and returns a segmentation mask. Args: image: An RGB image represented as a numpy ndarray. Raises: RuntimeError: If the underlying graph throws any error. ValueError: If the input image is not three channel RGB. Returns: A NamedTuple object with a "segmentation_mask" field that contains a float type 2d np array representing the mask. r) input_data)rprocess)rrrs rrzSelfieSegmentation.process=s" 77??w&6? 7 77r)r) __name__ __module__ __qualname____doc__rnpndarrayrr __classcell__)rs@rr r #su ' ' ' ' ' '82:8*8888888888rr )r"typingrnumpyr#mediapipe.calculators.corermediapipe.calculators.tensorrrrmediapipe.calculators.utilrmediapipe.framework.toolr mediapipe.python.solution_baser rr rrr.s%$KJJJJJGGGGGGAAAAAAOOOOOOIIIIII999999877777^)8)8)8)8)8)8)8)8)8)8r