o
    ߥi=                     @   s	  d dl Zd dlmZ d dlmZ G dd deZG dd deZej	e
ejejfeje
eje
eejfeje
ejdfejeejfejeedfejeedfejeiZej	d	d
dejd	ddejd	ddejd	ddejddddidejddiejddiejddiiZdd Zi ejej	ejdejej	ej	ejejiejej	ejej	ej	ejejiejej	ej ej	ej!ej	ejejej"ej	ej#ejej$ejej%ej	ej&ej	ej'ej	ej(ej	ej)ej	ej*ej	i ej+ej	ej,ej	ej-ej	ej.ej	ej/ej	ej0ej	ej1ej	ej2ej	ej3ej	ej4ej	ej5ejej6ej	ej7ejej8ej	ej9ej	ej:ej	ej;ej	i ej<ej	ej=ej	ej>ej	ej?ej	ej@ej	ejAej	ejBej	ejCej	ejDej	ejEej	ejFej	ejdejGej	ejHej	ejIej	ejJej	ejKej	ejLej	i ejMej	ejNej	ejOej	ejPej	ejQej	ejdejRej	ej	dejSej	ej	ej	dejTej	ej	dejUej	ejdejVej	ej	dejWej	ejXejejYej	ej	ej	ej	ej	fgejZej	ej[ej	ej	dej\ej	ej]eji ej^ejej_ej	ej`ejejaej	ejbej	ejcejejfejdejejeejejfej	ejgejejhej	ej	ej	fej	ej	ej	d gejiej	ej	ejejiejjej	ejkejejlej	ejmej	ej	ej	ej	ej	ej	iejnej	ej	d!ejii ejoejejd"ejpejejejejd#ejqej	ej	d$ejiejrej	ej	d%ejiejsd&ejiejtejejejfejejd'gejuejejfejejd(gejvejejfejwejejxejejyejejzejej{ejd)ejigej|ejej}ejd)ejigej~ejejeji ejejejejejd*ejejejfejejd*gejejejejejejejd+ejejejejejd,ejejejd-ejejejejd.ejejejejejd/ejejejejd0ejd1ejiejejejejd2ejejejejejejejejd3gi ejejejejejd4ejejejejejejejejejejejejejejejejejejejejejejejgejej	d5ej	igejejd6ejigejd&ejiejej	ejdi ejd)ejiejej	ejd7ejej	ejdejej	ejd7ejejejd8ejej	ejdejejejd8ejej	ejejd9ejejejej	ejejejejejejejd:ejej	ejejejejd;ejd)ejiejejejd<ejiejej	ejd=ZdS )>    N)Image)Tasksc                   @   s   e Zd ZdZdZdZdS )	InputKeysimagetextvideoN)__name__
__module____qualname__IMAGETEXTVIDEO r   r   N/home/ubuntu/.local/lib/python3.10/site-packages/modelscope/pipeline_inputs.pyr   	   s    r   c                   @   s,   e Zd ZdZdZdZdZdZdZdZ	dZ
d	S )
	InputTyper   r   audior   boxdictlistnumberN)r   r	   r
   r   r   AUDIOr   BOXDICTLISTNUMBERr   r   r   r   r      s    r   zcv2.VideoCapturestringz(Base64 encoded image file or url string.)typedescriptionz)Base64 encoded audio file or url string..z)Base64 encoded video file or url string..zThe input text.arrayzBox coordinate, should be int.r   r   )r   r   itemsobjectintegerc              	   C   s   t |  }| tjkr+t|jdks't||s)J d|  d| dt| d| d S d S t||sBJ d|  d| dt| d| d S )NVideoCapturezinvalid input type for z, expected z	 but got z
 )
