/jmdddlmZddlZddlZddlZddlZddlZddlZddlZddl Z ddl Z ddl Z ddl Z ddl Z ddlmZddlmZddlmZddlZddlZddlZddlmZmZmZmZmZmZmZm 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/m0Z0m1Z1m2Z2m3Z3m4Z4m5Z5e)j6dz dfd Z7dXd Z8ej9dYdZdZ:d[dZ;d\dZd_d"Z?d#Z@e/ej9d`d%ZAdadbd)ZBe0d*ZCe0e)j6dz d+d'dfd,ZDd-ZEdcddd0ZFd1ZGded4ZHdfdgd6ZIdhdid:ZJdjd<ZKdkd>ZLdldBZMej9dmdCZNdndDZOdEZPdFZQdGZRdodpdJZSdqdLZTd[dMZUdNZVdOZWd[dPZXeBdQdd'ReGe'jYdQZZe'jYdSZ[e'jYdTZ\e'jYdUZ]e'jYdVZ^eBdSdWZ_eBdTdWZ`eBdUdWZadS)r) annotationsN)metadata)Path)SimpleNamespace)ARM64ASSETS ASSETS_URL AUTOINSTALLGITIS_COLAB IS_DOCKER IS_JETSON IS_KAGGLEIS_PIP_PACKAGELINUXLOGGERMACOSONLINEPYTHON_VERSION RKNN_CHIPSROOT TORCH_VERSIONTORCHVISION_VERSIONUSER_CONFIG_DIRWINDOWSRetryThreadingLocked TryExcept clean_urlcolorstr downloadsis_github_action_runningurl2filezrequirements.txtc<|r$dtj|jD}n3t|}g}|D]}|}|r|ds|dd}tj d|x}rL| t|d|dr|dnd|S) a/Parse a requirements.txt file, ignoring lines that start with '#' and any text after '#'. Args: file_path (Path): Path to the requirements.txt file. package (str, optional): Python package to use instead of requirements.txt file. Returns: requirements (list[SimpleNamespace]): List of parsed requirements as SimpleNamespace objects with `name` and `specifier` attributes. Examples: >>> from ultralytics.utils.checks import parse_requirements >>> parse_requirements(package="ultralytics") cg|]}d|v| S)z extra == .0xs ]/home/longshao/multi-rider-rag/.venv/lib/python3.11/site-packages/ultralytics/utils/checks.py z&parse_requirements..Js#___!+]^J^J^AJ^J^J^#rz ([a-zA-Z0-9-_]+)\s*([<>!=~]+.*)?r$name specifier) r distributionrequiresr read_text splitlinesstrip startswith partitionrematchappendr) file_pathpackager5 requirementsliner<s r+parse_requirementsrB:s<__x4W==F___ ??,,..99;;Lttzz||  t,, t>>#&&q)//11D!DdKKKu t##OqafghaiMqUSTX^^M]M]M]oq$r$r$rsss r- import_namestrreturnctjD]>}|dpd}||vr|jdcS?|S)z`Get the pip distribution name for a given import name (e.g., 'cv2' -> 'opencv-python-headless').z top_level.txtr$Name)r distributionsr6split)rCdist top_levels r+get_distribution_namerLYse&(())^^O44:AACC ) # #=( ( ( ( $ r-0.0.0tuplec  ttttjd|ddS#t $r%}t jd|d|Yd}~dSd}~wwxYw)a%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): 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)rNmapintr;findall Exceptionrwarning)versiones r+ parse_versionrXbsSbj99"1"=>>??? XGXXUVXXYYYwwwwwszis_ascii..|s,,,s1vv|,,,,,,r-)allrD)ss r+is_asciirbss' ,,SVV,,, , ,,r- r/r0ctttjrnt|tr|g}nt|t t frt |}nvt|tr9|rt|gntj |}n(td|dt|j dt||krTd}|dkrtd|d|t!jdt |d |t |g}fd |D}||krt!jd|d d ||d kr#t|dkr|d|dgn"|dkrt|dkr|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 | 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'. c`g|]*}ttj|z z+Sr')maxmathceil)r)r*floorstrides r+r,zcheck_imgsz..s6 D D D#diF ##f,e 4 4 D D Dr-z must be multiple of max stride z, updating to r0r)rR isinstancetorchTensorrflistrNrD isnumericast literal_eval TypeErrortype__name__len ValueErrorrrU)imgszrjmin_dimmax_dimrimsgszs ` ` r+ check_imgszr|sFEL!A!AMv N NF%   ED%= ) ) U  E3   % 1 1NU s7G7N7N Ye Y Y$u++2F Y Y Y    5zzG Y  a<<NeNNNNOO OASZZAACAABBBU  D D D D De D D DB U{{aaavaa]_aabbb#a<>> 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<=>>@@ G B ' " " ::!q&&FF 4ZZAFFFF 4ZZaFF 4ZZaFF 3YYAFF 3YYAF $e8eeteewee`cee  /%g.. .  $ N7 # # # MsAB!A<<B ultralyticscddl} |jj|d|dd}|jdkr |dd SdS#t$rYdSwxYw) aReturn the latest version of a PyPI package without downloading or installing it. Args: package_name (str): The name of the package to find the latest version for. Returns: (str | None): The latest version of the package, or None if unavailable. rNzhttps://pypi.org/pypi/z/jsonrP)timeoutinforV)requestspackagesurllib3disable_warningsget status_codejsonrT) package_namerresponses r+check_latest_pypi_versionrsOOO!22444<< L L L LVW<XX  3 & &==??6*95 5 ' & ttsA#A,, A:9A:ctrZtrS ddlm}t }t |d|rt jd|ddSn#t$rYnwxYwdS)zCheck if a new version of the ultralytics package is available on PyPI. Returns: (bool): True if an update is available, False otherwise. r) __version__rz)New https://pypi.org/project/ultralytics/u8 available 😃 Update with 'pip install -U ultralytics'TF) rrrrrrrrrT)rlatests r+check_pip_update_availabler)s  .   / / / / / /.00F[,f,,77  @@@@t       D  5s?A AA Arial.ttfc^ddlm}tj}t|z }|r|Sfd|D}t|r|dStd|}tj |drtj |||SdS) zFind 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 | str): Resolved font file path. r) font_managercg|]}|v| Sr'r')r)rafonts r+r,zcheck_font..SsFFFQDAIIqIIIr-/T)check)urlfileN) matplotlibrrr2rexistsfindSystemFontsanyr r!is_url safe_download)rrr2rmatchesrs` r+ check_fontr?s(''''' ::?D T !D {{}} GFFF,6688FFFG 7||qz $ C4(((Cd3333 r-3.8.0Tminimumc4tt|d||S)aCheck current python version against the required minimum version. Args: minimum (str): Required minimum version of python. hard (bool): If True, raise a ModuleNotFoundError 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)r2rr)rr)rrrs r+ check_pythonr^s xdT[ \ \ \\r-c tddd}g}|D] tjddgddd}|jd ks2t fd |jDs|q#t$r|YwxYw|rtj |d d t|d kzd|dtrdgngddgz}tj|ddd}trdgnggdz|z}tj|ddd}tj |dtj |dtddddSdS)zCheck if apt packages are installed and install missing ones. Args: requirements (list[str]): List of apt package names to check and install. redboldzapt requirements:dpkgz-lTF)rtextrrc3JK|]}|do|vVdS)iiN)r9)r)rAr?s r+r_z)check_apt_requirements..{sI11>B%%9'T/111111r- Ultralytics requirementrar/ $ not found, attempting AutoUpdate...sudoaptupdate)rrr)rinstallz-yu AutoUpdate success ✅;Restart runtime or rerun command for updates to take effect N)r rrrrstdoutr7r=rTrrruis_sudo_availablerU)r@prefixmissing_packagesrcmdr?s @r+check_apt_requirementsrls&eV%8 9 9F - - -^VT7$;DW[chiiiF A%%S1111FLmF^F^F`F`111..%!''000 - - -  # #G , , , , , -  x  J Jsc:J6K6Ka6O/P J JSc J J J   -..6xxB5(:KK44PPP-..6xxB:R:R:RRUee44PPP v666777&vv8F4q#r#rvvvwwwww x xsA-BB*)B*r'c.tddd}tjdddkrt j|ddSt |trT|}| sJ|d |d fd t|D}nt |tr|g}g}|D]6}t |ttfr|n|g}d } |D]} | d ddd} t!jd| } | d| dr| dnd}} t't)j| |rd} n#t,t(jf$rYwxYw| sV|d}d|vr5|d\}}}t!jdd||z|z}||8t7ddd}d d|D}|rL|rGt:r?t=|}t j|dd|dkzd |d t?j}t@s JdtB o tE}t j||||t?j|z }t j|d |d!dt j#|d tdd"d#nY#tH$rJ}|d$|}tK|d%r|j&r |d#|j&z }t j#|Yd&}~d Sd&}~wwxYwd SdS)'aCheck if installed dependencies meet Ultralytics YOLO models requirements and attempt to auto-update if needed. Args: requirements (Path | str | list[str|tuple] | tuple[str]): Path to a requirements.txt file, a single package requirement as a string, a list of package requirements as strings, or a list containing strings and tuples of interchangeable packages. exclude (tuple): 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.3.200", cmds="--index-url https://download.pytorch.org/whl/cpu") Check multiple packages >>> check_requirements(["numpy", "ultralytics"]) Check with interchangeable packages >>> check_requirements([("onnxruntime", "onnxruntime-gpu"), "numpy"]) rrz requirements:$ULTRALYTICS_SKIP_REQUIREMENTS_CHECKS01zN ULTRALYTICS_SKIP_REQUIREMENTS_CHECKS=1 detected, skipping requirements check.Trz not found, check failed.cBg|]}|jv |j|jSr'r1)r)r*excludes r+r,z&check_requirements..s7lllQVWV\dkVkVk1601;00VkVkVkr-Frz.gitr$z([a-zA-Z0-9-_]+)([<>!=~]+.*)?r/r0rzgit+;z [<>!=~]+.*$)timesdelayc |r6tjdtjd|d|ddtjdStjd|d|dtjdS)zGAttempt package installation with uv if available, falling back to pip.z(uv pip install --no-cache-dir --python "z" rz; --index-strategy=unsafe-best-match --break-system-packagesT)shellstderrrzpip install --no-cache-dir )r check_outputsys executableSTDOUT)rcommandsuse_uvs r+attempt_installz+check_requirements..attempt_installs  *N3>NNXNNX`NNN!(  & ?( ? ?X ? ?tT^Telp    r-c3"K|] }d|dV dS)"Nr'r(s r+r_z%check_requirements..s*((aQ((((((r-rrarzAutoUpdate skipped (offline))ru AutoUpdate success ✅ .1frru ❌ outputN)'r osenvironrrrrkrresolverrBrDrnrN rpartitionreplacer;r<r8rrrVAssertionErrorrr:subr=rjoinr rutimerrrrUrThasattrr)r@rrcmdsrrpkgsr candidates satisfied candidate r_strippedr<r2rpkgrsepmarkerrrantrdtrWrzs ` r+check_requirementsr s*6eV_ 5 5F z~~>vrJJJH=zJJE"1X58'KuQx~~'7'7'7(D  !1$!7!7BB $IE#H$AB     Q-C}}#&==#5#5 S&f^R55;fD KK    !   ((4(((((A  { D A K6uu3!a%=uu4uuu v v v IKK=====v"1xzz OOAtFCCCDDDY[[1_ vHHrHHHHIIIss1n o osss   ))a))1h''+AH+???*Cs###uuuuu  5 4s+0$FF0/F0B3L<< N?N  Nctrtrtrtdtddt jddtddS) zSCheck and install ExecuTorch requirements including platform-specific dependencies.zpackaging>=22.0 executorchztorch==+r)rz numpy<=2.3.5N)rrr r rrIr'r-r+check_executorch_requirementsrsi ..9.,---|*QM4G4L4LQ4O*Q*QRRRR~&&&&&r-7.0.0 min_versionctrBtjjdd}t d|d|ddSdS)zCheck and install TensorRT requirements including platform-specific dependencies. Args: min_version (str): Minimum supported TensorRT version (default: "7.0.0"). .rz tensorrt-curz ,!=10.1.0N)rrlrVcudarIr )r cuda_versions r+check_tensorrtr s^  Q})//44Q7 OOOOOOPPPPPQQr-cdgdgdgdgdgdgdgdgd gd gd gd gd gd }dtjdddddd}||vr||}dtjddddddt fd|Dr(t jdd|d|dddSdSdS)a,Check 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.25z0.24z0.23z0.22z0.21z0.20z0.19z0.18z0.17z0.16z0.15z0.14z0.13) z2.10z2.9z2.8z2.7z2.6z2.5z2.4z2.3z2.2z2.1z2.0z1.13z1.12rr r/rNr0c3$K|] }|kV dS)Nr')r)r v_torchvisions r+r_z$check_torchvision..1s(??a}!??????r-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)rrrIrr`rrU)compatibility_tablev_torchcompatible_versionsrs @r+check_torchvisionrsvxxxxxxxxxx"hh}*32215;;C@@!DEEG%%%1':!4!:3!B!B1!E!K!KC!P!PQSRSQS!TUU ????+>??? ? ?  Nd ddGdd1DQ1Gddd      &%  r- yolo26n.pt.ptc`|r|rt|tr|h}t|ttfr|n|gD]o}t|ddx}rd||vsJ||d|d|ldSdSdS)zCheck file(s) for acceptable suffix. Args: file (str | list[str]): File or list of files to check. suffix (str | tuple): Acceptable suffix or tuple of suffixes. msg (str): Additional message to display in case of error. rrz acceptable suffix is z, not .N)rkrDrnrNrlowerr8)rsuffixrzfras r+ check_suffixr#:s ^^ fc " " XF#D4-88Dtf ^ ^AFF%%c**2.4466<<>>>q ^1ww&(((S*]!*]*]6*]*]Z[*]*]((( ^^^^ ^ ^r-rc*d|vsd|vrd|vr|dd}nod|vrkd|vrg|}tjdd|}tjd d|}tjd d|}||kr|rtjd |d |d |S)aReplace legacy YOLOv5 filenames with updated YOLOv5u filenames. Args: file (str): Filename to check and potentially update. verbose (bool): Whether to print information about the replacement. Returns: (str): Updated filename. yolov3yolov5zu.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. )rr;rrr)rr original_files r+check_yolov5u_filenamer*Js48t++ t  <<'22DD d]]s$ M64iFFD65y$GGD68)TJJD}$$$ d=ddTXddd Kr-yolo11nmodel str | Pathct|}|js(|jtjvr|dS|S)zReturn a model filename from a valid model stem. Args: model (str): Model stem to check. Returns: (str | Path): Model filename with appropriate suffix. r)rr!stemr!GITHUB_ASSETS_STEMS with_suffix)r,paths r+check_model_file_from_stemr3esA ;;D ;'49 (EEE&&& Lr-rcBt||t|}t|}|rLd|vr!t |s'|dr|S|r |drddlm }|||}|gS|dd}t ||z t|z }|j d kr| d |r(tjd t|d |n3|jd d t%j||dt|S|r|dr|dr d|ddz}|}t |t|z }|r(tjd t|d |nt%j||dt|St)jtt*dz |z d p-t)jtt*j|z } | s|rt-d|dt/| dkr|rt-d|d| t/| r| dngS)aKSearch/download file (if necessary), check suffix (if provided), and return path. Args: file (str): File name or path, URL, platform URI (ul://), or GCS path (gs://). suffix (str | tuple): Acceptable suffix or tuple of suffixes to validate against the file. download (bool): Whether to download the file if it doesn't exist locally. download_dir (str): Directory to download the file to. hard (bool): Whether to raise an error if the file is not found. Returns: (str | list): Path to the file, or an empty list if not found. z://zgrpc://zul://r)resolve_platform_urirNz.ndjsonT) missing_okzFound z locally at )parentsexist_okF)rrunzip)zhttps://zhttp://zrtsp://zrtmp://ztcp://gs://r<zhttps://storage.googleapis.com/z**) recursive'z' does not existr/zMultiple files match 'z', specify exact path: )r#rDr8r*rrr r9$ultralytics.utils.callbacks.platformr5r#r!unlinkrrrparentmkdirr!rglobrrru) rr!download download_dirrr5ruri_path local_filefiless r+ check_filerIts6v t99??  D !$ ' 'D *.   $t**"3"3"5"5  ::<< " "9 - -  $.djjll--g66$.MMMMMM""4d333 ;I8,''(2Xc]]B   ) )     . . .      K KI3IIZII J J J J   # #D4 # @ @ @  #*E J J J J: .djjll--H. ??7 # # @4tABBx?DL!!HTNN2 ;;== E KC3CCTCC D D D D  #$e D D D D4yy #dTkD011TBBBhdiPSTXT_bfTfPgPgFhFh c c#$>$>$>$>?? ? ZZ!^^^#$aT$a$aZ_$a$abb bu::-uQxx2-r-r'z.ymlc&t|||S)aaSearch/download YAML file (if necessary) and return path, checking suffix. Args: file (str | Path): File name or path. suffix (tuple): Tuple of acceptable YAML file suffixes. hard (bool): Whether to raise an error if the file is not found or multiple files are found. Returns: (str): Path to the YAML file. r6)rI)rr!rs r+ check_yamlrLs dF . . ..r-basedir Path | strr2ct|}t|}|o)|jdt |j|jkS)aCheck 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)rrrpartsru)rMr2base_dir_resolved path_resolveds r+check_is_path_saferSsmW --//JJ&&((M    ! ! tm&9:XC@Q@WEnvironment does not support cv2.imshow() or PIL Image.show() NF)rr rrrcv2imshownpzerosuint8waitKeydestroyAllWindowsrTrrU)warnrWs r+ check_imshowras   Z 1 1 11 ***,Y*** 628IRX>>>??? A  At   b N`]^`` a a auuuuusBB!! C+C  Cc ddl}ddlm}trt jdd|rd}|j}t jd\}}}d tj d ||z d d ||z |z d d||z d d } ddl m } | n#t$rYnwxYwd} tjrt#||dt%jd| dS)zPrint a human-readable YOLO software and hardware summary. Args: verbose (bool): Whether to print verbose information. device (str | torch.device): Device to use for YOLO. rN) select_device sample_dataT) ignore_errors@r(z CPUs, rz GB RAM, z GB disk))displayr$F)devicenewlineuSetup complete ✅ )psutilultralytics.utils.torch_utilsrcr shutilrmtreevirtual_memorytotal disk_usager cpu_countIPythonrh clear_output ImportErrorr is_repocheck_multiple_installrr) rrirkrcgibramrp_usedfreerarhs r+ check_yolor|s]MMM;;;;;;9 m48888 ##%%+#.s33ud t  t tsSy t t t PS?S t t tZ_beZe t t t t  ' ' ' ' ' '  " " " "    D   {!   M//// K)a))*****s B(( B54B5c  ddl}ddlm}ddlm}m}d}t j}ttj d\}}}tj |ttjrdn t rdnd t#t$|j|z d d ||z |z d d||z d d |t+j|r |d nd|rt jnd|rt jjndd } t3jddd| Dzdzi} t;t=dD]} t?j| j } tC| t#| j"| j drdnd} n#t>j#$rd} d} YnwxYw| | | j"| | j <t3j| j d| | j | | d<tIrt+j%dt+j%dt+j%dt+j%dt+j%dt+j%dd }t3jddd!|Dz|| d"<| S)#zCollect 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_inforfrgitpipotherz.2fz GBr)index) OS EnvironmentrInstallrRAMDiskCPUz CPU countGPUz GPU countCUDArc3*K|]\}}|d|VdS)<23Nr'r)krs r+r_z&collect_system_info..*s0 M M41aA1 M M M M M Mr-r)r?T)r2ru✅ u❌ z(not installed)rz Package Info RUNNER_OSGITHUB_EVENT_NAMEGITHUB_WORKFLOW GITHUB_ACTORGITHUB_REPOSITORYGITHUB_REPOSITORY_OWNER)rrrrrrc3*K|]\}}|d|VdS): Nr'rs r+r_z&collect_system_info..Bs0$Q$QTQ[[Q[[$Q$Q$Q$Q$Q$Qr-z GitHub Info)&rkultralytics.utilsr~rlrrrlr is_availabler|rmrqplatformrr rvrrDrrorprrr device_countrVrrritemsrBrLrr2rr3rr"getenv)rkr~rrrxrrp_r{ info_dict package_inforris_met github_infos r+collect_system_infor s; MMM------HHHHHHHH C : " " $ $DLLL&s++NE1d!!" KQ55n-QUU'D ''))/#5>>>>DLC'BBBeckBBBB|~~\^^(,6||!$$$$$26@UZ,,...D&*4 ""  I Ktyy M M9??;L;L M M MMMMPTTUUUL (=m(L(L M M M;; &qv..G,Wc!+6F6FQVZ^___kVVekFF,   'GFFF #)@'@1;@@ QV qv99<#799:::: ,In!! /;//!#+>!?!?!y):;;In55!#+>!?!?')y1J'K'K     D499$Q$Q[=N=N=P=P$Q$Q$QQQQRRR#. - sAG##G98G9c> ddlmtj t d} jdvrdStjdtj }tj }t||rtj|d|ddS fd }t"d z }tj|d d } dd lm}||d|sJtj|dnw#t*$rtj|d|YnRt,t.f$rtj|d|Yn't0$rtj|dYdSwxYwdS)avCheck the PyTorch Automatic Mixed Precision (AMP) functionality of a YOLO 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 (torch.nn.Module): A YOLO model instance. Returns: (bool): Returns True if the AMP functionality works correctly with YOLO model, else False. Examples: >>> from ultralytics import YOLO >>> from ultralytics.utils.checks import check_amp >>> model = YOLO("yolo26n.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.c|gdz}tdtjdz}|||ddjj}d5|||ddjj}d d d n #1swxYwY~|j|jko(t j||d S) zAll close FP32 vs AMP results.rWF)rwrirrT)enabledNg?)atol) rfrRrjboxesdatashaperlallclosefloat) mimbatchrwabrrir,s r+ amp_allclosezcheck_amp..amp_allclosens'qCU\--//!34455 Ae5 ? ? ? B H M Xd # # # R R%uVUCCCAFLQA R R R R R R R R R R R R R R R w!'!LenQ &L&L&LLs* BBBzbus.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 YOLO26n for AMP checks. zichecks skipped. Unable to load YOLO26n 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)rlrnext parametersrir rsr;compile IGNORECASErlrget_device_namerYsearchrrUrrrrConnectionErrorAttributeErrorrrerror) r,rpatterngpurr warning_msgrrris ` @@r+ check_amprHsn$766666 %""$$ % % ,F g  F {n$$u* egigt  j((00 s## $ $  N\\c\\\   5MMMMMMM ) B K6LLLMMMtK$$$$$$|DD..33333 v0001111 xxx&vvitvvwwwww / 0    v vhs v v        X X X   uu  4s14D&&#F )F6 FFcRddl} tj|jddddgdd}d tjd }|jdkrInstall your local copy in editable mode with 'pip install -e z?' to avoid issues. See https://docs.ultralytics.com/quickstart/z not foundz2Ultralytics not found via pip but importing from: z. zlocation:\s+(.+))flagsrzFMultiple Ultralytics installations detected. The `yolo` command uses: z$, but current session imports from: z$. This may cause version conflicts. )rrrrrrArrr rrUrr;rSrIrsamefilerT)rr install_msg yolo_paths r+rwrws}JJJufm Teiptuuu CT[ C C C    ! !fm113333iTXii\giijjj F"*%8&-rtTTTUWXYY\iirrtt !!$,..11  NmYbmm59mm_jmm        sA6D>BD D&%D&args dict | Nonec  d tjj}tj|\}}}}}9tj|\}}}fd|D t |t d}n$#t$rt |j }YnwxYw|r|dnd|r|dndz}tjt|d fdt#DzdS)zPrint 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. ct|tr7|dr"t|dkrt |n|S)zTClean longer Ultralytics HUB URLs by stripping potential authentication information.httpd)rkrDr9rur)rs r+ strip_authzprint_args..strip_authsH *1c 2 2dq||F7K7KdPSTUPVPVY\P\P\y|||cddr-Nc$i|] \}}|v || Sr'r')r)rrrs r+ zprint_args..s$:::AT 1 r-r$rrc3>K|]\}}|d|VdS)=Nr')r)rrrs r+r_zprint_args..s='`'`41a1(>(>zz!}}(>(>'`'`'`'`'`'`r-)inspect currentframef_back getframeinfo getargvaluesrrr relative_torr1rvr/rrr rsorted) r show_file show_funcr*rrfuncfrmrars ` @r+ print_argsrsgeee %A!.q11D!T1a |!.q11aC:::::::Dzz!!##//55AA"EE Dzz! )DrY.NkkkkBOA K dii'`'`'`'`6RVR\R\R^R^K_K_'`'`'````aaaaas4A CC"!C"rRc>trtjS t jgdd}|ddd}t|S#tj ttf$rYdSwxYw)zGet 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/r) rrlrrrrr8rIrRCalledProcessErrorrrv)r first_lines r+cuda_device_countrs z&&((( ,TTT_fF  --dA66q9Jz?? "-/@*M   11 sAA<<BBc&tdkS)zCheck if CUDA is available in the environment. Returns: (bool): True if one or more NVIDIA GPUs are available, False otherwise. r)rr'r-r+cuda_is_availablers    ""r-ctrtr td5}|}|d^}}|dddddt vr dddd S dddn #1swxYwYdSdS#t$rYd SwxYwd S) zCheck 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/compatiblerr$-r/rNTF)rropenreadrIrrOSError)r"dev_strrsocs r+ is_rockchiprs9    455 &&((!--,,C;;vr**00a88;zII         J                      55 us;B-A%B B- B-B""B-%B"&B-- B;:B;cddlm}d|vrdS tjddgddd}d|jvS#t $rYd SwxYw) zCheck if the system has Intel hardware (CPU or GPU). Returns: (bool): True if Intel hardware is detected, False otherwise. r)rintelTzxpu-smi discoveryr7)rrrF)rlrr rrrrT)rrs r+is_intelrs ;:::::,,..&&((((tK 8TXbcddd&---//// uus4A A+*A+ctrdSd}tj|dtjtjjdkS)zCheck if the sudo command is available in the environment. Returns: (bool): True if the sudo command is available, False otherwise. Fzsudo --versionT)rrrr)rrrDEVNULLr)rs r+rr s> u C >#T*2DZM_ ` ` ` kop ppr-z3.8)rrz3.9z3.10z3.12z3.13r6)rCrDrErD)rM)rErN)rErY)rcr/r0r)rMrMrVFFr$)rrDrrDr2rDrrYrrYrzrDrErY)r)r)rTF)rrDrrYrrYrErY)r)rrD)rrr$)T)rrDrrYrErD)r+)r,rDrEr-)r$TrT)rJT)rMrNr2rNrErY)F)Tr$)NTF)rr)rErR)b __future__rrp functoolsrCrrgrrr;rmrrr importlibrpathlibrtypesrrYnumpyr[rlrrrr r r r r rrrrrrrrrrrrrrrrrrr r!r"r#rArBrL lru_cacherXrbr|rrrrrrrr rrrr#r*r3rIrLrSrar|rrrwrrrrrrr9 IS_PYTHON_3_8 IS_PYTHON_3_9IS_PYTHON_3_10IS_PYTHON_3_12IS_PYTHON_3_13IS_PYTHON_MINIMUM_3_9IS_PYTHON_MINIMUM_3_10IS_PYTHON_MINIMUM_3_12r'r-r+r sR#"""""     !!!!!!  B"&/A!A2>       - - - -2222j   TTTTTn(, : ] ] ] ] ] $x$x $xN $(K2D$DbZ^eghhh hV'''QQQQQ!!!H ^ ^ ^ ^ 6     :.:.:.:.z / / / / u u u u  0!+!+!+!+H999xEEEP0bbbbb42####&( q q q q U----*)%00 ))%00 **622**622**622$ U777%f5999%f5999r-