j dddlmZddlZddlZddlmZddlmZddl m Z gdZ ej fdejdej d ejfd Zej fd e dej d ejfd Zej fd edej d ejfdZej fdejdej d ejfdZdejd eeeffdZdS))BytesION)Image) to_tensor) AbstractPath)tensor_from_pilread_img_as_tensordecode_img_as_tensortensor_from_numpy get_img_shapepil_imgdtypereturnc|tjkrt|}n/tt j|tjd|}|S)zConvert a PIL Image to a PyTorch tensor Args: pil_img: a PIL image dtype: the output tensor data type Returns: decoded image as tensor T)copy)torchfloat32rr nparrayuint8)r r imgs Y/var/www/html/Carbon-Document/venv/lib/python3.11/site-packages/doctr/io/image/pytorch.pyrrsG    "( F F FNN Jimg_pathc|tjtjtjfvrt dt j|d5}t|d|cdddS#1swxYwYdS)aRead an image file as a PyTorch tensor Args: img_path: location of the image file dtype: the desired data type of the output tensor. If it is float-related, values will be divided by 255. Returns: decoded image as a tensor insupported value for dtypermodeRGBN) rrfloat16r ValueErrorropenrconvert)rr r s rrr$s U[%-???6777 H3 ' ' '>7wu55u==>>>>>>>>>>>>>>>>>>s #A;;A?A? img_contentc*|tjtjtjfvrt dt jt|d5}t| d|cdddS#1swxYwYdS)aRead a byte stream as a PyTorch tensor Args: img_content: bytes of a decoded image dtype: the desired data type of the output tensor. If it is float-related, values will be divided by 255. Returns: decoded image as a tensor rrrrN) rrr rr!rr"rrr#)r$r r s rr r 5s U[%-???6777 GK((s 3 3 3>wwu55u==>>>>>>>>>>>>>>>>>>s#BB B npy_imgc|tjtjtjfvrt d|tjkrt |}n~tj|}|d}|tjkr3| tj d}|S)a7Read an image file as a PyTorch tensor Args: npy_img: image encoded as a numpy array of shape (H, W, C) in np.uint8 dtype: the desired data type of the output tensor. If it is float-related, values will be divided by 255. Returns: same image as a tensor of shape (C, H, W) r)r)r ) rrr rr!r from_numpypermute contiguoustodiv)r&r rs rr r Fs U[%-???6777    w''kk)$$//11 EM ! !&&u}&--11#66C Jrrc |jddS)zGet the shape of an imageN)shape)rs rr r `s 9RSS>r)iornumpyrrPILr!torchvision.transforms.functionalrdoctr.utils.common_typesr__all__rr Tensorrrbytesr ndarrayr tupleintr rrr?s  777777111111 q q q@E}U[QVQ]$EJM>>>ek>V[Vb>>>>"CH->>e>EK>TYT`>>>>"AF rz%+RWR^4u|c3hr