Fi$fdZddlmZddlZddlmZddlmZddlmZddl m Z ddl Z ddl Z ddlZddlmZe j"Ze j&fdd Zej*Zd ad Zee_e j&fdd Zddd ZddZej6ZdZdZedddZedZ edddZ!y)zDMonkey patches to update/extend functionality of existing functions.) annotationsN)contextmanager)copy)Path)Any)Imagectj|tj}|jdrft j |tj \}}|r>> 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)filenameflags file_bytessuccessframesims [/var/www/pod-logistic/pod-ai/venv/lib/python3.12/site-packages/ultralytics/utils/patches.pyimreadr"sXrxx0J*+++J8L8LM  #F q 0VAY^^q5H6!9 fbhhW]deNf f \\*e , :(..*334FGXu-B "277a@Q@QR S S S s;B37B' B3AB' B3'B0,B30B33 B?>B?c tjt|j||dj |y#t $rYywxYw)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 r TF)rimencodersuffixtofiler))rr8paramss r!imwriter>lsF$ T(^**C8;BB8L s>> 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)winnamemats r!imshowrFs!" GNN+ , 3 3 5s;r#cNddlm}|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_utilsrHtorchload)r,r-rHs r! torch_loadrMs/ 9nF2!&~ ::t &v &&r#ctdD]} t|i|cSy#t$r,}|dk(r|tjd|zdz Yd}~Cd}~wwxYw)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") r r N)range _torch_save RuntimeErrortimesleep)r,r-ies r! torch_saverWsa1X# #// /# #Av JJ1z " " #s  A"AAc#K|r:|r8|dk(r3tjddfd }|t_dt_ydyw)zwWorkaround for ONNX torch.arange incompatibility with FP16. https://github.com/pytorch/pytorch/issues/148041. onnxN)dtypec2|i|j|S)zNWrap torch.arange to cast dtype after creation instead of passing it directly.)to)rZr,r-funcs r!arangezarange_patch..aranges((++E2 2r#)rKr^)dynamichalffmtr^r]s @r! arange_patchrbs; 4C6M|| $ 3    sAAc#Kddlm}|rNtjjfd}|tj_dtj_ydyw)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)r,r-r]s r! torch_exportz'onnx_export_patch..torch_exports666 6r#N)rJrdrKrYexport)rdrhr]s @r!onnx_export_patchrjsC8zz   7)     sAAc#"K|r[t|}|jD]\}}t||| ||jj |jy|y#|jj |jwxYww)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)r, overrides original_argskeyvalues r!override_configsrts|T #//+ &JC D#u % & 9J MM !7!7 8  MM !7!7 8s3BA%+B%'B  B)rstrrintreturnznp.ndarray | None)N)rrur8 np.ndarrayr=zlist[int] | Nonerwbool)rDrurErxrwNone)FF)r_ryr`ryraru)rpzdict[str, Any] | None)"__doc__ __future__rrS contextlibrrpathlibrtypingrrnumpyrrKPILrrFrA IMREAD_COLORr"r2r(r'r.rr>saverQrMrWrbrjrtrgr#r!rsJ" %   **(+'7'7H<jj 6> -0,<,<&2<*jj '0#.&$r#