
    vj                     "   d dl mZmZ d dlmZmZ d dlmZmZm	Z	 d dl
Zd dlZd dlmZ  G d de          Zi ej        eej        ej        ej        ee         ej        eej        eej        ee         ej        ej        ej        ej        ej        eej                 ej        eej        ej        ej        ej        ej         ej        ej!        eej                 ej"        eej                 ej#        ej        ej$        ei ej%        ej&        ej'        eej(        dej)        eej                 ej*        d	ej+        d
ej,        eej                 ej-        d
ej.        eej/        ej        ej0        ej        ej1        ej        ej2        ee         ej3        ej        ej4        eej5        eej6        ej        i ej7        eej                 ej8        ej        ej9        e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ejA        eejB        ee         ejC        eejD        eejE        eejF        eejG        eejH        eIejJ        eIejK        eeI         ejL        eeI         ejM        eej                 ejN        ej        ejO        ej        ejP        eejQ        eej                 i	ZRi ej        ddiej        dddidej        dddidej        ddiej        ddiej        dddidej        dddidej        dddidej        ddddiddej        ddiej        dddidej        dddidej         dddidej!        ddddiddej"        ddddiddej#        dddidej$        ddii ej%        dddidej'        ddiej(        dddej)        dddddej*        dddej+        dddej,        dddddej-        dddej.        ddiej/        dddidej0        dddidej1        dddidej2        dddidej3        dddidej4        ddiej5        ddiej6        dddidi ej7        ddddiddej8        dddidej9        ddiej:        dddidej;        dddidej<        dddidej?        dddidej@        ddiejA        ddiejB        dddidejC        ddiejD        ddiejE        ddiejF        ddiejG        ddiejH        ddiejJ        ddiejK        dddidejL        dddidejM        ddddiddejN        dddidejO        dddidejP        ddiejQ        ddddiddiZSi ejT        ej        ej(        ej3        gejU        ej%        gejV        ej%        gejW        ej%        ej        gejX        ej%        ej$        gejY        ej%        ej        ej)        ej        ej#        gejZ        ej$        gej[        ej$        gej\        ej$        gej]        ej0        ej3        gej^        ej        ej        ej        gej_        ej        ej        ej        gej`        ej        ej        ej        geja        ej        gejb        ej        ej         gejc        ej        ej        gejd        ej        ej        gi eje        ej        ej        gejf        ej        ej        gejg        ej(        ej        ej        ej        gejh        ej        ej        geji        ej0        gejj        ej        ej        ej        gejk        ej(        gejl        ej        ej        gejm        ej        ej        ej        gejn        ej        ej        ej        gejo        ej        ej        ej        gejp        ej        ej        ej        gejq        ej(        gejr        ej        ej        ej         gejs        ej        ej        ej         ej        ej@        gejt        ej         geju        ej(        gi ejv        ej(        gejw        ej(        gejx        ej(        gejy        ej        gejz        ej(        gej{        ej(        gej|        ej(        gej}        ej(        gej~        ej(        gej        ej(        gej        ej(        gej        ej(        gej        ej        ej(        gej        ej(        gej        ej(        gej        ej(        gej        ej(        gi ej        ej(        gej        ej(        gej        ej(        gej        ej(        gej        ej*        gej        ej*        gej        ej*        gej        ej'        gej        ej'        gej        ej'        gej        ej*        gej        ej'        gej        ej(        gej        ej        ej        gej        ej        ej        gej        ej        gej        ej        ej        ej        gi ej        ej        ejG        ej*        gej        ej        ej        gej        ej'        gej        ej'        gej        ej'        gej        ej'        gej        ej'        gej        ej'        gej        ej'        gej        ej        ej        gej        ej        ejG        gej        ej        ej        ejG        gej        ej        ej        gej        ej        ej        gej        ej0        gej        ej:        gej        ej;        ej=        ej<        ej>        ej?        gi ej        ej*        gej        ej(        gej        ej         gej        ej         gej        ejH        ejL        ejJ        ejK        gej        ej        ej        gej        ej'        gej        ej         ejG        ej*        gej        ej         ej*        gej        ej        ej        gej        ej        ej        gej        ej        ej        gej        ej        ej        gej        ej        ej        gej        ej2        gej        ej4        gej        ej'        gi ej        ej'        gej        ej'        gej        ej'        gej        ej'        gej        ej3        ej        gej        ej        gej        ej$        gej        ej5        ejE        gej        ej$        gej        ej$        gej        ej$        gej        ej$        gej        ej3        gej        ej'        gej        ej'        gej        ej$        gej        ej$        gi ej        ej        gej        ej        ej        gej        ej$        gej        ej$        gej        ej        ej$        ej6        gej        ej$        gej        ej$        gej        ej+        gej        ej+        gej        ej+        gej        ej,        gej        ej-        gej        ejB        gej        ej        gej        ej        gejZ        ej$        gej        ej        ej        gi ej        ej)        gej        ej-        gej        ej0        ej3        gej        ej0        ej3        ej        gej        ej0        ej3        ej        gej        ej$        gej        ej$        gej        ej$        gej        ej$        gej        ej        ej        gej        ejG        ej        ej        ej        gej        ej'        gej        ej0        gej        ej'        gej        ej'        gej        ej        ej        ej        gej        ej'        ej        gi ej        ej         gej        ej(        gej        ej        gej        ej         gej        ej        ej*        gej        ej        ej        gej        ej        ej        gej        ej$        gej        ej'        gej        ej'        gej        ej        ejP        gej        ej*        gej        ej(        gej        ej(        gej        ej(        gej        ej)        gej        ejQ        gej        ej'        giZ G d de          ZdS )    )OrderedDict
namedtuple)	dataclassfields)DictListTupleN)Tasksc                      e Zd ZdZdZdZdZdZdZdZ	dZ
d	Zd
ZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZ dZ!d Z"d!Z#d"Z$d#Z%d$Z&d%Z'd&Z(d'Z)d(Z*d)Z+d*Z,d+Z-d,Z.d-Z/d.Z0d/Z1d0Z2d1Z3d2Z4d3Z5d4Z6d5Z7d6Z8d7Z9d8Z:d9Z;d:Z<d;Z=d<Z>d=Z?d>Z@d?ZAd@ZBdAS )B
OutputKeyslosslogitsscoresscorelabellabels	input_ids	label_posposescaptionboxes	keypointsmasksdepthsdepths_colorflowsflows_colornormalsnormals_colorlayouttextpolygonsoutput
output_imgoutput_imgsoutput_video
output_pcmoutput_pcm_list
output_wav
output_objoutput_meshimg_embeddingspk_embeddingspo_listtext_embeddingtranslationresponse
predictionpredictionsprobabilitiesdialog_statesvideo_embeddingphrase_prototypeobject_prototypesentence_prototypeevent_prototypetextvideo_simuuidwordkws_list
sql_string	sql_queryhistoryquery_result
timestampsshot_num	scene_numscene_meta_listshot_meta_listmatchespcd12pcd12_aligntboundsMViewsN)C__name__
__module____qualname__LOSSLOGITSSCORESSCORELABELLABELS	INPUT_IDS	LABEL_POSPOSESCAPTIONBOXES	KEYPOINTSMASKSDEPTHSDEPTHS_COLORFLOWSFLOWS_COLORNORMALSNORMALS_COLORLAYOUTTEXTPOLYGONSOUTPUT
OUTPUT_IMGOUTPUT_IMGSOUTPUT_VIDEO
OUTPUT_PCMOUTPUT_PCM_LIST
OUTPUT_WAV
OUTPUT_OBJOUTPUT_MESHIMG_EMBEDDINGSPK_EMBEDDINGSPO_LISTTEXT_EMBEDDINGTRANSLATIONRESPONSE
PREDICTIONPREDICTIONSPROBABILITIESDIALOG_STATESVIDEO_EMBEDDINGPHRASE_PROTOTYPEOBJECT_PROTOTYPESENTENCE_PROTOTYPEEVENT_PROTOTYPETEXTVIDEO_SIMUUIDWORDKWS_LIST
SQL_STRING	SQL_QUERYHISTORYQUERY_RESULT
TIMESTAMPSSHOT_NUM	SCENE_NUMSCENE_META_LISTSHOT_META_LISTMATCHESPCD12PCD12_ALIGNTBOUNDSMV_IMGS     j/lsinfo/ai/hellotax_ai/data_center/backend/venv/lib/python3.11/site-packages/modelscope/outputs/outputs.pyr   r      sQ       DFFEEFIIEGEIEF!LEKG#MFDHFJK!LJ'OJJK#M#MH%NKHJK#M#M'O))-'O#MDDHJIG!LJHI'O%NGEKGGGGr   r   imagebytespcmtypenumberarray)r   itemsstringobjectzThe base64 encoded image.)r   descriptionzThe base64 encoded video.zThe base64 encoded PCM.zThe base64 encoded WAV.integerc                   P     e Zd Zd Zd Z fdZ fdZ fdZd Zd Z	d Z
 xZS )	ModelOutputBasec                 <    |                                   d| _        d S )NT)reconstruct	post_initselfs    r   __post_init__zModelOutputBase.__post_init__z  s    r   c                     |                                   t          |                                           D ](\  }}|                     t	          | |                     )d S N)clear	enumeratekeysappendgetattr)r   idxkeys      r   r   zModelOutputBase.reconstruct~  s[    

!$))++.. 	, 	,HCKKc**++++	, 	,r   c                    t          |t                    r!t          | |          rt          | |          S n=t          |t          t
          f          r!t                                          |          S t          d| d          )Nz	No Index z found in the dataclass.)	
