o vžÄi{ã@sŒUddlZddlmZmZddddddœdd d d d dœd dddddœddddddœddddddœddddddœdd d!d"d#dœd$d%d&d'd(dœd)œZeeeeeeefffe d*<ddddddœdd+d,d-d.dœd dddddœddddddœddddddœddddddœd d d!d"d#dœdd d!d"d#dœd$d%d&d'd(dœd/œ Z eeeeeffe d0<d1e ed2e efd3d4„Z d1e ed2eeeeffd5d6„Z dS)7éN)ÚListÚTuple)ér)é é )rr)é€r)rr)z1,1z4,3z3,4z16,9z9,16)éép)r r)éPé)r r )éÀr )r éÀ)r r )ér )r r)ér)é€r)rr)ré€)rr)éàr)rr)rr)r é°)rr )rr)é@r)rr)éÐr)r r)rr )rr)rr)ér)é@r)rr)réÀ)rr)Ú1080Ú1024Ú720Ú512Ú480Ú480pÚ 720robocasaÚ256ÚIMAGE_RES_SIZE_INFO)rr)rr)rr )r r) rrrrrrZ720prr ÚVIDEO_RES_SIZE_INFOÚwdinfosÚreturncCs.g}|D]}t d|¡}| | d¡¡q|S)Nzaspect_ratio_(\d+_\d+)é)ÚreÚsearchÚappendÚgroup)r#Ú aspect_ratiosÚwdinfoZaspect_ratio_match©r,úP/data/cameron/vidgen/cosmos-policy/cosmos_policy/_src/predict2/datasets/utils.pyÚget_aspect_ratios_from_wdinfosFs  r.cCst|ƒ}dd„t||ƒDƒS)NcSs g|] \}}|| dd¡f‘qS)Ú_Ú,)Úreplace)Ú.0r+Ú aspect_ratior,r,r-Ú Ss z.get_wdinfos_w_aspect_ratio..)r.Úzip)r#r*r,r,r-Úget_wdinfos_w_aspect_ratioOsr6)r&Útypingrrr!ÚdictÚstrÚtupleÚintÚ__annotations__r"Úlistr.r6r,r,r,r-ÚsV ûû&ñûûð&