a yµÿfÆ ã@s’dZddlZddlmZddlZddlZddlZejZ ej fe e dœdd„Z de ejdœdd „Ze ejd œd d „ZejZejZd d„Zdd„ZdS)zDMonkey patches to update/extend functionality of existing functions.éN)ÚPath©ÚfilenameÚflagscCst t |tj¡|¡S)zý Read an image from a file. Args: filename (str): Path to the file to read. flags (int, optional): Flag that can take values of cv2.IMREAD_*. Defaults to cv2.IMREAD_COLOR. Returns: (np.ndarray): The read image. )Úcv2ZimdecodeÚnpÚfromfileZuint8r©r úU/var/www/html/django/DPS/env/lib/python3.9/site-packages/ultralytics/utils/patches.pyÚimreads r )rÚimgcCs>z$t t|ƒj||¡d |¡WdSty8YdS0dS)a2 Write an image to a file. Args: filename (str): Path to the file to write. img (np.ndarray): Image to write. params (list of ints, optional): Additional parameters. See OpenCV documentation. Returns: (bool): True if the file was written, False otherwise. éTFN)rZimencoderÚsuffixÚtofileÚ Exception)rr Úparamsr r r Úimwrites  r©ZwinnameÚmatcCst| d¡ ¡|ƒdS)z— Displays an image in the specified window. Args: winname (str): Name of the window. mat (np.ndarray): Image to be shown. Úunicode_escapeN)Ú_imshowÚencodeÚdecoderr r r Úimshow0srcOs.ddlm}|r d|vr d|d<t|i|¤ŽS)aG 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. Note: 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_13Z weights_onlyF)Zultralytics.utils.torch_utilsrÚ _torch_load)ÚargsÚkwargsrr r r Ú torch_load@s  rc OsjtdƒD]\}zt|i|¤ŽWStyb}z*|dkr<|‚t d|d¡WYd}~qd}~00qdS)a8 Optionally use dill to serialize lambda functions where pickle does not, adding robustness with 3 retries and exponential standoff in case of save failure. Args: *args (tuple): Positional arguments to pass to torch.save. **kwargs (Any): Keyword arguments to pass to torch.save. éééN)ÚrangeÚ _torch_saveÚ RuntimeErrorÚtimeÚsleep)rrÚiÚer r r Ú torch_saveYs r))N)Ú__doc__r%ÚpathlibrrÚnumpyrZtorchrrZ IMREAD_COLORÚstrÚintr ZndarrayrÚloadrÚsaver#rr)r r r r Ús