o ?߱i{@sUddlZddlmZmZdddddddd d d d dd ddddddddddddddddddddddddd d!d"d#dd$d%d&d'd(dd)Zeeeeeeefffe d*<dddddddd+d,d-d.dd dddddddddddddddddddddddd d d!d"d#ddd d!d"d#dd$d%d&d'd(dd/ Z eeeeeffe d0<d1e ed2e efd3d4Z d1e ed2eeeeffd5d6Z dS)7N)ListTuple)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)10801024720512480480p 720robocasa256IMAGE_RES_SIZE_INFO)rr)rr)rr )r r) rrrrrr720prr VIDEO_RES_SIZE_INFOwdinfosreturncCs.g}|D]}td|}||dq|S)Nzaspect_ratio_(\d+_\d+))researchappendgroup)r$ aspect_ratioswdinfoZaspect_ratio_matchr-V/data/cameron/vidgen/cosmos-predict2.5/cosmos_predict2/_src/predict2/datasets/utils.pyget_aspect_ratios_from_wdinfosFs  r/cCst|}ddt||DS)NcSs g|] \}}||ddfqS)_,)replace).0r, aspect_ratior-r-r. Ss z.get_wdinfos_w_aspect_ratio..)r/zip)r$r+r-r-r.get_wdinfos_w_aspect_ratioOsr7)r'typingrrr!dictstrtupleint__annotations__r#listr/r7r-r-r-r.sV &&