o W ht@sddlZddlZddlZddlZddlZddlZddlZddlZddlZddl m Z ddl m Z ddl mZddlmZddlZddlZddlZddlZddlmZmZmZmZmZmZmZmZmZm 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$j0ddfd d Z1d`d e2fd dZ3d e4fddZ5daddZ6    dbde7de7de7de4de4de7d e4fddZ8dcd!d"Z9d#d$Z:e)ddd&d'Z;ded*e7de4de4d e4fd+d,Zdfd4d5Z?dgd6e7de4fd7d8Z@dhd:d;ZAdid=d>ZBdjd@dAZCdBdCZDdkdDdEZEdldFdGZFdHdIZGdJdKZHe$fdLdMZIdmdNeeJfdOdPZKd eLfdQdRZMd e4fdSdTZNdUdVZOd e4fdWdXZPee"QdYZRe"Qd[ZSe"Qd\ZTe"Qd]ZUe>> from ultralytics.utils.checks import parse_requirements >>> parse_requirements(package="ultralytics") cSsg|]}d|vr|qS)z extra == .0xr r L/var/www/vscode/kcb/lib/python3.10/site-packages/ultralytics/utils/checks.py Cz&parse_requirements..#rz ([a-zA-Z0-9-_]+)\s*([<>!=~]+.*)?rname specifier) r distributionrequiresr read_text splitlinesstrip startswithsplitrematchappendr) file_pathpackager. requirementsliner5r r r$parse_requirements3s*r;0.0.0returnc Cs^zttttd|ddWSty.}ztd|d|WYd}~dSd}~ww)a9 Convert a version string to a tuple of integers, ignoring any extra non-numeric string attached to the version. Args: version (str): Version string, i.e. '2.0.1+cpu' Returns: (Tuple[int, int, int]): Tuple of integers representing the numeric part of the version, i.e. (2, 0, 1) z\d+Nzfailure for parse_version(z), returning (0, 0, 0): )rrr)tuplemapintr4findall Exceptionrwarning)versioner r r$ parse_versionRs rGcCstddt|DS)a Check if a string is composed of only ASCII characters. Args: s (str | list | tuple | dict): Input to be checked (all are converted to string for checking). Returns: (bool): True if the string is composed only of ASCII characters, False otherwise. css|] }t|dkVqdS)N)ord)r"cr r r$ mzis_ascii..)allstr)sr r r$is_asciics rP r(r)cs`tttjr nt|tr|g}n-t|ttfr#t|}n!t|tr6|r1t|gnt |}nt d|dt |j dt ||krld}|dkrZtd|d|tdt|d |t|g}fd d |D}||krtd|d d ||dkrt |dkr|d|dg}|S|dkrt |dkr|d}|S|}|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 | List[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] | 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. zupdating to 'imgsz=z'. cs$g|]}tt|qSr )maxmathceilr!floorstrider r$r%s$zcheck_imgsz..z must be multiple of max stride z, updating to r)r)rA isinstancetorchTensorrRlistr?rN isnumericeval TypeErrortype__name__len ValueErrorrrD)imgszrWmin_dimmax_dimrVmsgszr rUr$ check_imgszps4     $rhrEFcurrentrequiredr+hardverboserfc Cs|std|d|ddS|ds>z |}t|}Wntjy=}z|r2t|d|WYd}~dSd}~ww|sBdSd |vr^trLd |vs\trRd |vs\t r^d |vr^d |vr^dSd}d}d} t |} | d dD]U} t d| \}}|sd}t |} |dkr| | krd} qp|dkr| | krd} qp|dkr| | ksd} qp|dkr| | ksd} qp|dkr| | ksd} qp|dkr| | ksd} qp| s||d|d|d|} |rt| |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): Name to be used in warning message. hard (bool): If True, raise an AssertionError if the requirement is not met. verbose (bool): If True, print warning message if requirement is not met. msg (str): Extra message to display if verbose. Returns: (bool): True if requirement is met, False otherwise. Examples: 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") zinvalid check_version(, z!) requested, please check values.Trz& package is required but not installedNF sys_platformwin32linuxmacosdarwinr,z([^0-9]*)([\d.]+)z>=z==z!=z<=>)timeoutinforEN)requestspackagesurllib3disable_warningsget status_codejsonrC) package_nameresponser r r$check_latest_pypi_versions   rcCsbtr/tr/z ddlm}t}t|d|r"td|dWdSWdSty.YdSwdS)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__ruz)New https://pypi.org/project/ultralytics/u8 available 😃 Update with 'pip install -U ultralytics'TF) rr rrrr~rrrC)rlatestr r r$check_pip_update_availables    r Arial.ttfcs~ddlm}tj}t|}|r|Sfdd|D}t|r(|dSd|}tj |ddr=tj ||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: (Path): Resolved font file path. r) font_managercsg|]}|vr|qSr r )r"rOfontr r$r%<r&zcheck_font..z?https://github.com/ultralytics/assets/releases/download/v0.0.0/T)check)urlfileN) matplotlibrrr+rexistsfindSystemFontsanyris_url safe_download)rrr+rmatchesrr rr$ check_font(s  r3.8.0TminimumcCstt|d||dS)a Check current python version against the required minimum version. Args: minimum (str): Required minimum version of python. hard (bool): If True, raise an AssertionError if the requirement is not met. verbose (bool): If True, print warning message if requirement is not met. Returns: (bool): Whether the installed Python version meets the minimum constraints. Python)r+rkrl)r~r)rrkrlr r r$ check_pythonGs rr c stddd}t|tr)|}|sJ|d|dfddt|D}nt|tr1|g}g}|D]A}|dd d d }t d |} | d | drU| d nd } } z t t | | sdJWq5tt jfyv||Yq5wtdd ddd} ddd|D} | r |r tr t|}t|dd|d kd|dzAt}tsJdt| | |t|}t|d|dd|dd|d kd| t|dtdddWd"Sty }zt|d|WYd }~d!Sd }~wwd!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. Examples: >>> 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%rs&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$rKrLz%check_requirements..z Ultralytics requirementrOz$ not found, attempting AutoUpdate...zAutoUpdate skipped (offline)u AutoUpdate success ✅ .1fz s, installed z package: z;Restart runtime or rerun command for updates to take effect u ❌ NFT)rrXrresolverr;rNr3replacer4r5r1r~rrEAssertionErrorrwr6rjoinrrarrtimerrDrC)r9rinstallcmdsprefixrpkgsr| r_strippedr5r+rjrrOntdtrFr rr$check_requirementsVsR    "   $  0 rc sdgdgdgdgdgdgdgdgd gd }d tjd d d dd}||vr^||}d td d d ddtfdd|Dr`tdd|d|d ddSdSdS)a2 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 compatibility table based on: https://github.com/pytorch/vision#installation. z0.21z0.20z0.19z0.18z0.17z0.16z0.15z0.14z0.13) z2.6z2.5z2.4z2.3z2.2z2.1z2.0z1.13z1.12.+rNr)c3s|]}|kVqdS)Nr )r"r} v_torchvisionr r$rKsz$check_torchvision..z 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)rrYrr3rrMrrD)compatibility_tablev_torchcompatible_versionsr rr$check_torchvisions,$ "r yolo11n.pt.ptcCs|r:|r)dtyper(Tz>Environment does not support cv2.imshow() or PIL Image.show() NF)r r r osenvironcv2imshownpzerosuint8waitKeydestroyAllWindowsrCrrD)warnrFr r r$ check_imshowCs    rc Csddl}ddlm}trtjddd|rVd}|j}td\}}}d t d ||d d |||d d||d d } z ddl m } | Wn tyUYnwd} ||ddtd| dS)z Return a human-readable YOLO software and hardware summary. Args: verbose (bool): Whether to print verbose information. device (str): Device to use for YOLO. rN) select_device sample_dataT) ignore_errors@r(z CPUs, rz GB RAM, z GB disk))displayrF)devicenewlineuSetup complete ✅ )psutilultralytics.utils.torch_utilsrr shutilrmtreevirtual_memorytotal disk_usager cpu_countIPythonr clear_output ImportErrorrr) rlrr rgibramrusedfreerOrr r r$ check_yolo\s$  8    rc Csddl}ddlm}ddlm}m}d}totj}t t d\}}}t |t tr/dntr3dnd tt|j|d d |||d d||d d |t|r\|dd nd|rdtjnd|rktjjndd } tdddd| Ddi} tddD]C} zt| j} t| t| j | jddrdnd} Wntj!yd} d} Ynw| | | j | | j<t| jd| | jq| | d<t"rt#dt#dt#dt#dt#dt#d d!}tddd"d|D|| d#<| S)$z Collect and print relevant system information including OS, Python, RAM, CPU, and CUDA. Returns: (dict): Dictionary containing system information. rN) ENVIRONMENT) get_cpu_info get_gpu_inforrgitpipotherz.2fz GBr)index) OS EnvironmentrInstallrRAMDiskCPUz CPU countGPUz GPU countCUDArcss"|] \}}|d|VqdS)<20Nr r"kr}r r r$rK z&collect_system_info..r)r8T)r+rku✅ u❌ z(not installed)r)z Package Info RUNNER_OSGITHUB_EVENT_NAMEGITHUB_WORKFLOW GITHUB_ACTORGITHUB_REPOSITORYGITHUB_REPOSITORY_OWNER)r-r.r/r0r1r2css"|] \}}|d|VqdS)rNr r*r r r$rKr,z GitHub Info)$r ultralytics.utilsrr rrrYcuda is_availablerr rplatformrr r rNrrrrr device_countrErrritemsr;rr+r~r,rwrgetenv)r rrrrr4rrr info_dict package_infor|riis_met github_infor r r$collect_system_info~sX & $"r>c sHddlmtjtd}jdvrdStdtj }t j }t ||r9t|d|ddSfd d }td }t|d d }zddlm}||d|s_Jt|dWdSty}t|d|YdSttfyt|d|YdStyt|dYdSw)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. Returns: (bool): Returns True if the AMP functionality works correctly with YOLO11 model, else False. Examples: >>> from ultralytics import YOLO >>> from ultralytics.utils.checks import check_amp >>> model = YOLO("yolo11n.pt").model.cuda() >>> check_amp(model) r)autocastzAMP: >cpumpsFzV(nvidia|geforce|quadro|tesla).*?(1660|1650|1630|t400|t550|t600|t1000|t1200|t2000|k40m)u#checks failed ❌. AMP training on zW GPU may cause NaN losses or zero-mAP results, so AMP will be disabled during training.cs|gd}tdtjd}|||dddjj}dd|||dddjj}Wd n1s9wY~|j|jkoNtj||d d S) zAll close FP32 vs AMP results.rF)rcrrlrT)enabledNg?)atol) rRrArWboxesdatashaperYallclosefloat)mimbatchrcabr?rrr r$ amp_allcloses   zcheck_amp..amp_allclosezbus.jpgz1running Automatic Mixed Precision (AMP) checks...z`Setting 'amp=True'. If you experience zero-mAP or NaN losses you can disable AMP with amp=False.)YOLOruchecks passed ✅zGchecks skipped. Offline and unable to download YOLO11n for AMP checks. zichecks skipped. Unable to load YOLO11n for AMP checks due to possible Ultralytics package modifications. zchecks 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)r r?next parametersrrr_r4compile IGNORECASErYr4get_device_nameboolsearchrrDrrrrRConnectionErrorAttributeErrorrxrerror)rrpatterngpurQrL warning_msgrRr rPr$ check_ampsN        r`cCs:ztjd|dddddWStyYdSw)z Return human-readable git description, i.e. v5.0-5-g3e25f1e https://git-scm.com/docs/git-describe. Args: path (Path): Path to git repository. Returns: (str): Human-readable git description. zgit -C z describe --tags --long --alwaysTrNrr)rrrrC)rr r r$ git_describes  $ raargsc sddtj}t|\}}}}}dur+t|\}}}fdd|Dzt|t  d}Wnt yGt|j }Ynw|rO|dnd|rW|dnd}t t|dfd d DdS) z Print function arguments (optional args dict). Args: args (dict, optional): Arguments to print. show_file (bool): Whether to show the file name. show_func (bool): Whether to show the function name. cSs,t|tr|drt|dkrt|S|S)zTClean longer Ultralytics HUB URLs by stripping potential authentication information.httpd)rXrNr2rar)r}r r r$ strip_auths,zprint_args..strip_authNcsi|] \}}|vr||qSr r r*)rbr r$ &szprint_args..rrrmc3s&|]\}}|d|VqdS)=Nr r*)rer r$rK,s$zprint_args..)inspect currentframef_back getframeinfo getargvaluesr8rr relative_torrrbrrrrr) rb show_file show_funcr#r_funcfrmrOr )rbrer$ print_argss  $.rsc CsLztjgddd}|dd}t|WStjttfy%YdSw)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)encodingrr)rrr1r3rACalledProcessErrorrrb)output first_liner r r$cuda_device_count/s rxcCs tdkS)z Check if CUDA is available in the environment. Returns: (bool): True if one or more NVIDIA GPUs are available, False otherwise. r)rxr r r r$cuda_is_availableEs rycCstrGtrGz8td&}|}|d^}}|ddtvr( WdWdSWdn 1s2wYWdSWdStyFYdSwdS)z Check if the current environment is running on a Rockchip SoC. Returns: (bool): True if running on a Rockchip SoC, False otherwise. z/proc/device-tree/compatiblersrNTF)r ropenreadr3rrOSError)rdev_strrpsocr r r$ is_rockchipOs ( rcCs(trdSd}tj|dtjtjdjdkS)z Check if the sudo command is available in the environment. Returns: (bool): True if the sudo command is available, False otherwise. Fzsudo --versionT)rstdoutstderrr)rrrunDEVNULL returncode)cmdr r r$is_sudo_availablecsrz3.8)rkrlz3.11z3.12z3.13z3.10r)r<)rQr(r)r)r<r<rEFFr)r)r)rTF)rrr)T)r)rTrT)rT)F)Tr)NTF)XrrhrSrr6r4r rr importlibrpathlibrtypesrtypingrrnumpyrrrYr3rrrr r r r r rrrrrrrrrrrrrrrrrrr;r?rGrXrPrhrNr~rrrrrrrrrrrrrrr>r`radictrsrArxryrrr2 IS_PYTHON_3_8IS_PYTHON_3_11IS_PYTHON_3_12IS_PYTHON_3_13IS_PYTHON_MINIMUM_3_10IS_PYTHON_MINIMUM_3_12r r r r$s    l 7 X E !   (  "= I