Datasets
Models
Downloads: 1
271 lines (255 with data), 42.1 kB
B
%ĒYfH» „ @ sĖ d Z ddlZddlZddlZddlZddlZddlZddlZddlZddl Z ddl
Z
ddl Z ddl Z ddl
Z
ddlZddlZddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddl mZ dd lmZ dd
l m Z dd l m Z m Z ddl!Z!ddl"Z#ddl$Z%ddl&Z'ddl(Z(ddl)Z)ddl*Z*ddl+m,Z, dd l-m.Z. y ddl/Z/e0e/d
ÉźsVt1āW n, e2e1fk
źrĄ e†3d° ddl/Z/Y nX ddl4m5Z5 ddl6m7Z7m8Z8 ddl9m:Z:m;Z; ddl<m=Z=m>Z> ee?Ɇ@° ZAeAjBd ZCeDe†Edd°ÉZFeGdeHde†I° d ÉÉZJee†EdeCjKd °ÉZLeMe†Edd°É†N° dkZOeMe†Ed d°É†N° dkZPd ZQd ZRe(jSd d d!d"ć e#jSd d#d$jTid%ć d&e%jUjV_We!†Xd° eMeJÉejYd'< e†3° d(kźr¶d)neMeJÉejYd*< d+ejYd,< d„d.d/ĄZZdšd1d2ĄZ[d3d4Ą Z\d5d6Ą Z]d7d8Ą Z^e_d9úd:d;ĄZ`dŚd=d>ĄZad?Zbebdfd@dAĄZcecebÉ e†deb°Zee†3° dBkźr^x(eejfeejgfD ]ZheieeehjjdCdDĄ É źqBW dśdGdHĄZkekÉ ZlG dIdJĄ dJejmÉZnG dKdLĄ dLejmÉZoG dMdNĄ dNejmÉZpdOdPĄ ZqdÁe er dQúdRdSĄZsdŤdTdUĄZtdťdWdXĄZudYdZĄ ZvdÍd\d]ĄZwe?fd^d_ĄZxe?fd`daĄZydbdcĄ ZzdddeĄ Z{eCfdfdgĄZ|e7É epeCÉdŽdjdkĄÉÉZ}epeCÉdždldmĄÉZ~dŪdodpĄZdÓdsdtĄZÄdÔdvdwĄZĀdūdxdyĄZādŮd|d}ĄZÉdÚddÄĄZĄdůdĀdāĄZÖeRd<fdÉdĄĄZÜdŰdÖdÜĄZádádąĄ ZądűdädčĄZČdČi fdĆdćĄZädŲdŹdźĄZčdĎdíĄ ZĆdųdĒdēĄZćdĖdóĄ ZédėdôĄ ZŹdÝdĚděĄZźdüd†Ą ZĎdýdĘd£ĄZíd°e#†ďd°°fd§d•ĄZĒd¶dߥ Zēd®d©Ą ZĖd™dęĄ Zódķd≠dģĄZėdŻdĮdįĄZôdŁdĪd≤ĄZödżd≥dīĄZõdĶd∂Ą ZúdĢdłdĻĄZĚdˇdļdĽĄZěźd dľdĹĄZüdĺdŅĄ Z†dņdŃĄ Z°źddŇd∆ĄZĘźdd»d…ĄZ£eĎd ÉfdňdŐĄZ§dÕdőĄ Z•źddŌd–ĄZ¶e!jßZ®e!j©fd—d“ĄZ™d”d‘Ą Zęd’d÷Ą Zßee†¨° d j≠ÉjKjK†ģ° e†¨° d j≠kźr6e™eęeß e!_™e!_ęe!_ßd◊dōĄ ZĮdŔdॠZįdŘd‹Ą ZĪddl"Z#ddl≤m≥Zī ddlĶm∂Z∂ ddl(Z(źdd›dřĄZ∑dŖdŗĄ ZłdŠd‚Ą ZĻdS ( z
General utils
ť N)ŕdeepcopy)ŕdatetime)ŕrepeat)ŕ
ThreadPool)ŕPath)ŕ check_output)ŕ
is_tarfile)ŕOptional)ŕZipFileŕ
is_zipfile)ŕ roi_alignŕ __version__zpip install -U ultralytics)ŕcheck_requirements)ŕ TryExceptŕemojis)ŕ
curl_downloadŕgsutil_getsize)ŕbox_iouŕfitnessť ŕRANKťˇˇˇˇť ZYOLOv5_DATASETS_DIRŕdatasetsZYOLOv5_AUTOINSTALLTŕtrueZYOLOv5_VERBOSEz{l_bar}{bar:10}{r_bar}z Arial.ttfi@ ť ŕlong)ŕ linewidthŕ precisionŕprofileŕ
float_kindz{:11.5g})r ŕ formatterť
ZNUMEXPR_MAX_THREADSŕdarwinŕ1ZOMP_NUM_THREADSŕ2ZTF_CPP_MIN_LOG_LEVELŕ c C s$ t | É} t| †° †dd°Ét| ÉkS )Nŕasciiŕignore)ŕstrŕlenŕencodeŕdecode)ŕs© r. ķL/home/iml/Desktop/Talha/yolov5_P2_ROI_HCM_Micro_1000_Github/utils/general.pyŕis_asciiN s r0 ű šļļŚ∑•śôļŤÉĹc C s t t†dt| É°ÉS )Nu [šłÄ-ťŅŅ])ŕboolŕreŕsearchr) )r- r. r. r/ ŕ
is_chineseT s r5 c C s
dt jkS )Nz google.colab)ŕsysŕmodulesr. r. r. r/ ŕis_colabY s r8 c C s, t †t°Ź ddlm} | É dk S Q R X dS )zŕ
Check if the current script is running inside a Jupyter Notebook.
Verified on Colab, Jupyterlab, Kaggle, Paperspace.
Returns:
bool: True if running inside a Jupyter Notebook, False otherwise.
r )ŕ get_ipythonNF)ŕ
contextlibŕsuppressŕ ExceptionŕIPythonr9 )r9 r. r. r/ ŕ
is_jupyter^ s r> c C s t j†d°dko t j†d°dkS )NZPWDz/kaggle/workingZKAGGLE_URL_BASEzhttps://www.kaggle.com)ŕosŕenvironŕgetr. r. r. r/ ŕ is_kagglel s rB )ŕreturnc C sP t dɆ° rdS y&tdÉŹ} tddĄ | D ÉÉS Q R X W n tk
rJ dS X dS )z4Check if the process runs inside a docker container.z /.dockerenvTz/proc/self/cgroupc s s | ]}d |kV qdS )ZdockerNr. )ŕ.0ŕliner. r. r/ ķ <genexpr>w s z is_docker.<locals>.<genexpr>NF)r ŕexistsŕopenŕanyŕOSError)ŕfiler. r. r/ ŕ is_dockerq s
rL Fc C sZ |st †| t j°S t| Éd }y"t|dÉŹ W d Q R X |†° dS tk
rT dS X d S )Nztmp.txtŕwTF)r? ŕaccessŕW_OKr rH ŕunlinkrJ )ŕdirŕtestrK r. r. r/ ŕ is_writeable| s
rS Zyolov5c C sf t t†dd°É}|r"|dkr"tjntj}tj†dd| ddii| d| |d úi| || gdd
úid ú° d S ) Nr r > r r r Fŕformatz %(message)szlogging.StreamHandler)ŕclassr! ŕlevel)rV ŕhandlersŕ propagate)ŕversionZdisable_existing_loggersZ
formattersrW ŕloggers)ŕintr? ŕgetenvŕloggingŕINFOŕERRORŕconfigZ
dictConfig)ŕnameŕverboseŕrankrV r. r. r/ ŕ set_loggingć s rd ŕWindowsc C s t t| ÉÉS )N)ŕfnr )ŕxr. r. r/ ŕ<lambda>ß ů rh ŕ UltralyticsŕYOLOV5_CONFIG_DIRc C sd t †|°}|rt|É}n<ddddú}t†° |†t†° d° }t|ÉrH|ntdÉ| }|jddć |S ) NzAppData/Roamingz.configzLibrary/Application Support)re ŕLinuxŕDarwinr& z/tmpT)ŕexist_ok) r? r\ r ŕhomerA ŕplatformŕsystemrS ŕmkdir)rQ ŕenv_varŕenvŕpathŕcfgr. r. r/ ŕuser_config_dir™ s
rw c @ s6 e Zd Zd ejdúddĄZddĄ Zdd Ą Zd
d Ą ZdS )
ŕProfileÁ N)ŕdevicec C s2 || _ || _|r(t|Éd dÖ dkr(dnd| _d S )Nť ŕcudaTF)ŕtrz r) r| )ŕselfr} rz r. r. r/ ŕ__init__ľ s zProfile.__init__c C s | † ° | _| S )N)ŕtimeŕstart)r~ r. r. r/ ŕ __enter__Ń s
zProfile.__enter__c C s$ | † ° | j | _| j| j7 _d S )N)rÄ rĀ ŕdtr} )r~ ŕtypeŕvalueŕ tracebackr. r. r/ ŕ__exit__Ň s zProfile.__exit__c C s | j rtj †| j° t†° S )N)r| ŕtorchŕ synchronizerz rÄ )r~ r. r. r/ rÄ … s z Profile.time)ry N) ŕ__name__ŕ
__module__ŕ __qualname__rą rz r rā rá rÄ r. r. r. r/ rx ļ s rx c @ s4 e Zd ZdddúddĄZddĄ Zdd Ą Zd
d Ą Zd S )
ŕTimeoutr& T)ŕ timeout_msgŕsuppress_timeout_errorsc C s t |É| _|| _t|É| _d S )N)r[ ŕsecondsŕtimeout_messager2 r; )r~ rź ré rŹ r. r. r/ r — s
zTimeout.__init__c C s t | jÉād S )N)ŕ TimeoutErrorrĎ )r~ ŕsignumŕframer. r. r/ ŕ_timeout_handler÷ s zTimeout._timeout_handlerc C s, t †° dkr(t†tj| j° t†| j° d S )Nre )rp rq ŕsignalŕSIGALRMrē ŕalarmrź )r~ r. r. r/ rā Ŕ s zTimeout.__enter__c C s, t †° dkr(t†d° | jr(|tkr(dS d S )Nre r T)rp rq rĖ rė r; rí )r~ ŕexc_typeŕexc_valŕexc_tbr. r. r/ rá ř s
zTimeout.__exit__N)rä rč rĆ r rē rā rá r. r. r. r/ rć Ō s rć c @ s$ e Zd ZddĄ ZddĄ ZddĄ ZdS )ŕWorkingDirectoryc C s || _ t†° †° | _d S )N)rQ r ŕcwdŕresolve)r~ Znew_dirr. r. r/ r Á s zWorkingDirectory.__init__c C s t †| j° d S )N)r? ŕchdirrQ )r~ r. r. r/ rā Ž s zWorkingDirectory.__enter__c C s t †| j° d S )N)r? rü rĚ )r~ rô rö rõ r. r. r/ rá Ó s zWorkingDirectory.__exit__N)rä rč rĆ r rā rá r. r. r. r/ rú Ś s rú c s á fddĄt ą ÉD ÉS )Nc s( g | ] }t tą |ÉÉr|†d °s|ĎqS )ŕ__)ŕcallableŕgetattrŕ
startswith)rD ŕf)ŕinstancer. r/ ķ
<listcomp>Ű s zmethods.<locals>.<listcomp>)rQ )r• r. )r• r/ ŕmethodsÚ s rß )ŕargsc sō t †° j}t †|°\}}}}}ą d krNt †|°\Č }}}á fddĄ|†° D ÉČ y t|Ɇ° †t °†
d°}W n t k
rą t|Éj }Y nX |rė|õ dĚnd|r®|õ dĚnd }t
†t|Éd†ddĄ ą †° D É° ° d S )Nc s i | ]\}}|ą kr||ďqS r. r. )rD ŕkŕv)r® r. r/ ķ
<dictcomp>ż s z print_args.<locals>.<dictcomp>r& z: z, c s s | ]\}}|õ d |õ ĚV qdS )ķ=Nr. )rD r© r™ r. r. r/ rF s z print_args.<locals>.<genexpr>)ŕinspectŕ currentframeŕf_backŕ getframeinfoŕ getargvaluesŕitemsr rě ŕ relative_toŕROOTŕ with_suffixŕ
ValueErrorŕstemŕLOGGERŕinfoŕcolorstrŕjoin) r® Z show_fileZ show_funcrg rK ŕ_ŕfuncZfrmr- r. )r® r/ ŕ
print_argsų s
$rĺ c C sx t †| ° tj †| ° t†| ° tj†| ° tj†| ° |rtttjdÉrtt† d° dtj
j _ dt
jd< t| Ét
jd< d S )Nz1.12.0Tz:4096:8ZCUBLAS_WORKSPACE_CONFIGZPYTHONHASHSEED)ŕrandomŕseedŕnprą ŕ manual_seedr| ŕmanual_seed_allŕ
check_versionr
ŕ use_deterministic_algorithmsŕbackendsŕcudnnŕ
deterministicr? r@ r) )rņ r» r. r. r/ ŕ
init_seeds s
r… r. c s á áfddĄ| † ° D ÉS )Nc sD i | ]<\Č }ą ąkrt á fd dĄąD ÉÉr|jąą jkr|ą ďqS )c 3 s | ]}|ą kV qd S )Nr. )rD rg )r© r. r/ rF s z-intersect_dicts.<locals>.<dictcomp>.<genexpr>)ŕallŕshape)rD r™ )ŕdbŕexclude)r© r/ rę s z#intersect_dicts.<locals>.<dictcomp>)r≤ )ŕdarŐ rÕ r. )rŐ rÕ r/ ŕintersect_dicts s rŌ c C s t †| °}ddĄ |j†° D ÉS )Nc S s& i | ] \}}|j tjjk r|j |ďqS r. )ŕdefaultr≠ ŕ Parameterŕempty)rD r© r™ r. r. r/ rę s z$get_default_args.<locals>.<dictcomp>)r≠ ŕ signatureŕ
parametersr≤ )rĹ r” r. r. r/ ŕget_default_args s
r’ ŕ.c C s, t j | õ dĚddć}|r(t|tjjdćS dS )Nz /**/last*.ptT)ŕ recursive)ŕkeyr& )ŕglobŕmaxr? ru ŕgetctime)Z
search_dirZ last_listr. r. r/ ŕget_latest_run s r‹ c C s" t †° t †t| Ɇ° j° }|jS )N)r ŕnowŕ
fromtimestampr ŕstatŕst_mtimeŕdays)ru rÉ r. r. r/ ŕfile_age& s r‚ c C s. t †t| Ɇ° j°}|jõ d|jõ d|jõ ĚS )Nķ-)r rř r rŖ rŗ ŕyearŕmonthŕday)ru r} r. r. r/ ŕ file_date, s rÁ c C sN d}t | É} | †° r"| †° j| S | †° rFtddĄ | †d°D ÉÉ| S dS d S )Ni c s s | ]}|† ° r|†° jV qd S )N)ŕis_filerŖ ŕst_size)rD r§ r. r. r/ rF 9 s z file_size.<locals>.<genexpr>z**/*g )r rŤ rŖ rť ŕis_dirŕsumrŔ )ru ŕmbr. r. r/ ŕ file_size2 s rŪ c s dd l Č á fddĄ} | É p | É S )Nr c s* yą † dd° dS tk
r$ dS X d S )N)z1.1.1.1iĽ r TF)ŕcreate_connectionrJ r. )ŕsocketr. r/ ŕrun_onceB s
z check_online.<locals>.run_once)rÔ )rū r. )rÔ r/ ŕ check_online> s rŮ c C sN y4t | Éd †° stātd| õ dĚdd憰 d dÖ S tk
rH dS X d S )Nz.gitzgit -C z describe --tags --long --alwaysT)ŕshellr r& )r rÍ ŕAssertionErrorr r, r< )ru r. r. r/ ŕ git_describeM s
rŰ ķultralytics/yolov5ŕmasterc
sp dą õ Ě}d|õ Ě}t dÉ}tdɆ° s8t|d | ÉātÉ sNt|d | Éātjdtdd d
憰 d ć}á fd d
Ą|D É}t |ÉrĖ||†
d °d }n d}td|õ d|õ Ěd d
ć td|õ Ěd ddć tdd d
憰 † ° }t td|õ d|õ d|õ dĚd d
ćÉ} | dkźrR|dkźrd nd |õ d|õ Ě}
|d | õ d d | dk õ d!|
õ d"|õ d#Ě 7 }n|d$|õ d%Ě7 }t
†|° d S )&Nzhttps://github.com/z, for updates see zgithub: z.gitz%skipping check (not a git repository)zskipping check (offline)z\sz
git remote -vT)rÚ )ŕpatternŕstringc s g | ] }ą |kĎqS r. r. )rD r- )ŕrepor. r/ r¶ a s z$check_git_status.<locals>.<listcomp>r ŕ ultralyticszgit remote add ķ z
git fetch r )rÚ ŕtimeoutz git rev-parse --abbrev-ref HEADz
git rev-list z..ķ/z --countr ŕoriginzgit pullz git pull u ‚ö†ÔłŹ YOLOv5 is out of date by z commitr- z. Use 'z' or 'git clone z ' to update.zup to date with u ‚úÖ)rļ r rG rů rŮ r3 ŕsplitr r, rI ŕindexŕstripr[ rł rĻ ) rý ŕbranchŕurlŕmsgr- ŕsplitsŕmatchesŕremoteZ local_branchŕnZpullr. )rý r/ ŕcheck_git_statusV s&
$
,r c C sé t dÉ dd l}yX|†| °}|jjj†dd°}|jjj }y |j
j }W n t k
r\ d }Y nX |||dúS |j
jk
rą d d d dúS X d S )NZ gitpythonr z.gitr& )r r ŕcommit)r ŕgitZRepoZremotesrĢ r ŕreplaceŕheadr
ZhexshaZ
active_branchra ŕ TypeErrorŕexcZInvalidGitRepositoryError)ru r rý r r
r r. r. r/ ŕcheck_git_infor s
r ķ3.8.0c C s t t†° | dddć d S )NzPython T)ra ŕhard)rń rp ŕpython_version)ŕminimumr. r. r/ ŕ check_pythonĄ s r ķ0.0.0ķversion c C sn ddĄ | |fD É\} }|r"| |kn| |k}d|õ |õ d|õ | õ dĚ}|rX|sXt t|ÉÉā|rj|sjt†|° |S )Nc s s | ]}t †|°V qd S )N)ŕpkgZ
parse_version)rD rg r. r. r/ rF č s z check_version.<locals>.<genexpr>u WARNING ‚ö†ÔłŹ z is required by YOLOv5, but z is currently installed)rů r rł ŕwarning)ŕcurrentr ra Zpinnedr rb ŕresultr- r. r. r/ rń Č s
rń ť c sd t | tÉr tt| tąÉÉą É}n t| É} á áfddĄ| D É}|| kr`t†d| õ dąõ d|õ Ě° |S )Nc s g | ]}t t|tąÉÉą ÉĎqS r. )rŕ ŕmake_divisibler[ )rD rg )ŕfloorr- r. r/ r¶ õ s z"check_img_size.<locals>.<listcomp>u WARNING ‚ö†ÔłŹ --img-size z must be multiple of max stride z, updating to )ŕ
isinstancer[ rŕ r ŕlistrł r )ŕimgszr- r ŕnew_sizer. )r r- r/ ŕcheck_img_sizeē s
r# c
C sā yFt É r tātÉ rtāt†dt†d°° t†d° t†° t†d° dS t k
r| } z| rlt
† d|õ Ě° dS d }~X Y nX d S )NrR )r r ť r TuM WARNING ‚ö†ÔłŹ Environment does not support cv2.imshow() or PIL Image.show()
F) r> rů rL ŕcv2ŕimshowrŃ ŕzerosZwaitKeyZdestroyAllWindowsr< rł r )ŕwarnŕer. r. r/ ŕ check_imshow° s
r* ķ
yolov5s.pt©z.ptc C sr | rn|rnt |tÉr|g}xTt | ttfÉr,| n| gD ]8}t|Éj†° }t|Ér2||ks2t|õ |õ d|õ ĚÉāq2W d S )Nz acceptable suffix is ) r r) r ŕtupler ŕsuffixŕlowerr* rů )rK r. r r§ r- r. r. r/ ŕ check_suffixĪ s
r0 ©z.yamlz.ymlc C s
t | |ÉS )N)ŕ
check_file)rK r. r. r. r/ ŕ
check_yamlľ s r3 c C sf t | |É t| É} tj†| °s"| s&| S | †d°r | }ttj† | °†
d°d Éj } tj†| °rtt †
d|õ d| õ Ě° nRt †
d|õ d| õ dĚ° tj†|| ° t| Ɇ° rłt| Ɇ° jdks∆td |õ ĚÉā| S | †d
°rÍd tjksśtd Éā| S g }x0d
D ](}|†tjtt| d | Éddć° qŰW t|Éźs8td| õ ĚÉāt|ÉdkźsZtd| õ d|õ ĚÉā|d S d S )N)zhttp:/zhttps:/ķ?r zFound z locally at z Downloading z to z...zFile download failed: z
clearml://Zclearmlz[ClearML is not installed, so cannot use ClearML dataset. Try running 'pip install clearml'.)ŕdataZmodelsŕutilsz**T)r◊ zFile not found: r zMultiple files match 'z', specify exact path: )r0 r) r? ru ŕisfiler£ r ŕurllibŕparseŕunquoterˇ ra rł rĻ rą ŕhubŕdownload_url_to_filerG rŖ rť rů r6 r7 ŕextendrŔ rī r* )rK r. r ŕfilesŕdr. r. r/ r2 Ń s,
,
("r2 c C s` t | É} t| j }| †° s\|†° s\d| jõ Ě}t†d|õ d|õ dĚ° tjj|t |É|dć d S )Nz https://ultralytics.com/assets/z Downloading z to z...)ŕprogress)
r ŕ
CONFIG_DIRra rG rł rĻ rą r; r< r) )Zfontr@ rK r r. r. r/ ŕ
check_font› s
rB c
sĄ d}t ą ttfÉrjtą És"tą Érjtą tõ dtą Éjõ Ědddddć tttą Éj † d°ÉČ ą j
d }}t ą ttfÉrÄt ą ÉČ x(dD ] }|ą ksÜt t
d |õ d
ĚÉÉāqÜW t ą d ttfÉr–ttą d ÉÉą d < td d
Ą ą d †° D ÉÉsÚt dÉātą d Éą d< t|źpą †d°źpdÉČą†° źs:tą †° Čąą d< xĒdD ]Ć}ą †|°źr@t ą | tÉźrįąą | †° }|†° źsĘą | †d°źrĘąą | dd Ö †° }t|Éą |< náfddĄą | D Éą |< źq@W á fdd
ĄdD É\}}}}|źrbddĄ t |tÉźr|n|gD É}tdd
Ą |D ÉÉźsbt† dddĄ |D É ° |źrH|źsPt d Éāt † ° } |†d °źr–|† d °źr–t|Éj }
t† d |õ d |
õ d!Ě° t!j"†#||
° ttÉj$ddd"ć t%|
td#ć t|
Ɇ&° d } n<|†d$°źrĢt† d%|õ d&Ě° t'j(|dd'ć} nt)|d(ą iÉ} d)t*t † ° | dÉõ d*Ě} | d+kźrFd,| õ d-t+d.tÉõ Ěn
d/| õ d0Ě}t† d1|õ Ě° t,t-ą d Éźrvd2nd3dd4ć ą S )5Nr& rż TFr )rQ ŕunzipŕdeleteŕcurlŕthreadsz*.yaml)ŕtrainŕvalŕnamesz data.yaml 'u :' field missing ‚ĚĆrI c s s | ]}t |tÉV qd S )N)r r[ )rD r© r. r. r/ rF ķ s z check_dataset.<locals>.<genexpr>z2data.yaml names keys must be integers, i.e. 2: carŕncru )rG rH rR z../r$ c s g | ]}t ą | †° ÉĎqS r. )r) rě )rD rg )ru r. r/ r¶
s z!check_dataset.<locals>.<listcomp>c 3 s | ]}ą † |°V qd S )N)rA )rD rg )r5 r. r/ rF
s )rG rH rR ŕdownloadc S s g | ]}t |Ɇ° ĎqS r. )r rě )rD rg r. r. r/ r¶ s c s s | ]}|† ° V qd S )N)rG )rD rg r. r. r/ rF s u+
Dataset not found ‚ö†ÔłŹ, missing paths %sc S s g | ]}|† ° st|ÉĎqS r. )rG r) )rD rg r. r. r/ r¶ s u Dataset not found ‚ĚĆŕhttpz.zipz Downloading z to z...)ŕparentsrn )ru zbash zRunning z ...)rÚ ŕyamlķ(zs))r Nu success ‚úÖ z , saved to ŕboldzfailure u ‚ĚĆzDataset download z Arial.ttfzArial.Unicode.ttf)r@ ).r r) r r r rK ŕ DATASETS_DIRr∑ ŕnextŕrglobŕparentŕ yaml_loadrů r r r- ŕdictŕ enumerater ŕkeysr* rA ŕ is_absoluterī rě rG r£ rł rĻ r< rÄ ŕendswithra rą r; r< rr ŕ
unzip_filerP ŕ
subprocessŕrunŕexecŕroundrļ rB r0 )
r5 Z autodownloadŕ extract_dirr© rg rG rH rR r- r} r§ ŕrrÉ r. )r5 ru r/ ŕ
check_datasetÁ sd $
"
, rb c sō ddl mČ m} á fddĄ}tdÉ}t| †° Éj}|jdkr@dS td d d
}|† ° r\|nt
É rfd nt † d °}y6|t
| É|ÉsĒ||d
|É|ÉsĒtāt†|õ dĚ° dS tk
r“ d}t†|õ d|õ Ě° dS X d S )Nr )ŕ AutoShapeŕDetectMultiBackendc sJ ą | ddć}||Éj d }d|_||Éj d }|j|jkoHtj||ddćS )NF)rb r TgöôôôôôĻ?)ŕatol)Zxywhnŕamprň rą ŕallclose)ŕmodelŕimŕmŕaŕb)rc r. r/ ŕ amp_allclose- s
z check_amp.<locals>.amp_allclosezAMP: )ŕcpuŕmpsFr5 Zimageszbus.jpgz&https://ultralytics.com/images/bus.jpg)iÄ iÄ r$ z
yolov5n.ptu checks passed ‚úÖTz1https://github.com/ultralytics/yolov5/issues/7908u< checks failed ‚ĚĆ, disabling Automatic Mixed Precision. See )Z
models.commonrc rd rļ rR r‘ rz rĄ rī rG rŮ rŃ ŕonesr rů rł rĻ r< r )rh rd rm ŕprefixrz r§ ri Zhelp_urlr. )rc r/ ŕ check_amp) s
"rr ķ data.yamlc C s" t | ddćŹ }t†|°S Q R X d S )Nr( )ŕerrors)rH rN ŕ safe_load)rK r§ r. r. r/ rU E s rU c C s8 t | dÉŹ$}tjddĄ |†° D É|ddć W d Q R X d S )NrM c S s( i | ] \}}t |tÉr t|Én||ďqS r. )r r r) )rD r© r™ r. r. r/ rę N s z yaml_save.<locals>.<dictcomp>F)ŕ sort_keys)rH rN ŕ safe_dumpr≤ )rK r5 r§ r. r. r/ ŕ yaml_saveK s rx ©z .DS_StoreZ__MACOSXc s` |d krt | Éj}t| ÉŹ<}x4|†° D ](Č tá fddĄ|D ÉÉr&|ją |dć q&W W d Q R X d S )Nc 3 s | ]}|ą kV qd S )Nr. )rD rg )r§ r. r/ rF W s z unzip_file.<locals>.<genexpr>)ru )r rT r
ŕnamelistr ŕextract)rK ru rÕ ZzipObjr. )r§ r/ r[ Q s
r[ c C s0 t t| ÉɆdd°} ttj†| °Éj†d°d S )Nz:/z://r4 r )r) r r r8 r9 r: ra rˇ )r r. r. r/ ŕurl2file[ s r| r$ c sö á ááááfddĄČt |É}|jdddć ądkrjtąÉ}|†áfddĄt| t|ÉÉ° |†° |†° n,x*t| t t fÉrÄ| gn| D ]}ą||É qĄW d S )Nc
sÄ d}t j†| °rt| É}nł|t| Éj }t†d| õ d|õ dĚ° xźtąd ÉD ]Ä}ą rjt| |ądkdć}n t j
j | |ądkdć |† ° }|rźP qN|ąk rľt†
d|d õ d ąõ d
| õ dĚ° qNt†
d | õ dĚ° qNW ąźr||źr||jd ksķt|Ésķt|Éźr|t†d
|õ dĚ° t|Éźr"t||É nLt|ÉźrHtjdd|d|jgddć n&|jd kźrntjdd|d|jgddć ąźr||†° d S )NTz Downloading z to z...r )Zsilent)r@ u" ‚ö†ÔłŹ Download failure, retrying rż rŻ u ‚ĚĆ Failed to download z.gzz
Unzipping ŕtarŕxfz --directory)ŕcheckZxfz)r? ru r7 r ra rł rĻ ŕranger rą r; r< rŤ r r. r r r[ r\ r] rT rP )r rQ ŕsuccessr§ ŕi)rE rD ŕretryrF rC r. r/ ŕ download_onec s2
$(
z download.<locals>.download_oneT)rM rn r c s ą | é S )Nr. )rg )rĄ r. r/ rh á ri zdownload.<locals>.<lambda>)
r rr r ŕimapŕzipr ŕcloserĽ r r) ) r rQ rC rD rE rF rÉ ŕpoolŕur. )rE rD rĄ rÉ rF rC r/ rK a s
rK c C s* t |tjÉrt|†° É}t†| | °| S )N)r rą ŕTensorr[ rŕ ŕmathŕceil)rg ŕdivisorr. r. r/ r Ź s r c C s t jdd| dćS )Nu" [|@#!¬°¬∑$‚ā¨%&()=?¬Ņ^*;:,¬®¬ī><+]rľ )rų ŕreplrÝ )r3 ŕsub)r- r. r. r/ ŕ clean_strĖ s rź Á Á ū?ťd c s á ááfddĄS )Nc s( dt †| t j ą ° d ąą ą S )Nr ť )rč ŕcosŕpi)rg )ŕstepsŕy1ŕy2r. r/ rh Ě ri zone_cycle.<locals>.<lambda>r. )rė rô ró r. )ró rė rô r/ ŕ one_cycleõ s rö c sv t | Édkr| n dd| d fź^ }}ddddd d
d d d
dddddddddddúČ d†á fddĄ|D É°|õ ą d S ) Nr ŕbluerP r z[30mz[31mz[32mz[33mz[34mz[35mz[36mz[37mz[90mz[91mz[92mz[93mz[94mz[95mz[96mz[97mz[0mz[1mz[4m)ZblackZredZgreenZyellowrõ ZmagentaZcyanZwhiteZ bright_blackZ
bright_redZ bright_greenZ
bright_yellowZ bright_blueZbright_magentaZ bright_cyanZ bright_whiteŕendrP Z underliner& c 3 s | ]}ą | V qd S )Nr. )rD rg )ŕcolorsr. r/ rF ∑ s zcolorstr.<locals>.<genexpr>rú )r* rĽ )ŕinputr® rÝ r. )rĚ r/ rļ † s* $rļ ťP c C sr | d d krt †° S t†| d°} | d d Ödf †t°}tj||dć}d||dk< d| }||†° }t †|°† ° S )Nr )ŕ minlengthr )
rą rä rŃ ŕ concatenateŕastyper[ ŕbincountrŽ ŕ
from_numpyŕfloat)ŕlabelsrJ ŕclassesŕweightsr. r. r/ ŕlabels_to_class_weightsļ s r© c s. t †á fddĄ| D É°}|†dą °| †d°S )Nc s, g | ]$}t j|d d Ödf †t°ą dćĎqS )Nr )r† )rŃ r£ rĘ r[ )rD rg )rJ r. r/ r¶ – s z+labels_to_image_weights.<locals>.<listcomp>r )rŃ ŕarrayŕreshaperŽ )r¶ rJ ŕ
class_weightsZ class_countsr. )rJ r/ ŕlabels_to_image_weightsÕ s r≠ c P C s§ ddddddddd d
d d d
ddddddddddddddd d d d d d!d"d#d$d%d&d'd(d)d*d+d,d-d.d/d0d1d2d3d4d5d6d7d8d9d:d;d<d=d>d?d@dAdBdCdDdEdFdGdHdIdJdKdLdMdNdOdPgPS )QNr rĒ r$ r{ r ť ť r ť r" ť ť
ť ť ť ť ť ť ť ť ť ť ť ť ť ť ť r ť! ť" ť# ť$ ť% ť& ť' ť( ť) ť* ť+ ť, ť. ť/ ť0 ť1 ť2 ť3 ť4 ť5 ť6 ť7 ť8 ť9 ť: ť; ť< ť= ť> ť? ť@ ťA ťC ťF ťH ťI ťJ ťK ťL ťM ťN ťO rü ťQ ťR ťT ťU ťV ťW ťX ťY ťZ r. r. r. r. r/ ŕcoco80_to_coco91_class‘ s <8rű c C sz t | tjÉr| †° nt†| °}| d | d d |d< | d | d d |d< | d | d |d< | d | d |d< |S )N).r ).rĒ rĒ ).r ).r$ )r rą rä ŕclonerŃ ŕcopy)rg ŕyr. r. r/ ŕ xyxy2xywhŗ s rý c C sā t | tjÉr| †° nt†| °}| d | d d |d< | d | d d |d< | d | d d |d< | d | d d |d< |S )N).r ).rĒ rĒ ).r ).r$ )r rą rä rŲ rŃ rų )rg rÝ r. r. r/ ŕ xywh2xyxyÍ s rķ ťÄ c C sĘ t | tjÉr| †° nt†| °}|| d | d d | |d< || d | d d | |d< || d | d d | |d< || d | d d | |d< |S )N).r ).rĒ rĒ ).r ).r$ )r rą rä rŲ rŃ rų )rg rM ŕhŕpadwŕpadhrÝ r. r. r/ ŕ
xywhn2xyxyŰ s rˇ c C s§ |rt | || || fÉ t| tjÉr.| †° nt†| °}| d | d d | |d< | d | d d | |d< | d | d | |d< | d | d | |d< |S )N).r ).rĒ rĒ ).r ).r$ )ŕ
clip_boxesr rą rä rŲ rŃ rų )rg rM rŁ ŕclipŕepsrÝ r. r. r/ ŕ
xyxy2xywhnĢ s r c C sJ t | tjÉr| †° nt†| °}|| d | |d< || d | |d< |S )N).r ).r )r rą rä rŲ rŃ rų )rg rM rŁ rż rĢ rÝ r. r. r/ ŕxyn2xy
s r c C sp | j \}}|dk|dk@ ||k@ ||k@ }|| || }}t|Érft†|†° |†° |†° |†° g°S t†d°S )Nr )r r{ )ŕTrI rŃ r™ ŕminrŕ r' )ŕsegmentŕwidthŕheightrg rÝ Zinsider. r. r/ ŕ segment2box s
r
c C sL g }x8| D ]0}|j \}}|†|†° |†° |†° |†° g° q
W tt†|°ÉS )N)r ŕappendr rŕ rý rŃ r™ )ŕsegmentsŕboxesr- rg rÝ r. r. r/ ŕsegments2boxes s
&r ťŤ c sé xąt | ÉD ]|\}Č tją ą ddÖd d Öf fddćČ t†dtą Éd |°Čt†tą É°Čt†á ááfddĄtdÉD É°†dd°j| |< q
W | S )Nr r )ŕaxisc s& g | ] }t †ąąą d d Ö|f °ĎqS )N)rŃ ŕinterp)rD rā )r- rg ŕxpr. r/ r¶ ) s z%resample_segments.<locals>.<listcomp>rĒ r ) rW rŃ r° ŕlinspacer* ŕarangerÄ rę r )r r rā r. )r- rg r r/ ŕresample_segments# s "2r c C ső |d kr\t | d |d | d |d É}| d |d | d | d |d | d f}n|d d }|d }|dddgf |d 8 < |dddgf |d 8 < |dd dÖf | < t||É |S )Nr r rĒ .r$ r{ )r r )ŕ
img1_shaper
ŕ
img0_shapeŕ ratio_padŕgainŕpadr. r. r/ ŕ scale_boxes- s "2
r c C sķ |d kr\t | d |d | d |d É}| d |d | d | d |d | d f}n|d d }|d }|d d Ödf |d 8 < |d d Ödf |d 8 < || }t||É |rŲ|d d Ödf |d < |d d Ödf |d < |S )Nr r rĒ )r ŕ
clip_segments)r r r r ŕ normalizer r r. r. r/ ŕscale_segments= s "2
r c C s≤ t | tjÉr^| d †d|d ° | d †d|d ° | d †d|d ° | d †d|d ° nP| dddgf †d|d °| dddgf< | ddd gf †d|d °| ddd gf< d S )
N).r r r ).r ).rĒ ).r$ .rĒ r$ )r rą rä ŕclamp_r )r
rň r. r. r/ r P s (r c C sö t | tjÉrF| d d Ödf †d|d ° | d d Ödf †d|d ° nP| d d Ödf †d|d °| d d Ödf< | d d Ödf †d|d °| d d Ödf< d S )Nr r )r rą rä r r )r rň r. r. r/ r \ s
(r Á –?ÁÕŐŐŐŐŐ‹?ť, c % C sě d| krdks&n t d|õ dĚÉād| kr:dksLn t d|õ dĚÉāt| ttfÉrb| d } | j} d| jk}
|
r~| †° } | jd } | jd | d } | d |k}
d
}d }d d
| }d}|| dkM }d}t†° }d| }t j
dd| f| jdćg| }źxöt | ÉD ź]Ć\}}||
| }|źr∆t || Éźr∆|| }t j
t |É| | d f|jdć}|ddÖddÖf |ddÖddÖf< d|ddÖdf< d|t
t |ÉÉ|ddÖdf †° d f< t †||fd°}|jd źs÷źq|ddÖddÖf |ddÖddÖf 9 < t|ddÖddÖf É}|ddÖ|dÖf }|źrĖ|ddÖd|Öf |kjddćj\} } t †|| || d| df | ddÖdf †° || fd°}nH|ddÖd|Öf jdddć\} } t †|| | †° |fd°| †d°|k }|dk źr||ddÖddÖf t j||jdćk†d° }|jd } | źs*źq||ddÖdf jddćd|Ö }|ddÖddÖf |źrjdn| } |ddÖddÖf | |ddÖdf }!}"tj†|!|"|°} | d|Ö } |źrHd| k źr‘dk źrHn npt |!| |!É|k}#|#|"d }$t † |$|ddÖddÖf °†° |$j dddć || ddÖf< |źrH| |#† d°dk } || ||< |
źrl|| † | °||< t†° | |kźrt †!d|dõd Ě° P źqW |S ) zĮNon-Maximum Suppression (NMS) on inference results to reject overlapping detections
Returns:
list of detections, on (n,6) tensor per image [xyxy, conf, cls]
r r z Invalid Confidence threshold z&, valid values are between 0.0 and 1.0z Invalid IoU ro rĒ r ).r{ i i0u g ŗ?göôôôôô©?TFrģ )rz Nr{ g ū?)ŕas_tuple)ŕkeepdimr )ŕ
descendingg pß@u WARNING ‚ö†ÔłŹ NMS time limit z.3fz
s exceeded)"rů r r r- rz rĄ rn rň rÄ rą r' rW r* rÄ r ŕcatrķ ŕnonzeror r• rŕ ŕviewŕtensorrI ŕargsortŕ torchvisionŕopsZnmsr ŕmmrŽ ŕtorł r )%Z
predictionZ
conf_thresŕ iou_thresrß ZagnosticZ multi_labelr¶ Zmax_detŕnmrz ro ŕbsrJ ZxcZmax_whZmax_nmsZ
time_limitZ redundantŕmerger} ŕmiŕoutputŕxirg ŕlbr™ ŕboxŕmaskrā ŕjŕconfr ŕcr
ŕscoresZiour® r. r. r/ ŕnon_max_suppressionf s| &&
$( ,&<"&
.
$"* " : r= ķbest.ptc C s» t j| t †d°dć}|†d°r*|d |d< xdD ] }d ||< q0W d|d< |d †° x|d †° D ]
}d|_qbW t †||p|| ° tj †
|pĆ| °d }t † d
| õ d |rįd |õ d Ěnd
õ d|dõdĚ° d S )Nrn )ŕ map_locationŕemarh )ŕ optimizerŕ best_fitnessr@ ŕupdatesr ŕepochFg ÄĄ.AzOptimizer stripped from ķ,z
saved as r& rŻ z.1fŕMB)
rą ŕloadrz rA ŕhalfr‘ ŕ
requires_gradŕsaver? ru ŕgetsizerł rĻ )r§ r- rg r© ŕprž r. r. r/ ŕstrip_optimizer◊ s
rM zevolve: c C sP |d }|d }t | Ét |†° É } t ddĄ | D ÉÉ} |t |†° É }t| É} |ríd|õ dĚ}
t|
É|†° rv|†° jndkrít† dd |
õ |õ g° |†° rěd
nd | | †
d °d
} t |dÉŹ&} | † | d| | †
d ° d
° W d Q R X t |dÉŹ‚} t
j|ddć}
|
jddĄ dć}
t†t|
jd d Öd dÖf É°}t|
É}| † dd|õ d
Ě d|d õ d
Ě d d †d dĄ | d d Ö D É° d
d d †d dĄ |
j|d d Öf D É° d ° tj|
j| d d Ö †° | d!d"ć W d Q R X t†||õ d#Ě | d †d$dĄ | D É° d
| d †d%dĄ |D É° d ° |źrLt† dd |õ |õ d|õ Ěg° d S )&Nz
evolve.csvzhyp_evolve.yamlc s s | ]}|† ° V qd S )N)r )rD rg r. r. r/ rF Ž s z!print_mutation.<locals>.<genexpr>zgs://z /evolve.csvr ŕgsutilŕcpr& z%20s,rE ŕ
rk z%20.5g,rM T)ŕskipinitialspacec S s | † ° S )N)r )rg r. r. r/ rh ż ri z print_mutation.<locals>.<lambda>)ŕcolumnsr{ z*# YOLOv5 Hyperparameter Evolution Results
z# Best generation: z# Last generation: r z# z, c s s | ]}|† ° d õV qdS )z>20sN)r )rD rg r. r. r/ rF s rĮ c s s | ]}|d õV qdS )z>20.5gNr. )rD rg r. r. r/ rF s z
F)rv z' generations finished, current result:
c s s | ]}|† ° d õV qdS )z>20sN)r )rD rg r. r. r/ rF s c s s | ]}|d õV qdS )z20.5gNr. )rD rg r. r. r/ rF s )r- rX ŕvaluesr* r rG rŖ rť r\ r] ŕrstriprH ŕwriteŕpdZread_csvŕrenamerŃ ŕargmaxr rĽ rN rw ŕlocŕto_dictrł rĻ )rX ŕresultsŕhypŕsave_dirŕbucketrq ŕ
evolve_csvŕ evolve_yamlŕvalsr r r- r§ r5 rā Z generationsr. r. r/ ŕprint_mutationÁ s2 " * p,Hrb c
C sū t |tjÉr|gn|}źx“t| ÉD ź]ń\}}|d k r"t|Ér"|†° }t|d d Öd dÖf É}|d d Ödd Öf †d°d †d°|d d Ödd Öf< |d d Ödd Öf d d |d d Ödd Öf< t |Ɇ
° |d d Öd dÖf< t |j dd Ö |d d Öd dÖf || j É |d d Ödf †
° }g }xö|D ]í} || t
| d Ét
| d ÉÖt
| d Ét
| d ÉÖf }
t†|
d °} | d d Öd d Öd d d
Öf †ddd°} tj| tjd ć} | d } |†| ° źq"W |t†|°†|j°É†d°} | | || k | |< q"W | S )
Nr{ rĒ r r gÕŐŐŐŐŐŰ?ť r r$ )ťŗ rd r )ŕdtypeťˇ )r rŃ ŕndarrayrW r* rŲ rý rŕ ŕ unsqueezerķ r r rň r[ r% ŕresizeŕ transposeŕascontiguousarrayŕfloat32r rą rä r. rz rX )
rg rh ŕimgZim0rā r? rl Z pred_cls1ŕimsrk Zcutoutri Z pred_cls2r. r. r/ ŕapply_classifier s* 4, ,
8 & ro c C sĆ t | É} | †° rv|sv| †° r,| †d°| jfn| df\} }x4tddÉD ]&}| õ |õ |õ |õ Ě}tj†|°sDP qDW t |É} |rą| jdddć | S )Nr& rĒ i' T)rM rn ) r rG rŤ rĶ r. rÄ r? ru rr )ru rn ŕseprr r. r rL r. r. r/ ŕincrement_path1 s $ rq c C s t †t†| tj°|°S )N)r% ZimdecoderŃ ŕfromfileŕuint8)ŕfilenameŕflagsr. r. r/ ŕimreadO s rv c C s: y t †t| Éj|°d †| ° dS tk
r4 dS X d S )Nr TF)r% Zimencoder r. ŕtofiler< )rt rm r. r. r/ ŕimwriteS s
rx c C s t | †d°†° |É d S )Nŕunicode_escape)ŕimshow_r+ r, )ru ri r. r. r/ r& [ s r& c s ą j dd Ö }|d d Ödf |d }|d d Ödf |d }|d d Ödf |d }|d d Ödf |d }t†||d †° d|d d °Čt†||d †° d|d d °Čt†||d †° d|d d °Čt†||d †° d|d d °Čá ááááfddĄt|j d ÉD É}|S ) NrĒ r r$ r r{ r c sD g | ]<}ą d d Öd d Öą| ą| d Öą| ą| d Öf ĎqS )Nr r. )rD rā )ŕ feature_mapŕx_maxŕx_minŕy_maxŕy_minr. r/ r¶ s s z1get_object_level_feature_maps.<locals>.<listcomp>)rň rą ŕclampr[ rÄ )r{ ŕtargetsŕfeature_map_shapeŕx_centerŕy_centerr r ŕextracted_regionsr. )r{ r| r} r~ r r/ ŕ get_object_level_feature_mapsc s """"$rÜ c s ą j dd Ö }|d d Ödf |d }|d d Ödf |d }|d d Ödf |d }|d d Ödf |d }t†||d †° d|d d °Čt†||d †° d|d d °Čt†||d †° d|d d °Čt†||d †° d|d d °Čá ááááfddĄt|j d ÉD É}|S )NrĒ r r r$ r{ c sD g | ]<}ą d d Öd d Öą| ą| d Öą| ą| d Öf ĎqS )Nr r. )rD rā )r{ r| r} r~ r r. r/ r¶ á s z2get_object_level_feature_maps2.<locals>.<listcomp>)rň rą rÄ r[ rÄ )r{ rĀ rā rÉ rĄ r r rÖ r. )r{ r| r} r~ r r/ ŕ get_object_level_feature_maps2w s """"$rá c C sŗ g }x÷t |†d°ÉD ]ń}||ddÖf }|\}}}}|| †d° | †d° } || †d° | †d° }
|| †d° | †d° } || †d° | †d° } tt| |
| | gÉ\} }
} } | ddÖddÖ|
| Ö| | Öf }
|†|
° qW |S )a∆
Extracts regions of interest (ROIs) from the concatenated_features based on resize_boxes.
Args:
concatenated_features (torch.Tensor): Feature map with shape [batch, channels, height, width].
resize_boxes (torch.Tensor): Boxes with shape [num_boxes, 5], where each row is [batch, x1, y1, x2, y2].
Returns:
torch.Tensor: Tensor containing the ROI features with shape [num_boxes, channels, roi_height, roi_width].
r r Nr$ rĒ )rÄ ŕsizeŕmapr[ r )Zconcatenated_featuresZ resize_boxesZroi_features_listZbox_idxZ
box_coordsZbox_x1Zbox_y1Zbox_x2Zbox_y2Zroi_x1Zroi_y1Zroi_x2Zroi_y2Z roi_featuresr. r. r/ ŕextract_roi_featuresč s rä c C s> t jdt|Éd ddć\}}xPt|ÉD ]D\}}|| j| d|f †° †° †° dddć || †d|õ Ě° q&W |d j| d|d f †° †° †° dddć |d †d
° x^t t|j
ÉÉD ]L} |†° †° †° \}
} } }
t j |
| f| |
|
| dd d d
ć}|d †
|° qłW |†|° |źr2t †|° td|õ ĚÉ nt †° d S )Nr )ť r{ )ŕfigsizer Zviridisŕauto)ZcmapZaspectzChannel r zBounding Boxesra ŕnone)r Z edgecolorZ facecolorzImage saved at: )ŕpltŕsubplotsr* rW r& rn ŕdetachŕnumpyŕ set_titlerÄ rň ŕpatchesŕ RectangleZ add_patchZsuptitleZsavefigŕprintŕshow)r{ r
ŕchannelsŕtitleZ save_pathŕfigŕaxsrā Zchannelr7 ŕxminŕyminŕxmaxŕymaxŕrectr. r. r/ ŕ)plot_multi_channel_feature_map_with_boxesĶ s (,"
r° c C sN | \}}}}||d }||d }||d }||d }t †||||g°S )NrĒ )rą r) ) ZxywhrÉ rĄ r r r} r r| r~ r. r. r/ ŕ xywh_to_xyxyŌ s rĘ c C sx | † ° \}}}}||kr |d7 }||kr0|d7 }||krN||†d°krN|d8 }||krl||†d°krl|d8 }||||fS )Nr rĒ )r[ rą )Znormalized_xyxyŕint_featr} r r| r~ r. r. r/ ŕget_fixed_xyxyō s r§ )r& )r1 )F)rj rk )NTF)r F)r. )r÷ )rű rŲ )r÷ )r )r r r FFF)r r )F)r+ r, r& )r1 )r& )T)rs )Nry )r÷ TTFr r$ )rĎ rí rď )rü )rŻ rŻ r r )rŻ rŻ FrĎ )rŻ rŻ r r )rŻ rŻ )r )N)NF)r r! NFFr. r" r )r> r& )Fr& F)N)ļŕ__doc__r: rŔ r≠ r] Zlogging.configrč r? rp rŅ r3 rĖ r\ r6 rÄ r8 rų r r ŕ itertoolsr Zmultiprocessing.poolr ŕpathlibr r ŕtarfiler ŕtypingr ŕzipfiler
r r% rí rŃ ŕpandasrV ŕ
pkg_resourcesr rą r+ rN ŕtorch.nnŕnnŕtorchvision.opsr rķ ŕhasattrrů ŕ ImportErrorrq ŕultralytics.utils.checksr r6 r r ŕutils.downloadsr r ŕ
utils.metricsr r ŕ__file__rě ŕFILErM rī r[ r\ r r rŕ ŕ cpu_countZ NUM_THREADSrT rQ r) r/ Z AUTOINSTALLŕVERBOSEŕTQDM_BAR_FORMATZFONTŕset_printoptionsrT ŕoptionsŕdisplayZ max_columnsZ
setNumThreadsr@ r0 r5 r8 r> rB r2 rL rS Z LOGGING_NAMErd ŕ getLoggerrł rĻ r rf ŕsetattrrä rw rA ŕContextDecoratorrx rć rú rß rV rĺ r… rŌ r’ r‹ r‚ rÁ rŪ rŮ rŰ r r r rń r# r* r0 r3 r2 rB rb rr rU rx r[ r| rK r rź rö rļ r© rp r≠ rű rý rķ rˇ r r r
r r r r r r r= rM rb ro rq r& rz Z IMREAD_COLORrv rx ŕstackrt ŕas_posixrÜ rá rä ŕmatplotlib.pyplotŕpyplotrŹ Zmatplotlib.patchesrĒ r° rĘ r§ r. r. r. r/ ŕ<module> s8
B
.
h '# *%