o W h@@sdZddlZddlmZddlZddlZddlZejZ ej fde de fddZ dde dejfd d Zd e d ejfd dZejZejZddZddZdS)zDMonkey patches to update/extend functionality of existing functions.N)PathfilenameflagscCsnt|tj}|dr1t|tj\}}|r/t|dkr(|djdkr(|dStj |ddSdSt ||S)av Read an image from a file. Args: filename (str): Path to the file to read. flags (int): Flag that can take values of cv2.IMREAD_*. Controls how the image is read. Returns: (np.ndarray): The read image. Examples: >>> img = imread("path/to/image.jpg") >>> img = imread("path/to/image.jpg", cv2.IMREAD_GRAYSCALE) )z.tiffz.tifr)axisN) npfromfileuint8endswithcv2 imdecodemultiIMREAD_UNCHANGEDlenndimstackimdecode)rr file_bytessuccessframesrM/var/www/vscode/kcb/lib/python3.10/site-packages/ultralytics/utils/patches.pyimreads 0 rimgcCs:ztt|j||d|WdStyYdSw)a% Write an image to a file. 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)r imencodersuffixtofile Exception)rrparamsrrrimwrite)s  r winnamematcCst|d|dS)a Display an image in the specified window. This function is a wrapper around OpenCV's imshow function that displays an image in a named window. It is particularly useful for visualizing images during development and debugging. 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)r!r"rrrimshowCsr'cOs.ddlm}|rd|vrd|d<t|i|S)aH Load 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 2.0 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_utilsr( _torch_load)argskwargsr(rrr torch_load]s  r.c OshtdD]-}z t|i|WSty1}z|dkr|td|dWYd}~qd}~wwdS)aJ Save 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. Returns: (Any): Result of torch.save operation if successful, None otherwise. Examples: >>> model = torch.nn.Linear(10, 1) >>> torch_save(model.state_dict(), "model.pt") rrN)range _torch_save RuntimeErrortimesleep)r,r-ierrr torch_savevs r7)N)__doc__r3pathlibrr numpyr torchr'r$ IMREAD_COLORstrintrndarrayr loadr+saver1r.r7rrrrs