j0%HdZddlmZddlZddlmZddlmZddlmZddl m Z ddl Z ddl Z ddlZddlmZe jZe jfd(dZejZdadZee_e jfd(dZd)d*dZd+dZejZdZdZed,d-d#Zed$Z ed)d.d'Z!dS)/zDMonkey patches to update/extend functionality of existing functions.) annotationsN)contextmanager)copy)Path)Any)Imagefilenamestrflagsintreturnnp.ndarray | Nonec8 tj|tj}n#ttf$rYdSwxYw|drht j|t j\}}|rBt|dkr|dj dkr|dntj |dSdSt j ||}|7| drt||}||j dkr|d n|S) aRead an image from a file with multilanguage filename support. Args: filename (str): Path to the file to read. flags (int, optional): Flag that can take values of cv2.IMREAD_*. Controls how the image is read. Returns: (np.ndarray | None): The read image array, or None if reading fails. Examples: >>> img = imread("path/to/image.jpg") >>> img = imread("path/to/image.jpg", cv2.IMREAD_GRAYSCALE) N)z.tiffz.tifr)axis)z.avifz.heic).N)npfromfileuint8FileNotFoundErrorOSErrorendswithcv2 imdecodemultiIMREAD_UNCHANGEDlenndimstackimdecodelower _imread_pil)r r file_bytessuccessframesims \/var/www/html/Carbon-Document/venv/lib/python3.11/site-packages/ultralytics/utils/patches.pyimreadr(s[2844 w 'tt*++ H+J8LMM  g #F q 0 0VAY^q5H5H6!99bhW]deNfNfNf ft \*e , , :(..**334FGG:Xu--B "27a< >@QRR S S S S S S S S S S S S S S S S S S ttsAB77B* B7AB* B7*B..B71B.2B77 CCr= np.ndarrayparamslist[int] | Noneboolc tjt|j||d|dS#t $rYdSwxYw)aDWrite an image to a file with multilanguage filename support. Args: filename (str): Path to the file to write. img (np.ndarray): Image to write. params (list[int], optional): Additional parameters for image encoding. Returns: (bool): True if the file was written successfully, False otherwise. Examples: >>> import numpy as np >>> img = np.zeros((100, 100, 3), dtype=np.uint8) # Create a black image >>> success = imwrite("output.jpg", img) # Write image to file >>> print(success) True rTF)rimencodersuffixtofiler.)r r=r?s r'imwriterFos^$ T(^^*C88;BB8LLLt uusAA AAwinnamematNonecpt|d|dS)a8Display an image in the specified window with multilanguage window name support. This function is a wrapper around OpenCV's imshow function that displays an image in a named window. It handles multilanguage window names by encoding them properly for OpenCV compatibility. Args: winname (str): Name of the window where the image will be displayed. If a window with this name already exists, the image will be displayed in that window. mat (np.ndarray): Image to be shown. Should be a valid numpy array representing an image. Examples: >>> import numpy as np >>> img = np.zeros((300, 300, 3), dtype=np.uint8) # Create a black image >>> img[:100, :100] = [255, 0, 0] # Add a blue square >>> imshow("Example Window", img) # Display the image unicode_escapeN)_imshowencodedecode)rGrHs r'imshowrOs3" GNN+ , , 3 3 5 5s;;;;;cFddlm}|r d|vrd|d<tj|i|S)aDLoad a PyTorch model with updated arguments to avoid warnings. This function wraps torch.load and adds the 'weights_only' argument for PyTorch 1.13.0+ to prevent warnings. Args: *args (Any): Variable length argument list to pass to torch.load. **kwargs (Any): Arbitrary keyword arguments to pass to torch.load. Returns: (Any): The loaded PyTorch object. Notes: For PyTorch versions 1.13 and above, this function automatically sets `weights_only=False` if the argument is not provided, to avoid deprecation warnings. r) TORCH_1_13 weights_onlyF)ultralytics.utils.torch_utilsrRtorchload)r1r2rRs r' torch_loadrWsG 988888'nF22!&~ :t &v & &&rPctdD]I} t|i|cS#t$r,}|dkr|tjd|zdz Yd}~Bd}~wwxYwdS)aSave PyTorch objects with retry mechanism for robustness. This function wraps torch.save with 3 retries and exponential backoff in case of save failures, which can occur due to device flushing delays or antivirus scanning. Args: *args (Any): Positional arguments to pass to torch.save. **kwargs (Any): Keyword arguments to pass to torch.save. Examples: >>> model = torch.nn.Linear(10, 1) >>> torch_save(model.state_dict(), "model.pt") rrN)range _torch_save RuntimeErrortimesleep)r1r2ies r' torch_saveras1XX## #/// / / / # # #Avv J1z " " " " " " " " ###s ! A"AAdynamichalffmtc#K|r:|r8|dkr2tjddfd }|t_dVt_dSdVdS)zwWorkaround for ONNX torch.arange incompatibility with FP16. https://github.com/pytorch/pytorch/issues/148041. onnxN)dtypec:|i||S)zNWrap torch.arange to cast dtype after creation instead of passing it directly.)to)rhr1r2funcs r'arangezarange_patch..aranges&4(((++E22 2rP)rUrl)rcrdrerlrks @r' arange_patchrmsw  4 C6MM| $ 3 3 3 3 3 3 3    rPc#Kddlm}|r>tjjfd}|tj_dVtj_dSdVdS)zFWorkaround for ONNX export issues in PyTorch 2.9+ with Dynamo enabled.r) TORCH_2_9c|i|ddiS)zCExport model to ONNX format with Dynamo disabled for compatibility.dynamoF)r1r2rks r' torch_exportz'onnx_export_patch..torch_exports#4666666 6rPN)rTrorUrgexport)rorsrks @r'onnx_export_patchrusx877777 z  7 7 7 7 7)     rP overridesdict[str, Any] | Nonec#K|rt|}|D]\}}t||| |V|j|jdS#|j|jwxYw|VdS)aFContext manager to temporarily override configurations in args. Args: args (IterableSimpleNamespace): Original configuration arguments. overrides (dict[str, Any] | None): Dictionary of overrides to apply. Yields: (IterableSimpleNamespace): Configuration arguments with overrides applied. N)ritemssetattr__dict__update)r1rv original_argskeyvalues r'override_configsrs T #//++ & &JC D#u % % % % 9JJJ M !7 8 8 8 8 8DM !7 8 8 8 8 s A%%!B)r r r r r r)N)r r r=r>r?r@r rA)rGr rHr>r rI)FFrb)rcrArdrArer )rvrw)"__doc__ __future__rr] contextlibrrpathlibrtypingrrnumpyrrUPILrrOrL IMREAD_COLORr(r7r-r,r3r"rFsaver[rWrarmrurrrrPr'rsJJ"""""" %%%%%%   *(+'7HHHHHBj 666> -0,<&2<<<<*j '''0###.&$rP