isinstancestrhasattrr   intslicesuper__getitem__
IndexError)r   item	__class__s     r   r   zModelOutputBase.__getitem__  s    dC   	-tT"" +tT***+sEl++ 	-77&&t,,,CTCCCDDDr   c                 n   t          |t                    r(|d t          |           D             v r||                                 vr8t	                                          ||           |                                  d S t          t          | |                    t          |          k    rkt	                                          ||           t	                      	                    |                                 
                    |          |           d S d S t	                                          ||           d S t          |t                    r`t	                      	                    ||           |                                 |         }t	                                          ||           d S d S )Nc                     g | ]	}|j         
S r   )name).0fs     r   
<listcomp>z/ModelOutputBase.__setitem__.<locals>.<listcomp>  s    444!qv444r   )r   r   r   r   r   __setattr__r   idr   __setitem__indexr   )r   r   valuekey_namer   s       r   r   zModelOutputBase.__setitem__  s   c3 	144vd||44444diikk))GG''U333$$&&&&&c**++r%yy88GG''U333GG''		(9(9#(>(>FFFFF 98 ##C/////S!! 	1GGU+++yy{{3'HGG%00000	1 	1r   c                     t          | dd          r|                     ||          S t                                          ||          S )Nr   F)r   r   r   r   )r   r   r   r   s      r   r   zModelOutputBase.__setattr__  sG    4e,, 	3##C///77&&sE222r   c                 :      fdt                     D             S )Nc                 H    g | ]}t          |j                  |j        S r   )r   r   )r   r   r   s     r   r   z(ModelOutputBase.keys.<locals>.<listcomp>  s3     
 
 
GD!&,A,A,MAF,M,M,Mr   )r   r   s   `r   r   zModelOutputBase.keys  s4    
 
 
 
"4LL
 
 
 	
r   c                 N    |                                                                  S r   )to_dictr   r   s    r   r   zModelOutputBase.items  s    ||~~##%%%r   c                 v    t                      }|                                 D ]}t          | |          ||<   |S r   )r   r   r   )r   r#   r   s      r   r   zModelOutputBase.to_dict  s<    99;; 	- 	-C!$,,F3KKr   )rM   rN   rO   r   r   r   r   r   r   r   r   __classcell__)r   s   @r   r   r   x  s          , , ,E E E E E1 1 1 1 1 3 3 3 3 3
 
 

