a yfU @sddlZddlZddlZddlZddlmZddlmZddlm Z ddl m Z m Z ddl Z ddlZddlmZmZmZmZmZmZmZdZdd d Dd d d Dd d d Dd d dDdd dDdd dDdd dDdd dDdd dDdd dDdd dDdd dDdgdgZd d eDZd?d"d#Zd@d%d&ZdAd(d)ZdBd*d+Zd,e d-d'fd.d/Zd0d1Z dCd4d5Z!dDd7d8Z"dEd:d;Z#e d'd!d!dZ$dS)FN)repeat) ThreadPool)Path)parserequest)LOGGERTQDMchecks clean_urlemojis is_onlineurl2fileultralytics/assetscCs&g|]}dD]}d||dq qS))-cls-seg-pose-obbz-oiv7yolov8.pt.0ksuffixrrW/var/www/html/django/DPS/env/lib/python3.9/site-packages/ultralytics/utils/downloads.py rZnsmlxcCs&g|]}dD]}d||dq qS))rrrrrZyolo11rrrrrrrrcCs&g|]}dD]}d||dq qS))r6Zyolov5u.ptr)rr resolutionrrrrrcCsg|]}d|dqS)Zyolov3rrrrrrrrr)rz-sppz-tinycCsg|]}d|dqS)rz -world.ptrr!rrrrrZsmlxcCsg|]}d|dqS)rz -worldv2.ptrr!rrrrrcCsg|]}d|dqS)Zyolov9rrr!rrrrrZtsmcecCsg|]}d|dqS)Zyolov10rrr!rrrrrZnsmblxcCsg|]}d|dqS)Z yolo_nas_rrr!rrrrrsmlcCsg|]}d|dqS)Zsam_rrr!rrrrrblcCsg|]}d|dqS)zFastSAM-rrr!rrrrrsxcCsg|]}d|dqS)zrtdetr-rrr!rrrrrlxz mobile_sam.ptz:calibration_image_sample_data_20x128x128x3_float32.npy.zipcCsg|]}t|jqSr)rstemr!rrrr#rFc Csttt|}t|}t|j|jgs2J|rzt |*}| dkWdWdS1sp0YWddS1s0YdS)a( Validates if the given string is a URL and optionally checks if the URL exists online. Args: url (str): The string to be validated as a URL. check (bool, optional): If True, performs an additional check to see if the URL exists online. Defaults to False. Returns: (bool): Returns True for a valid URL. If 'check' is True, also returns True if the URL exists online. Returns False otherwise. Example: ```python valid = is_url("https://www.example.com") ``` NTF) contextlibsuppress Exceptionstrrurlparseallschemenetlocrurlopengetcode)urlcheckresultresponserrris_url&s   8 r6z .DS_StoreZ__MACOSXcCsH|D]>}tt||}td|d||D] }|q4qdS)ao Deletes all ".DS_store" files under a specified directory. Args: path (str, optional): The directory path where the ".DS_store" files should be deleted. files_to_delete (tuple): The files to be deleted. Example: ```python from ultralytics.utils.downloads import delete_dsstore delete_dsstore("path/to/dir") ``` Note: ".DS_store" files are created by the Apple operating system and contain metadata about folders and files. They are hidden system files and can cause issues when transferring files between different operating systems. z Deleting z files: N)listrrglobrinfounlink)pathZfiles_to_deletefilematchesfrrrdelete_dsstoreCs r@Tc sddlm}m}m}t|t|}|szrz+zip_directory...)is_filer-rexcluderIrrzrz!zip_directory..*.zipwzZipping  to ...r=descunitdisableN) zipfilerArBrCr@ris_dirFileNotFoundErrorr9 with_suffixrwrite relative_to) directorycompressrNprogressrArBrCZ files_to_zipzip_file compressionr?r=rrMr zip_directory]s  $2rccsddlm}m}m}t|r(||s8|d|d|durJt|j}||}fdd|D} dd | D} t| d k} | r|} t|t | d}nt|t|j }} |rt | r|st d |d |d |WdSt| d|dt|dd| dD]8} dt| jvrNt d| dq$|| | q$Wdn1st0Y|S)a Unzips a *.zip file to the specified path, excluding files containing strings in the exclude list. If the zipfile does not contain a single top-level directory, the function will create a new directory with the same name as the zipfile (without the extension) to extract its contents. If a path is not provided, the function will use the parent directory of the zipfile as the default path. Args: file (str): The path to the zipfile to be extracted. path (str, optional): The path to extract the zipfile to. Defaults to None. exclude (tuple, optional): A tuple of filename strings to be excluded. Defaults to ('.DS_Store', '__MACOSX'). exist_ok (bool, optional): Whether to overwrite existing contents if they exist. Defaults to False. progress (bool, optional): Whether to display a progress bar. Defaults to True. Raises: BadZipFile: If the provided file does not exist or is not a valid zipfile. Returns: (Path): The path to the directory where the zipfile was extracted. Example: ```python from ultralytics.utils.downloads import unzip_file dir = unzip_file("path/to/file.zip") ``` r) BadZipFilerC is_zipfilezFile 'z&' does not exist or is a bad zip file.Ncs&g|]tfddDrqS)c3s|]}|vVqdSrDrrGrIrrrJrz(unzip_file...)r-rLrMrIrrrzunzip_file..cSsh|]}t|jdqS)r)rparts)rr?rrr rzunzip_file..uWARNING ⚠️ Skipping z unzip as destination directory z is not empty. Unzipping rRrSr=rTz..z Potentially insecure file path: z, skipping extraction.)rXrdrCrerexistsparentnamelistlenr8r&anyiterdirrwarningrresolverfextract)r=r<rNexist_okr`rdrCreZzipObjfilesZtop_level_dirsZ unzip_as_dir extract_pathr?rrMr unzip_files,   ,0rvz(https://ultralytics.com/assets/coco8.zipg?c sz6t|}|jdks4Jd|d|jd|jWntyJYdS0dt|jdd}fd d t |D\}}}|||krdSd |d d ||dd|||d d} |rt | t | dS)aP Check if there is sufficient disk space to download and store a file. Args: url (str, optional): The URL to the file. Defaults to 'https://ultralytics.com/assets/coco8.zip'. path (str | Path, optional): The path or drive to check the available free space on. sf (float, optional): Safety factor, the multiplier for the required free space. Defaults to 1.5. hard (bool, optional): Whether to throw an error or not on insufficient disk space. Defaults to True. Returns: (bool): True if there is sufficient disk space, False otherwise. izURL error for :  Ti@Content-Lengthrc3s|]}|VqdSrDrrGZgibrrrJrz#check_disk_space..u,WARNING ⚠️ Insufficient free disk space z.1fz GB < z.3fz GB required, Please free z( GB additional disk space and try again.F) requestshead status_codereasonr*intheadersgetshutil disk_usage MemoryErrorrrp) r2r<Zsfhardrdatatotalusedfreetextrrzrcheck_disk_spaces$  ,    rc Cs|dddd}d|}d}t}|j|dd}d t|jvrdttd |d |j D] \}}| d rn|d |7}qn|j d}|rt d|d}Wdn1s0Y||fS)ax Retrieves the direct download link and filename for a shareable Google Drive file link. Args: link (str): The shareable link of the Google Drive file. Returns: (str): Direct download URL for the Google Drive file. (str): Original filename of the Google Drive file. If filename extraction fails, returns None. Example: ```python from ultralytics.utils.downloads import get_google_drive_file_info link = "https://drive.google.com/file/d/1cqT-cJgANNrhIHCrEufUYhQ4RqiWG_lJ/view?usp=drive_link" url, filename = get_google_drive_file_info(link) ``` z/d/rhz/viewrz/https://drive.google.com/uc?export=download&id=NT)streamzquota exceededui❌ Google Drive file download quota exceeded. Please try again later or download this file manually at .Zdownload_warningz &confirm=zcontent-dispositionzfilename="(.+)")splitr{Sessionrr+contentlowerConnectionErrorr cookiesitems startswithrrefindall) linkZfile_idZ drive_urlfilenamesessionr5rvcdrrrget_google_drive_file_infos(    .r?c  Cs|d} | rt|\}}t|p"d|p.t|} dt|vrVt|rVt|} nv| s| rh|nt|dd} d| d| d} t | d | j j d d d t || j d t |d D]}zj|s|dkr(d| }tddd|d|d| ddddg j}|dks Jd|nd}|dkrJtjj|| | dnt|}tt|dd| | dd dd`}t| d4}|D]}|||t|qWd n1s0YWd n1s0YWd n1s0Y| r4| j|kr,Wq| Wqt!y}zx|dkrrt"srt#t$d!| d"|n ||krt#t$d!| d#|t%d$|d d%|d&| d WYd }~qd }~00q|r~| r~| j&d'vr~dd(l'm(}|p| j )}|| r t*| ||| d)}nL| j&d*vrlt d+| d,|d tjd-| j&d.krZd/nd0| d1|gd d2|rz| |Sd S)3a Downloads files from a URL, with options for retrying, unzipping, and deleting the downloaded file. Args: url (str): The URL of the file to be downloaded. file (str, optional): The filename of the downloaded file. If not provided, the file will be saved with the same name as the URL. dir (str, optional): The directory to save the downloaded file. If not provided, the file will be saved in the current working directory. unzip (bool, optional): Whether to unzip the downloaded file. Default: True. delete (bool, optional): Whether to delete the downloaded file after unzipping. Default: False. curl (bool, optional): Whether to use curl command line tool for downloading. Default: False. retry (int, optional): The number of times to retry the download in case of failure. Default: 3. min_bytes (float, optional): The minimum number of bytes that the downloaded file should have, to be considered a successful download. Default: 1E0. exist_ok (bool, optional): Whether to overwrite existing contents during unzipping. Defaults to False. progress (bool, optional): Whether to display a progress bar during the download. Default: True. Example: ```python from ultralytics.utils.downloads import safe_download link = "https://ultralytics.com/assets/bus.jpg" path = safe_download(link) ``` zhttps://drive.google.com/r://z?https://github.com/ultralytics/assets/releases/download/v0.0.0/zhttps://ultralytics.com/assets/z Downloading z to ''rSTparentsrs)r<rhrZsScurlz-#-Lz-oz--retry3z-CzCurl return value torch)r`ryBi)rrUrWrVZ unit_scaleZ unit_divisorwbNu❌ Download failure for z. Environment is not online.z. Retry limit reached.u"⚠️ Download failure, retrying /rx>rrP.gz.tar)re)r=r<rsr`>rrrirRtarrZxfZxfzz --directory)r3)+rrrr r+rKr replacerr:rkmkdirrrange subprocessrun returncoderZhubZdownload_url_to_filerr0rr getheaderopenr\updatermrjstatst_sizer;r*r rr rprrXrerqrv)r2r=dirunzipdeleterretry min_bytesrsr`Zgdriver?urirUisrmethodr5ZpbarZf_openedrereZ unzip_dirrrr safe_downloadst&     (   r   :   (rlatestcCs|dkrd|}d|d|}t|}|jdkrN|jdkrN|rNt|}|jdkrtd|d|jd |jd gfS|}|d d d |dDfS)a Retrieve the specified version's tag and assets from a GitHub repository. If the version is not specified, the function fetches the latest release assets. Args: repo (str, optional): The GitHub repository in the format 'owner/repo'. Defaults to 'ultralytics/assets'. version (str, optional): The release version to fetch assets from. Defaults to 'latest'. retry (bool, optional): Flag to retry the request in case of a failure. Defaults to False. Returns: (tuple): A tuple containing the release tag and a list of asset names. Example: ```python tag, assets = get_github_assets(repo="ultralytics/assets", version="latest") ``` rztags/zhttps://api.github.com/repos/z /releases/r'zrate limit exceededu'⚠️ GitHub assets check failure for rwrxrZtag_namecSsg|] }|dqSrErrGrrrrrz%get_github_assets..assets)r{rr}r~rrpjson)repoversionrr2rrrrrget_github_assets}s     rv8.3.0c Ksddlm}t|}t|}t|dd}|rBt|S|d|rbt|d|Stt t|j }d|d}t| drt|d d }t |}t|rtd t|d |ntf||d d|n|tkr$|tvr$tf|d|d||d d|nPt||\}} | sDt|\}} || vrttf|d|d||d d|t|SdS)a  Attempt to download a file from GitHub release assets if it is not found locally. The function checks for the file locally first, then tries to download it from the specified GitHub repository release. Args: file (str | Path): The filename or file path to be downloaded. repo (str, optional): The GitHub repository in the format 'owner/repo'. Defaults to 'ultralytics/assets'. release (str, optional): The specific release version to be downloaded. Defaults to 'v8.3.0'. **kwargs (any): Additional keyword arguments for the download process. Returns: (str): The path to the downloaded file. Example: ```python file_path = attempt_download_asset("yolov8n.pt", repo="ultralytics/assets", release="latest") ``` r)SETTINGSrrZ weights_dirzhttps://github.com/z/releases/download)zhttp:/zhttps:/z:/rzFound z locally at gj@)r2r=rrN)ultralytics.utilsrr+r Zcheck_yolov5u_filenamerstriprrjrunquoterFrr rKrr:r rGITHUB_ASSETS_REPOGITHUB_ASSETS_NAMESr) r=rreleasekwargsrrF download_urlr2tagrrrrattempt_download_assets0    (  &rrhc st|}|jddddkrtH}|fddt|t|||Wdq1st0Yn4t|t tfr|gn|D]} t | |dqdS)a' Downloads files from specified URLs to a given directory. Supports concurrent downloads if multiple threads are specified. Args: url (str | list): The URL or list of URLs of the files to be downloaded. dir (Path, optional): The directory where the files will be saved. Defaults to the current working directory. unzip (bool, optional): Flag to unzip the files after downloading. Defaults to True. delete (bool, optional): Flag to delete the zip files after extraction. Defaults to False. curl (bool, optional): Flag to use curl for downloading. Defaults to False. threads (int, optional): Number of threads to use for concurrent downloads. Defaults to 1. retry (int, optional): Number of retries in case of download failure. Defaults to 3. exist_ok (bool, optional): Whether to overwrite existing contents during unzipping. Defaults to False. Example: ```python download("https://ultralytics.com/assets/example.zip", dir="path/to/dir", unzip=True) ``` Trrhc s$t|d|ddkdS)Nrrh)r2rrrrrrsr`)r)rHrrrsrthreadsrrrszdownload..N)r2rrrrrrs) rrrmapziprclosejoin isinstancer+r) r2rrrrrrrspoolurrrdownloads   (r)F)r7)Tr7T)Nr7FT) NNTFFrrFT)rrF)rr)%r(rrr itertoolsrZmultiprocessing.poolrpathlibrurllibrrr{rrrrr r r r r rrZGITHUB_ASSETS_STEMSr6r@rcrvcwdrrrrrrrrrrsv   $                 ' B&, g  6