o W hj@sddlZddlZddlZddlZddlZddlmZddlmZddlmZGdddej Z eddZ dd d Z e fd dZe fddZddZdddZdedd fddZdS)N)contextmanager)datetime)Pathc@s(eZdZdZddZddZddZdS) WorkingDirectorya A context manager and decorator for temporarily changing the working directory. This class allows for the temporary change of the working directory using a context manager or decorator. It ensures that the original working directory is restored after the context or decorated function completes. Attributes: dir (Path | str): The new directory to switch to. cwd (Path): The original current working directory before the switch. Methods: __enter__: Changes the current directory to the specified directory. __exit__: Restores the original working directory on context exit. Examples: Using as a context manager: >>> with WorkingDirectory('/path/to/new/dir'): >>> # Perform operations in the new directory >>> pass Using as a decorator: >>> @WorkingDirectory('/path/to/new/dir') >>> def some_function(): >>> # Perform operations in the new directory >>> pass cCs||_t|_dS)zgSets the working directory to 'new_dir' upon instantiation for use with context managers or decorators.N)dirrcwdresolve)selfnew_dirr K/var/www/vscode/kcb/lib/python3.10/site-packages/ultralytics/utils/files.py__init__)szWorkingDirectory.__init__cCt|jdS)z[Changes the current working directory to the specified directory upon entering the context.N)oschdirr)r r r r __enter__.zWorkingDirectory.__enter__cCr)zARestores the original working directory when exiting the context.N)rrr)r exc_typeexc_valexc_tbr r r __exit__2rzWorkingDirectory.__exit__N)__name__ __module__ __qualname____doc__r rrr r r r r s  rc cs0dt|vrt|t}t|}tu}t||jdd}|r+t ||n| r=|j j dddt ||z"|rDt|n|VW|rUtj ||ddn,| r_t ||n|rmtj ||ddw| rxt ||wwWddSWddS1swYdS|VdS)au Context manager to handle paths with spaces in their names. If a path contains spaces, it replaces them with underscores, copies the file/directory to the new path, executes the context code block, then copies the file/directory back to its original location. Args: path (str | Path): The original path that may contain spaces. Yields: (Path | str): Temporary path with spaces replaced by underscores if spaces were present, otherwise the original path. Examples: >>> with spaces_in_path('/path/with spaces') as new_path: >>> # Your code here >>> pass  _Tparentsexist_ok) dirs_exist_okN)str isinstancertempfileTemporaryDirectorynamereplaceis_dirshutilcopytreeis_fileparentmkdircopy2)pathstringtmp_dirtmp_pathr r r spaces_in_path7s8     " r2FcCst|}|r:|s:|r|d|jfn|df\}}tddD]}||||}tj|s5nq!t|}|rC|jddd|S)ao Increment a file or directory path, i.e., runs/exp --> runs/exp{sep}2, runs/exp{sep}3, ... etc. If the path exists and `exist_ok` is not True, the path will be incremented by appending a number and `sep` to the end of the path. If the path is a file, the file extension will be preserved. If the path is a directory, the number will be appended directly to the end of the path. Args: path (str | Path): Path to increment. exist_ok (bool): If True, the path will not be incremented and returned as-is. sep (str): Separator to use between the path and the incrementation number. mkdir (bool): Create a directory if it does not exist. Returns: (Path): Incremented path. Examples: Increment a directory path: >>> from pathlib import Path >>> path = Path("runs/exp") >>> new_path = increment_path(path) >>> print(new_path) runs/exp2 Increment a file path: >>> path = Path("runs/exp/results.txt") >>> new_path = increment_path(path) >>> print(new_path) runs/exp/results2.txt r3i'Tr) rexistsr* with_suffixsuffixrangerr.r,)r.rsepr,r7npr r r increment_pathks $ r<cCs"ttt|j}|jS)z>Return days since the last modification of the specified file.)rnow fromtimestamprstatst_mtimedays)r.dtr r r file_agesrCcCs.tt|j}|jd|jd|jS)z8Returns the file modification date in 'YYYY-M-D' format.-)rr>rr?r@yearmonthday)r.tr r r file_datesrIcCsXt|ttfr*d}t|}|r|j|S|r*tdd|dD|SdS)z:Returns the size of a file or directory in megabytes (MB).icss"|] }|r|jVqdS)N)r*r?st_size).0fr r r s zfile_size..z**/*g) r"r!rr*r?rJr'sumglob)r.mbr r r file_sizesrQ.cCs,tj|ddd}|rt|tjjdSdS)zdReturns the path to the most recent 'last.pt' file in the specified directory for resuming training.z /**/last*.ptT) recursive)keyr3)rOmaxrr.getctime) search_dir last_listr r r get_latest_runsrY)z yolo11n.ptc Csddlm}ddlm}|d}|jddd|D]0}||}td|||}||r6|d|j_||} td |d | | | qd S) ae Update and re-save specified YOLO models in an 'updated_models' subdirectory. Args: model_names (Tuple[str, ...]): Model filenames to update. source_dir (Path): Directory containing models and target subdirectory. update_names (bool): Update model names from a data YAML. Examples: Update specified YOLO models and save them in 'updated_models' subdirectory: >>> from ultralytics.utils.files import update_models >>> model_names = ("yolo11n.pt", "yolov8s.pt") >>> update_models(model_names, source_dir=Path("/models"), update_names=True) r)YOLO)default_class_namesupdated_modelsTrzLoading model from z coco8.yamlz Re-saving z model to N) ultralyticsrZultralytics.nn.autobackendr[r,printhalfmodelnamessave) model_names source_dir update_namesrZr[ target_dir model_name model_pathra save_pathr r r update_modelss    rk)Fr3F)rR) contextlibrOrr(r#rrpathlibrContextDecoratorrr2r<__file__rCrIrQrYrkr r r r s"   *  3 0