& & &      r   r   )collectionsr   r   dataclassesr   r   typingr   r   r	   numpynptorchmodelscope.utils.constantr
   r   r   rP   floatrQ   ndarrayrR   rS   rT   r   rU   rV   rW   rX   rY   rZ   r[   r\   r]   r^   rc   rd   re   r   rf   rg   rh   ri   rj   rk   rl   rm   rn   ro   rp   rq   rr   rs   rt   ru   rv   rw   rx   ry   rz   r{   r|   r}   r~   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   OutputTypesOutputTypeSchematask_templateocr_detectiontable_recognitionlineless_table_recognitionlicense_plate_detectioncard_detection_correctionocr_recognitionsudokutext2sqldocument_vl_embeddingface_2d_keypointsface_detectioncard_detectioncontent_checkimage_driving_perceptionface_livenessface_quality_assessmentfacial_expression_recognitiongeneral_recognitionface_processing_baseface_attribute_recognitionface_recognitionhuman_detectionface_image_generationimage_classificationimage_object_detection domain_specific_object_detectionopen_vocabulary_detectionvideo_object_detectionobject_detection_3dimage_segmentationvideo_panoptic_segmentationsemantic_segmentationportrait_mattinguniversal_mattingimage_deblurringimage_face_fusionimage_quality_assessment_mosskin_retouchingimage_super_resolutionimage_super_resolution_pasdimage_colorizationimage_color_enhancementimage_denoisingimage_editingimage_portrait_enhancementcrowd_countingimage_inpaintingimage_paintbyexamplecontrollable_image_generationimage_to_image_generationimage_to_image_translationimage_style_transferimage_portrait_stylizationimage_body_reshapingvideo_frame_interpolationvideo_super_resolutionvideo_deinterlacenerf_recon_accnerf_recon_vq_compressionsurface_recon_commonvideo_colorizationimage_control_3d_portrait self_supervised_depth_completion$image_quality_assessment_degradationlive_categoryaction_recognitionbody_2d_keypointsbody_3d_keypoints pedestrian_attribute_recognitionface_reconstructionhuman3d_renderhuman3d_animationhead_reconstructiontext_to_headhuman_reconstructiontext_texture_generationhand_2d_keypointsvideo_single_object_trackingvideo_multi_object_trackingvideo_categoryproduct_retrieval_embeddingr6   text_video_retrievalvideo_stabilizationvirtual_try_ontext_driven_segmentationshop_segmentationmovie_scene_segmentationhuman_wholebody_keypointvideo_summarization#referring_video_object_segmentationvideo_human_mattingtext_classificationsentence_similaritynlisentiment_classificationzero_shot_classificationrelation_extractionr0   word_segmentationnamed_entity_recognitionpart_of_speechtext_error_correctionword_alignmentsentence_embeddingtext_rankingtext_generationchatfid_dialoguetext_summarizationtext2text_generation	fill_maskfeature_extractiontask_oriented_conversationtable_question_answeringauto_speech_recognitioninverse_text_processingspeaker_verification&speaker_diarization_dialogue_detectionspeech_language_recognitionpunctuation3speaker_diarization_semantic_speaker_turn_detectionlanguage_score_predictionspeech_timestampspeech_signal_processacoustic_echo_cancellationacoustic_noise_suppressionspeech_separationtext_to_speechkeyword_spottingimage_captioningvideo_captioningvisual_groundingtext_to_image_synthesismulti_modal_embedding generative_multi_modal_embeddingmulti_modal_similarityvisual_question_answeringvideo_question_answeringmultimodal_dialoguevisual_entailmentaction_detectionfaq_question_answeringimage_reid_personvideo_inpaintinghand_staticface_human_hand_detectionface_emotionproduct_segmentationimage_skychangetranslation_evaluationvideo_object_segmentationmotion_generationbad_image_detectingvision_efficient_tuning!document_grounded_dialog_generatedocument_grounded_dialog_rerank"document_grounded_dialog_retrievalvideo_temporal_groundingtext_to_video_synthesistext_to_360panorama_imageimage_try_onhuman_image_generationimage_view_transformimage_to_3dsiamese_uieTASK_OUTPUTSlistr   r   r   r   <module>r     sV   / / / / / / / / ) ) ) ) ) ) ) ) $ $ $ $ $ $ $ $ $ $      + + + + + +@ @ @ @ @ @ @ @F>OU>rz> tE{> e	>
 c> tCy> "*> "*> > > bj> "*> bj> tBJ'>  T"*-!>" rz#>$ OS%> >& '>( t)>* 7+>, D,->. W/>0 51>2 RZ 03>4 55>6 47>8 BJ9>: bj;>< bj=>> e?>@ rzA>B CC>D E>F 2:G> > >H D,I>J bjK>L fM>N 
