Datasets
Models
Downloads: 1
39 lines (38 with data), 4.1 kB
B
¢eT ã @ s d Z ddlZddlZddlZddlmZ ddlZddlZdddZddd Z dd d Z
d
de e dddZ d ddZ
d ddZdS ) z
Download utils
é N)ÚPathTc C sf yDt | } tj | ¡}t|j|jgs*t|rBtj | ¡
¡ dkS dS ttjj fk
r` dS X d S )NéÈ TF) ÚstrÚurllibÚparseÚurlparseÚallÚschemeÚnetlocÚAssertionErrorÚrequestÚurlopenÚgetcodeÚ HTTPError)ÚurlÚcheckÚresult© r ú>/home/iml/Desktop/Talha/YOLOV5_Model/yolov5/utils/downloads.pyÚis_url s r Ú c C s. t jdd| gddd}|r*t| ¡ d S dS )NÚgsutilÚduTzutf-8)ÚshellÚencodingr )Ú
subprocessÚ check_outputÚintÚsplit)r Úoutputr r r Úgsutil_getsize s r ú&https://ultralytics.com/images/bus.jpgc C s t j| dd}t|j dd¡S )NT)Úallow_redirectszcontent-lengthéÿÿÿÿ)ÚrequestsÚheadr ÚheadersÚget)r Úresponser r r Ú url_getsize" s r) F)Úsilent)r* Úreturnc C s< |rdnd}t ddd| d| d|dd d
dg
¡}|jd kS ) z>
Download a file from a url to a filename using curl.
ÚsSr Úcurlz-#Ú-ÚLz--outputz--retryÚ9z-Cr )r ÚrunÚ
returncode)r Úfilenamer* Z
silent_optionÚprocr r r Ú
curl_download( s
r5 ç ð?c C s6 ddl m} t| } d| d| }zÀyX| d| d| d¡ tjj|t| |jt j
kd | ¡ rt| ¡ j
|ks|t|W nb tk
rà } zD| ¡ r | ¡ | d | d
|p²| d| d¡ t|pÊ|| W d d }~X Y nX W d | ¡ rþ| ¡ j
|k r&| ¡ r| ¡ | d | d | ¡ | d ¡ X d S )
Nr )ÚLOGGERzDownloaded file 'z(' does not exist or size is < min_bytes=z Downloading z to z...)ÚprogresszERROR: z
Re-attempting Ú
r )Ú
utils.generalr7 r ÚinfoÚtorchÚhubÚdownload_url_to_filer ÚlevelÚloggingÚINFOÚexistsÚstatÚst_sizer Ú ExceptionÚunlinkr5 )Úfiler Zurl2Ú min_bytesÚ error_msgr7 Z
assert_msgÚer r r Ú
safe_download; s" ""$
rK úultralytics/yolov5úv7.0c C s¢ ddl m} d dd}tt| ¡ dd¡} | ¡ sttj t| ¡j
}t| d¡r°t| d d
¡}| d ¡d } t|
¡ r| d | d
| ¡ nt| |dd | S dd dD }y|||\}}W n| tk
rL y||\}}W nT tk
rF y"tjddtjd ¡ ¡ d }W n tk
r@ |}Y nX Y nX Y nX ||kr| jjddd t| d| d| d| d| d| d | d t| S ) Nr )r7 Úlatestc S sF |dkrd| }t d| d| ¡ ¡ }|d dd |d D fS ) NrN ztags/z https://api.github.com/repos/z
/releases/Útag_namec S s g | ] }|d qS )Únamer )Ú.0Úxr r r ú
<listcomp>\ s z;attempt_download.<locals>.github_assets.<locals>.<listcomp>Úassets)r$ r' Újson)Ú
repositoryÚversionr( r r r Ú
github_assetsW s
z'attempt_download.<locals>.github_assetsú'r )zhttp:/zhttps:/z:/z://ú?zFound z locally at g jø@)rG r rH c S s& g | ] }d D ]}d| | dq qS ))r Ú6z-clsz-segÚyolov5z.ptr )rQ ÚsizeÚsuffixr r r rS l s z$attempt_download.<locals>.<listcomp>Únsmlxzgit tagT)r Ústderrr# )ÚparentsÚexist_okzhttps://github.com/z/releases/download/ú/z2 missing, try downloading from https://github.com/z
/releases/)r rH rI )rN )r: r7 r r ÚstripÚreplacerB r r ÚunquoterP Ú
startswithr Úis_filer; rK rE r r ÚSTDOUTÚdecodeÚparentÚmkdir) rG ÚrepoÚreleaser7 rX rP r rT Útagr r r Úattempt_downloadS s<
"
rp )T)r )r! )Nr6 r )rL rM )Ú__doc__r@ r r Úpathlibr r$ r< r r r) Úboolr5 rK rp r r r r Ú<module> s