sg- BddlZddlmZddlmZmZmZmZmZddl m Z ddl m Z m Z e rddlmZe rddlZed Gd d ZGd d e ZdeeefdededeeeffdZddddeeddfdZdeedeefdZdeddeefdZy)N) dataclass)AnyIterableListOptionalTuple)BaseImageProcessor)is_torch_availableis_torchvision_available)ComposeT)frozencleZdZUdZdZeed<dZeed<dZeed<dZ eed<dZ eed<dZ eed<d Z y) SizeDictz> Hashable dictionary to store image size information. Nheightwidth longest_edge shortest_edge max_height max_widthcPt||r t||Std|d)NzKey z not found in SizeDict.)hasattrgetattrKeyError)selfkeys [/var/www/html/venv/lib/python3.12/site-packages/transformers/image_processing_utils_fast.py __getitem__zSizeDict.__getitem__,s. 4 4% %cU"9:;;) __name__ __module__ __qualname____doc__rint__annotations__rrrrrrrrrrsJFCE3L#M3JIs output_size: (25, 50) - input_size: (100, 200), max_height: 200, max_width: 500 -> output_size: (200, 400) Args: image_size (`Tuple[int, int]`): The image to resize. max_height (`int`): The maximum allowed height. max_width (`int`): The maximum allowed width. )minr$) rDrrrr height_scale width_scale min_scale new_height new_widths r#get_image_size_for_max_height_widthrLKsV,MFE&Le#KL+.IVi'(JEI%&I y  rtensorz torch.Tensoraxiscn||jS |j|S#t$r|cYSwxYw)zF Squeezes a tensor, but only if the axis specified has dim 1. )rN)squeezer3)rMrNs r safe_squeezerQjs@ |~~~~4~((  s & 44valuescJt|Dcgc] }t|c}Scc}w)zO Return the maximum value across all indices of an iterable of values. )zipmax)rRvalues_is rmax_across_indicesrWws +.v, 7hCM 77 7s imagescbt|Dcgc]}|jc}\}}}||fScc}w)zH Get the maximum height and width across all images in a batch. )rWshape)rXimg_rrs rget_max_height_widthr]~s5 22O3992OPAz9  ""3Ps,)N)rA dataclassesrtypingrrrrrimage_processing_utilsr utils.import_utilsr r torchvision.transformsr torchrr(r$rLrQrWr]r&rrrds !776L. $<<<$/2!c3h!!! 38_ !>  x}  8x}8c8#n!5#%*#r