O>P Q>R S>T !2:U>V 
W>X bjY>Z OS[>\ OS]>^ c_>` 3a>b #c>d e>f Tg>h 3i> >j #S	tCyRZ(bjBJRZ({> >@mOm H
 	m H
 m  !m& 'm, H
 -m8 H
 9mD H
 EmP 
 
 Qmb cmh H
 imt H
 um@ H
 AmL 
 
 Mm^ 
 
 _mp H
 qm| O}m mB H
 CmN OmT 2 Um\ 6
 
 ]mj 2 kmr 0 smz 4
 
! !{mH 0 ImP QmV H
 Wmb H
 cmn H
 omz H
 {mF H
   GmR SmX Ym^ H
 _m m mj 
 
 km| H
 }mH ImN H
! !OmZ H
" "[mf H
" "gmr H
 sm~ OmD OEmJ H
 KmV Wm\ ]mb cmh imn omt 	umz 	{m m@ I
! ! I
    
 
  H
  H
   
 
 I	m m ^	w	z,j.GHw 
*-.w 
j12w 
$z':J<L&Mw 
!J$7#Iw 
#Z.
0F:,&w. 
JO,/w0 
L:?#1w2 
NZ_%3w@ 
z89Cwj 
:+Z-=>mwP 

(**>?Swv 

(**>?ywD 
*+,Ewb 
"Z%5z7G$Hcwp 
*+Z-=>qw~ 
!J$5z7G#Hw wL 
'
)*OwP 

 1:3DEQw` 