INPUT_TYPEr   r   r   r   
isinstance)
input_typeinputexpected_typer   r   r   check_input_typeI   s   
r(   )r   r   )imgcategory_names)r   prompt)r)   mask)r)   r,   	reference)	sky_imagescene_image)templateuser)contentstyle)masked_modelposeclothtarget_pose_path)
dataset_idcase_id)r8   r9   action_datasetactiontarget_viewsave_dirmessages)r   text2)source_texttarget_textr   )source_sentencesentences_to_compare)history	knowledgebot_profileuser_profile)questionhistory_sql)	query_setsupport_set)hypsrcref)r   database)querycontextlabeldataset)rP   positivenegative)wavr   )nearend_micfarend_speechr   r   )r)   r   )r   r   )r   r   r?   )	mesh_pathtexture_pathr+   uvsize
image_size
output_dir)video_input_pathvideo_output_path	mask_pathr+   )r)   prompts)numpynpPILr   modelscope.utils.constantr   r    r   r   r   strndarrayr   r   bytesr   r   r   r   r   r   r   r   intr#   INPUT_TYPE_SCHEMAr(   task_templateimage_text_retrievalgeneral_recognitionvideo_depth_estimationindoor_layout_estimationimage_demoireingpanorama_depth_estimationanimal_recognitionmotion_generationvideo_panoptic_segmentationocr_detectionocr_recognitionface_2d_keypointsface_livenessface_quality_assessmentcard_detectionlicense_plate_detectioncard_detection_correctionlineless_table_recognitiontable_recognitionface_detectionfacial_expression_recognitionface_attribute_recognitionface_recognitionface_reconstructionhead_reconstructiontext_to_headhuman_detectionface_image_generationimage_classificationimage_quality_assessment_mos$image_quality_assessment_degradationimage_object_detection domain_specific_object_detectionhuman_wholebody_keypointimage_segmentationportrait_mattinguniversal_mattingproduct_segmentationsemantic_segmentationface_human_hand_detectionhand_staticimage_fewshot_detectionopen_vocabulary_detectionimage_driving_perceptionvision_efficient_tuningskin_retouchingimage_super_resolutionimage_colorizationimage_color_enhancementimage_denoisingimage_body_reshapingimage_portrait_enhancementcrowd_countingimage_super_resolution_pasdimage_inpaintingimage_paintbyexampleimage_skychangecontrollable_image_generationimage_face_fusionimage_deblurringvideo_colorizationimage_to_image_generationimage_to_image_translationimage_style_transferimage_portrait_stylizationlive_categoryaction_recognitionbody_2d_keypointsbody_3d_keypointshand_2d_keypoints pedestrian_attribute_recognitionvideo_single_object_trackingvideo_multi_object_trackingvideo_categoryproduct_retrieval_embeddingvideo_embeddingvirtual_try_ontext_driven_segmentationshop_segmentationmovie_scene_segmentationbad_image_detectingimage_try_onhuman_image_generationhuman3d_renderhuman3d_animationimage_view_transformimage_control_3d_portraitchattext_classificationsentence_similaritynlisentiment_classificationzero_shot_classificationrelation_extractiontranslationtext_summarizationcompetency_aware_translationword_segmentationpart_of_speechnamed_entity_recognitiontext_error_correctionsentence_embeddingtext_rankingtext_generationfid_dialogue	fill_masktable_question_answeringfaq_question_answeringtranslation_evaluationsudokutext2sql!document_grounded_dialog_generatedocument_grounded_dialog_rerank"document_grounded_dialog_retrievalmachine_reading_comprehensionsiamese_uieauto_speech_recognitionspeech_signal_processacoustic_echo_cancellationspeech_separationacoustic_noise_suppressiontext_to_speechkeyword_spotting&speaker_diarization_dialogue_detectionlanguage_score_predictionpunctuationspeech_language_recognition3speaker_diarization_semantic_speaker_turn_detectioninverse_text_processingspeaker_verificationimage_captioningvideo_captioningmultimodal_dialoguevisual_groundingtext_to_image_synthesismulti_modal_embedding generative_multi_modal_embeddingmulti_modal_similaritytext_video_retrievalvisual_question_answeringvideo_question_answeringvisual_entailmentaction_detectionhuman_reconstructiontext_texture_generationimage_reid_personvideo_inpaintingtext_to_video_synthesisvideo_summarizationtext_to_360panorama_imageimage_editingTASK_INPUTSr   r   r   r   <module>   s  $!#%'+-/13579;=?ACEGIKMOQSUWY[]_acegikmqswy{}                    !  #  '  +  -  1  3  5  7  9  ;  =  ?  @  B  D  F  H  P  T  V  X  Z  _  c  g  m  q  w   	
    
                        !    $    &    )    +    -    /    3
    :    <    B    D    H    L    Q    S    W    \    _    d    g    j    o    q    u    w    y    {    }                                  
                                          !      %      )      -      1      5      :      <      >      F      H      M      P      Q      T   
