a yfFq @stddlZddlZddlZddlZddlZddlZddlZddlZddlZddl Z ddl m Z ddl m Z ddlmZddlZddlZddlZddlZddlmZmZmZmZmZmZmZmZmZmZm Z m!Z!m"Z"m#Z#m$Z$m%Z%m&Z&m'Z'm(Z(m)Z)m*Z*m+Z+m,Z,m-Z-m.Z.m/Z/e"j0ddfdd Z1dRe2d d d Z3e4d ddZ5dSddZ6dTe7e7e7e4e4e7e4dddZ8dUddZ9ddZ:e(dVd d!Z;dWe7e4e4d$d%d&ZdXd.d/Z?dYe7e4d0d1d2Z@dZd4d5ZAd[d7d8ZBd\d:d;ZCdd?ZEd^d@dAZFdBdCZGdDdEZHe"fdFdGZId_eeJdHdIdJZKeLd dKdLZMe4d dMdNZNeGz&parse_requirements..#rz ([a-zA-Z0-9-_]+)\s*([<>!=~]+.*)?rname specifier) r distributionrequiresr read_text splitlinesstrip startswithsplitrematchappendr) file_pathpackager. requirementsliner5r r r$parse_requirements4s ,r;0.0.0)returnc Csbz ttttd|ddWSty\}z$td|d|WYd}~dSd}~00dS)a Convert a version string to a tuple of integers, ignoring any extra non-numeric string attached to the version. This function replaces deprecated 'pkg_resources.parse_version(v)'. Args: version (str): Version string, i.e. '2.0.1+cpu' Returns: (tuple): Tuple of integers representing the numeric part of the version and the extra string, i.e. (2, 0, 1) z\d+Nu)WARNING ⚠️ failure for parse_version(z), returning (0, 0, 0): )rrr)tuplemapintr4findall Exceptionr warning)versioner r r$ parse_versionWs  rGcCst|}tdd|DS)z Check if a string is composed of only ASCII characters. Args: s (str): String to be checked. Returns: (bool): True if the string is composed only of ASCII characters, False otherwise. css|]}t|dkVqdS)N)ord)r"cr r r$ wr&zis_ascii..)strall)sr r r$is_asciiis rO r(r)csbtttjrnt|tr.|g}nZt|ttfrFt|}nBt|trl|rbt|gnt |}nt d|dt |j dt ||krd}|dkrtd|d|tdt|d |t|g}fd d |D}||krtd |d d||dkr:t |dkr:|d|dgn"|dkrZt |dkrZ|dn|}|S)a( Verify image size is a multiple of the given stride in each dimension. If the image size is not a multiple of the stride, update it to the nearest multiple of the stride that is greater than or equal to the given floor value. Args: imgsz (int | cList[int]): Image size. stride (int): Stride value. min_dim (int): Minimum number of dimensions. max_dim (int): Maximum number of dimensions. floor (int): Minimum allowed value for image size. Returns: (List[int]): Updated image size. z'imgsz=z' is of invalid type zK. Valid imgsz types are int i.e. 'imgsz=640' or list i.e. 'imgsz=[640,640]'z'train' and 'val' imgsz must be an integer, while 'predict' and 'export' imgsz may be a [h, w] list or an integer, i.e. 'yolo export imgsz=640,480' or 'yolo export imgsz=640'r(zimgsz=z is not a valid image size. u"WARNING ⚠️ updating to 'imgsz=z'. cs$g|]}tt|qSr )maxmathceilr!floorstrider r$r%r&zcheck_imgsz..uWARNING ⚠️ imgsz=z must be multiple of max stride z, updating to r)r)rA isinstancetorchZTensorrQlistr?rL isnumericeval TypeErrortype__name__len ValueErrorr rD)imgszrVZmin_dimZmax_dimrUmsgszr rTr$ check_imgszzs,      LrdrEF)currentrequiredr+hardverboserbr=c Cs|s td|d|ddS|dsz|}t|}WnPtjy}z6|rlttd|d|nWYd}~d SWYd}~n d}~00|sdSd |vrtrd |vst rd |vst rd |vrd|vrdSd}d}d} t |} | d dD]} td| \}}t |} |dkr.| | kr.d } q|dkrH| | krHd } q|dvrb| | ksbd } q|dkr|| | ks|d } q|dkr| | ksd } q|dkr| | ksd } q| sd|||d|d|d| } |rtt| |rt| | S)a Check current version against the required version or range. Args: current (str): Current version or package name to get version from. required (str): Required version or range (in pip-style format). name (str, optional): Name to be used in warning message. hard (bool, optional): If True, raise an AssertionError if the requirement is not met. verbose (bool, optional): If True, print warning message if requirement is not met. msg (str, optional): Extra message to display if verbose. Returns: (bool): True if requirement is met, False otherwise. Example: ```python # Check if current version is exactly 22.04 check_version(current="22.04", required="==22.04") # Check if current version is greater than or equal to 22.04 check_version(current="22.10", required="22.04") # assumes '>=' inequality if none passed # Check if current version is less than or equal to 22.04 check_version(current="22.04", required="<=22.04") # Check if current version is between 20.04 (inclusive) and 22.04 (exclusive) check_version(current="21.10", required=">20.04,<22.04") ``` u%WARNING ⚠️ invalid check_version(, z!) requested, please check values.TruWARNING ⚠️ z& package is required but not installedNF sys_platformwin32linuxZmacosdarwinr,z([^0-9]*)([\d.]+)z==z!=>rz>=z<=>)timeoutinforEN) contextlibsuppressrCrequestspackagesurllib3disable_warningsget status_codejson) package_nameresponser r r$check_latest_pypi_versions   rcCsztrvtrvttTddlm}t}t|d|rXt d|dWddSWdn1sl0YdS) z Checks if a new version of the ultralytics package is available on PyPI. Returns: (bool): True if an update is available, False otherwise. r) __version__rpz)New https://pypi.org/project/ultralytics/u8 available 😃 Update with 'pip install -U ultralytics'NTF) rr r~rrCrzrrryr r})rZlatestr r r$check_pip_update_availables   .r Arial.ttfcs~ddlm}tj}t|}|r*|Sfdd|D}t|rP|dSd|}tj |ddrztj ||d|Sd S) z Find font locally or download to user's configuration directory if it does not already exist. Args: font (str): Path or name of font. Returns: file (Path): Resolved font file path. r) font_managercsg|]}|vr|qSr r )r"rNfontr r$r%Br&zcheck_font..z?https://github.com/ultralytics/assets/releases/download/v0.0.0/T)check)urlfileN) Z matplotlibrrr+rexistsZfindSystemFontsanyris_url safe_download)rrr+rmatchesrr rr$ check_font.s   r3.8.0T)minimumrgr=cCstt|d|dS)aS Check current python version against the required minimum version. Args: minimum (str): Required minimum version of python. hard (bool, optional): If True, raise an AssertionError if the requirement is not met. Returns: (bool): Whether the installed Python version meets the minimum constraints. Python)r+rg)ryr)rrgr r r$ check_pythonMs rr cs$tddd}ttt|tr^|}|sFJ|d|dfddt|D}nt|trn|g}g}|D]}| dd  d d }t d |} | d | dr| d nd } } ztt| | sJWqvttjfy||Yqv0qvtdd ddd} ddd|D} | r |rtrt|}t|dd|d kd|dzzt}tszJdt| | |t|}t|d|dd|dd|d kd|d|dtdddWn<ty}z"t|d |WYd!}~d"Sd!}~00nd"Sd#S)$a Check if installed dependencies meet YOLOv8 requirements and attempt to auto-update if needed. Args: requirements (Union[Path, str, List[str]]): Path to a requirements.txt file, a single package requirement as a string, or a list of package requirements as strings. exclude (Tuple[str]): Tuple of package names to exclude from checking. install (bool): If True, attempt to auto-update packages that don't meet requirements. cmds (str): Additional commands to pass to the pip install command when auto-updating. Example: ```python from ultralytics.utils.checks import check_requirements # Check a requirements.txt file check_requirements("path/to/requirements.txt") # Check a single package check_requirements("ultralytics>=8.0.0") # Check multiple packages check_requirements(["numpy", "ultralytics>=8.0.0"]) ``` redboldz requirements: z not found, check failed.cs&g|]}|jvr|j|jqSr r*r!excluder r$r%{r&z&check_requirements../z.gitrz([a-zA-Z0-9-_]+)([<>!=~]+.*)?r(r))timesdelaycSstjd|d|ddS)z4Attempt pip install command with retries on failure.zpip install --no-cache-dir rTshell) subprocess check_outputdecode)rcommandsr r r$attempt_installsz+check_requirements..attempt_installcss|]}d|dVqdS)"Nr r!r r r$rKr&z%check_requirements..z Ultralytics requirementrNz$ not found, attempting AutoUpdate...zAutoUpdate skipped (offline)u AutoUpdate success ✅ .1fz s, installed z package:  u ⚠️ z;Restart runtime or rerun command for updates to take effectu ❌ NFT)rrcheck_torchvisionrWrresolverr;rLr3replacer4r5r1ryrrEAssertionErrorrrr6rjoinrr_r r}timerrCrD)r9rinstallZcmdsprefixrpkgsrwZ r_strippedr5r+rfrrNntdtrFr rr$check_requirements[sT    "   $ (rcsdgdgdgdgdgdgdgd}d tjd d d d d }||vr||}d td d d d d tfdd|Drtdd|d|d dd S)a Checks the installed versions of PyTorch and Torchvision to ensure they're compatible. This function checks the installed versions of PyTorch and Torchvision, and warns if they're incompatible according to the provided compatibility table based on: https://github.com/pytorch/vision#installation. The compatibility table is a dictionary where the keys are PyTorch versions and the values are lists of compatible Torchvision versions. z0.19z0.18z0.17z0.16z0.15z0.14z0.13)z2.4z2.3z2.2z2.1z2.0z1.13z1.12.+rNr)c3s|]}|kVqdS)Nr )r"rxZ v_torchvisionr r$rKr&z$check_torchvision..uWARNING ⚠️ torchvision==z is incompatible with torch==z . Run 'pip install torchvision==z' to fix torchvision or 'pip install -U torch torchvision' to update both. For a full compatibility table see https://github.com/pytorch/vision#installation)rrXrr3rrMprint)Zcompatibility_tableZv_torchZcompatible_versionsr rr$rs$  $"r yolov8n.pt.ptcCsx|rt|rtt|tr|f}t|ttfr*|n|gD]B}t|j}t|r0||vs0J||d|d|q0dS)z$Check file(s) for acceptable suffix.z acceptable suffix is z, not N) rWrLrYr?rsuffixlowerr1r_)rrrbfrNr r r$ check_suffixs r)rrhcCsd|vsd|vrd|vr&|dd}nbd|vrd|vr|}tdd|}td d|}td d|}||kr|rtd |d |d |S)z?Replace legacy YOLOv5 filenames with updated YOLOv5u filenames.Zyolov3Zyolov5zu.yaml.yamlruz(.*yolov5([nsmlx]))\.ptz\1u.ptz(.*yolov5([nsmlx])6)\.ptz(.*yolov3(|-tiny|-spp))\.ptuPRO TIP 💡 Replace 'model=z' with new 'model=z'. YOLOv5 'u' models are trained with https://github.com/ultralytics/ultralytics and feature improved performance vs standard YOLOv5 models trained with https://github.com/ultralytics/yolov5. )rr4subr r})rrhZ original_filer r r$check_yolov5u_filenames ryolov8ncCs4|r,t|js,t|jtjvr,t|dS|SdS)z0Return a model filename from a valid model stem.rN)rrstemrZGITHUB_ASSETS_STEMS with_suffix)modelr r r$check_model_file_from_stemsrrcCs4t||t|}t|}|rDd|vr6t|sD|drH|S|r|dr|}t|t|}|rt dt |d|nt j ||ddt|Stjttd|d d ptttj|}|s|rtd |d n(t|d kr|rtd|d|t|r,|dSgSdS)z4Search/download file (if necessary) and return path.z://zgrpc://)zhttps://zhttp://zrtsp://zrtmp://ztcp://zFound z locally at F)rrZunzipz**T) recursive'z' does not existr(zMultiple files match 'z', specify exact path: rN)rrLr1rrrrr2rr r}rrrglobrparentFileNotFoundErrorr_)rrdownload download_dirrgrfilesr r r$ check_files2    .rrz.ymlcCst|||dS)zJSearch/download YAML file (if necessary) and return path, checking suffix.rg)r)rrrgr r r$ check_yaml srcCs:t|}t|}|o8|jdt|j|jkS)a Check if the resolved path is under the intended directory to prevent path traversal. Args: basedir (Path | str): The intended directory. path (Path | str): The path to check. Returns: (bool): True if the path is safe, False otherwise. N)rrrpartsr_)ZbasedirpathZbase_dir_resolvedZ path_resolvedr r r$check_is_path_safes  rc Csz\tr$tstrJdtjvs$Jdtdtjdtj dt dt t dWdSt y}z"|rt d|WYd }~d Sd }~00d S) z-Check if environment supports image displays.ZDISPLAYz+The DISPLAY environment variable isn't set.test)rr>)Zdtyper(TuMWARNING ⚠️ Environment does not support cv2.imshow() or PIL Image.show() NF)r rr osenvironcv2ZimshownpZzerosZuint8ZwaitKeyZdestroyAllWindowsrCr rD)warnrFr r r$ check_imshow!s   rc Csddl}ddlm}tr@tdddr.tdtr@tj dd d |rd }| j }t d \}}}d t d||dd|||dd ||dd } tt$ddlm} | Wdq1s0Ynd} ||ddtd| dS)z;Return a human-readable YOLO software and hardware summary.rN) select_deviceZwandbF)rzpip uninstall -y wandbZ sample_dataT) ignore_errors@r(z CPUs, rz GB RAM, z GB disk))displayr)devicenewlineuSetup complete ✅ )psutilultralytics.utils.torch_utilsrr rrsystemrshutilrmtreevirtual_memorytotal disk_usage cpu_countr~rrCZIPythonr clear_outputr r}) rhrrrZgibramrusedfreerNrr r r$ check_yolo2s$    8  ( rcCsddl}ddlm}ddlm}|jd}tt dddt dd d|dd dt dd dt rpd n t rxd ndddd|dddd|dddtrtjrtjjnddtddD]p}z,t|j}t|t|jddrdnd}Wntjyd}d}Yn0t |jd|||jqtrt dtddtddtd d!td"d#td$d%td&d dS)'zWCollect and print relevant system information including OS, Python, RAM, CPU, and CUDA.rN) ENVIRONMENT) get_cpu_inforrZOSz<20 EnvironmentrZInstallgitpipotherZRAMz.2fz GB ZCPUCUDArz)r8Tru✅ u❌ z(not installed)z RUNNER_OS: Z RUNNER_OSz GITHUB_EVENT_NAME: ZGITHUB_EVENT_NAMEz GITHUB_WORKFLOW: ZGITHUB_WORKFLOWz GITHUB_ACTOR: Z GITHUB_ACTORz GITHUB_REPOSITORY: ZGITHUB_REPOSITORYz GITHUB_REPOSITORY_OWNER: ZGITHUB_REPOSITORY_OWNER)rultralytics.utilsrrrrrrr r}platformrrr rXcudaZ is_availablerEr;rr+ryrLr,rrrrgetenv)rrrZram_inforwreZis_metr r r$collect_system_infoOsl      "rc sddlmtjjdvr(dSfdd}td}td}t |d d }z2dd l m }||d |s|Jt |d Wnrt yt |d|YnNttfyt |d|Yn&tyt |dYdS0dS)a Checks the PyTorch Automatic Mixed Precision (AMP) functionality of a YOLO11 model. If the checks fail, it means there are anomalies with AMP on the system that may cause NaN losses or zero-mAP results, so AMP will be disabled during training. Args: model (nn.Module): A YOLO11 model instance. Example: ```python from ultralytics import YOLO from ultralytics.utils.checks import check_amp model = YOLO("yolo11n.pt").model.cuda() check_amp(model) ``` Returns: (bool): Returns True if the AMP functionality works correctly with YOLO11 model, else False. r)autocast>cpuZmpsFcs|gd}tdtjd}|||dddjj}dd(|||dddjj}Wd n1sr0Y~|j|jkotj||d d S) zAll close FP32 vs AMP results.rF)rarrhrT)enabledNg?)Zatol) rQrArVZboxesdatashaperXZallclosefloat)mimbatchraabr rrr r$ amp_allcloses  6zcheck_amp..amp_allclosezbus.jpgzAMP: z>running Automatic Mixed Precision (AMP) checks with YOLO11n...z`Setting 'amp=True'. If you experience zero-mAP or NaN losses you can disable AMP with amp=False.)YOLOz yolo11n.ptuchecks passed ✅u?checks skipped ⚠️, offline and unable to download YOLO11n. uachecks skipped ⚠️. Unable to load YOLO11n due to possible Ultralytics package modifications. uchecks failed ❌. Anomalies were detected with AMP on your system that may lead to NaN losses or zero-mAP results, so AMP will be disabled during training.T)rr next parametersrr]rrr r}rzrConnectionErrorrDAttributeErrorrsr)rrrrZ warning_msgrr rr$ check_ampvs6     rcCsPtt2tjd|dddddWdS1sB0YdS)zbReturn human-readable git description, i.e. v5.0-5-g3e25f1e https://git-scm.com/docs/git-describe.zgit -C z describe --tags --long --alwaysTrNrr)r~rrCrrr)rr r r$ git_describes @rargsc sddtj}t|\}}}}}durVt|\}}}fdd|Dzt|t  d}Wnt yt|j }Yn0|r|dnd|r|dnd}t t|dfd d DdS) z.Print function arguments (optional args dict).cSs,t|tr(|dr(t|dkr(t|S|S)zTClean longer Ultralytics HUB URLs by stripping potential authentication information.httpd)rWrLr2r_r)rxr r r$ strip_authszprint_args..strip_authNcsi|]\}}|vr||qSr r r"krxrr r$ r&zprint_args..rrric3s$|]\}}|d|VqdS)=Nr r$)r#r r$rKr&zprint_args..)inspect currentframef_back getframeinfo getargvaluesitemsrr relative_torrr`rr r}rr) r Z show_fileZ show_funcr#r_funcfrmrNr )r r#r$ print_argss  $r2c CsPz.tjgddd}|dd}t|WStjttfyJYdS0dS)z Get the number of NVIDIA GPUs available in the environment. Returns: (int): The number of NVIDIA GPUs available. )z nvidia-smiz--query-gpu=countz--format=csv,noheader,nounitszutf-8)encodingrrN)rrr1r3rACalledProcessErrorrr`)output first_liner r r$cuda_device_counts r7cCs tdkS)z Check if CUDA is available in the environment. Returns: (bool): True if one or more NVIDIA GPUs are available, False otherwise. r)r7r r r r$cuda_is_availablesr8z3.10rz3.12)r<)rPr(r)r)r<r<rEFFr)rz)r)rT)rrr)T)r)rTrT)rT)F)Tr)NTF)Rr~rr(rRrrr4rrr importlibrpathlibrtypingrrnumpyrrrXrrrrrr r r r r rrrrrrrrrrrrrrrrrrr;r?rGboolrOrdrLryrrrrrrrrrrrrrrrrrdictr2rAr7r8ZIS_PYTHON_MINIMUM_3_10r2ZIS_PYTHON_3_12r r r r$sx   p# 7 X  I$      '>