/j$DdZddlmZddlZddlmZddlmZddlmZddl m Z ddl Z ddl Z ddlZddlmZe jZe jfd$dZejZdadZee_e jfd$dZd%d&dZd'dZejZdZdZedZed Z ed%d(d#Z!dS))zDMonkey patches to update/extend functionality of existing functions.) annotationsN)contextmanager)copy)Path)Any)Imagefilenamestrflagsintreturnnp.ndarray | Nonectj|tj}|drht j|tj\}}|rBt|dkr|djdkr|dntj |dSdSt j ||}|7| drt||}||jdkr|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) )z.tiffz.tifr)axisN)z.avifz.heic).N) npfromfileuint8endswithcv2 imdecodemultiIMREAD_UNCHANGEDlenndimstackimdecodelower _imread_pil)r r file_bytessuccessframesims ^/home/longshao/multi-rider-rag/.venv/lib/python3.11/site-packages/ultralytics/utils/patches.pyimreadr&sXrx00J*++ 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%imwriterEls^$ 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)rFrGs r%imshowrNs3" GNN+ , , 3 3 5 5s;;;;;r'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_utilsrPtorchload)r0r1rPs r% torch_loadrUsG 988888'nF22!&~ :t &v & &&r'ctdD]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)r0r1ies r% torch_saver_s1XX## #/// / / / # # #Avv J1z " " " " " " " " ###s ! A"AAc#K|jrD|jr=|jdkr2tjddfd }|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)rbr0r1funcs r%arangezarange_patch..aranges&4(((++E22 2r')dynamichalfformatrSrf)r0rfres @r% arange_patchrjs  |  dkV&;&;| $ 3 3 3 3 3 3 3    r'c#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)r0r1res r% torch_exportz'onnx_export_patch..torch_exports#4666666 6r'N)rRrlrSraexport)rlrpres @r%onnx_export_patchrrsx877777 z  7 7 7 7 7)     r' 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)r0rs original_argskeyvalues r%override_configsr}s 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 r@)rFr rGr=r rH)rsrt)"__doc__ __future__rr[ contextlibrrpathlibrtypingrrnumpyrrSPILrrNrK IMREAD_COLORr&r6r,r+r2r rEsaverYrUr_rjrrr}ror'r%rsJJ"""""" %%%%%%   *(+'7HHHHH<j 666> -0,<&2<<<<*j '''0###.&$r'