z0*2B!awt 
$z'8*:K&Luw@ 
Z56AwZ 

):+;<]wh 
*"7!8iwv 
!2J4E FwwN 
 
):+;<QwR 
*
):+;<UwV 
#
):+;<YwJ 
 
):+;<MwX 

 56Ywl 

):+;<owd 
%:,j.>*/(ewv 
*"2!3wwF	 
Z23G	w w wH	 
j34I	wJ	 
Z23K	wL	 
j34M	wT	 
&)9(:U	w\	 
J12]	w^	 
 :#8"9_	w`	 
%
(='>a	wb	 
z45c	wd	 
!J$9#:e	wf	 
J12g	wh	 
*/0i	wj	 
$z'<&=k	wl	 
:,j.CDm	wn	 
Z23o	wp	 
!6 7q	wr	 
'**?)@s	wz	 
#j&;%<{	w w w|	 
$z'<&=}	w~	 
!6 7	w@
 
$z'<&=A
wB
 
!6 7C
wJ
 
#j&=%>K
wL
 
 :#:";M
wN
 
j56O
wP
 
:,-Q
wR
 
#j&7%8S
wT
 
!2 3U
wV
 
z67W
wX
 
#j&7%8Y
wZ
 
*Z-B,C[
wh
 
.:,1i
wz
 
*+Z->?{
wF 
z01Gwn 
j/1Aow w wT 
j3Z5LUw| 
**+-}wf 

 12gwh 
:,-iwj 
j/0kwL 

 12Mwp 
*+qwD 
!2 3EwT 
!J$5#6Uwr 
j2J4DEswH 
&*/)Iwz 
%*+Z-B({wL 
*+Z->?MwZ 
:,j.?@[wf 
%
(@'Agwr 
J67sw^ 
#Z%B#Z%? !_w w wn 

 78owz 
:01{wH 
"Z%5$6IwV 
j./WwD 
"Z6
8L"%Ewh 
"Z%9:;K$LiwH 

 12IwZ 
-*/1H0[wl 

 0*2IJmw~ 

 1:3DEwL 

 1:3DEMwZ 
I
!:#45[wh 
"Z%6
8I$Jiwv 
"Z%6
8I$JwwF 

 34GwR 

./Swd 
j/0ew w wx 
"Z%6$7ywz 
:,-{wF 
*"3!4GwP 
:,-QwR 
z8*:KLSwT 
*+Uw` 
JO,awn 
J$j&89owz 
){wF 
z/GwR 
 1Sw^ 
Ojo&_w~ 
z89w~ 
$z'8&9wL 
"Z%6$7MwV 
!JO#4Ww^ 
!JO#4_w w wf 
!2 3gwr 
0:,3swx 
%
'8yw@ 

(AwP 
=:?J,A@Qwn 
#jo%6owN 
Z_-OwZ 
*"7!8[w\ 
$z'<&=]w^ 
$z'<&=_w` 
j89awl 
:01mwF 
Z01GwV 
Z/0Wwb 
Z/0cwd 
JO,ewz 
Z-z/@A{w w wF  
!J$:#;G wR  
:01S w`  
 *";"a wt  
* *";Z=O-u wH! 
  *";Z=N#I!wT! 
#jo%6U!w\! 
"Z_$5]!wd! 

0e!wp! 
!JO#4q!w|! 
j/1BC}!wL" 
	M"wl" 
 :#4"5m"wx" 
j67y"wB# 
Z./C#wL# 

)*M#wX# 
#:+Z->&Y#wf# 
*J,<=g#w w wt# 
!1 2u#w@$ 
J12A$wH$ 
 :#3"4I$wT$ 
#j&6%7U$wb$ 
j2J4KLc$wn$ 

 1:3DEo$w|$ 
!J$5z7H#I}$w~$ 
+jo->$w@% 
)J,=+>A%wB% 
,z/@.AC%wD% 
"Z%6
8J$KE%wF% 
!J$;#<G%wH% 
#j&;%<I%wT% 
./U%w^% 
 :#8"9_%wh% 
!7 8i%wj% 

*+k%w wl% 

)*m%w wt%6 6 6 6 6d 6 6 6 6 6r   