Datasets
Models
Downloads: 1
269 lines (253 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. ķF/home/iml/Desktop/Talha/yolov5_P2_ROI_HCM_Micro_1000_Github/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 )NŕPWDz/kaggle/workingŕKAGGLE_URL_BASEzhttps://www.kaggle.com)ŕosŕenvironŕgetr. r. r. r/ ŕ is_kagglel s rD )ŕ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 )ŕdockerNr. )ŕ.0ŕliner. r. r/ ķ <genexpr>w s z is_docker.<locals>.<genexpr>NF)r ŕexistsŕopenŕanyŕOSError)ŕfiler. r. r/ ŕ is_dockerq s
rO 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)rA ŕaccessŕW_OKr rK ŕunlinkrM )ŕdirŕtestrN r. r. r/ ŕ is_writeable| s
rV ŕyolov5c 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)rZ ŕhandlersŕ propagate)ŕversionŕdisable_existing_loggersŕ
formattersr[ ŕloggers)ŕintrA ŕgetenvŕloggingŕINFOŕERRORŕconfigŕ
dictConfig)ŕnameŕverboseŕrankrZ r. r. r/ ŕ set_loggingć s rk ŕWindowsc C s t t| ÉÉS )N)ŕfnr )ŕxr. r. r/ ŕ<lambda>ß ů ro ŕ UltralyticsŕYOLOV5_CONFIG_DIRc C sd t †|°}|rt|É}n<ddddú}t†° |†t†° d° }t|ÉrH|ntdÉ| }|jddć |S ) NzAppData/Roamingz.configzLibrary/Application Support)rl ŕLinuxŕDarwinr& z/tmpT)ŕexist_ok) rA rb r ŕhomerC ŕplatformŕsystemrV ŕmkdir)rT ŕenv_varŕenvŕpathŕcfgr. r. r/ ŕuser_config_dir™ s
r~ 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)ŕtrĀ r) rÉ )ŕselfrĄ rĀ 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ŕ synchronizerĀ rá )rÖ r. r. r/ rá … s z Profile.time)rÄ N) ŕ__name__ŕ
__module__ŕ __qualname__rŹ rĀ rÜ rČ ré rá r. r. r. r/ r ļ s r 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)ra ŕ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 )Nrl )rw rx ŕsignalŕSIGALRMrú ŕalarmró )rÖ r. r. r/ rČ Ŕ s zTimeout.__enter__c C s, t †° dkr(t†d° | jr(|tkr(dS d S )Nrl r T)rw rx 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)rT r ŕcwdŕresolve)rÖ ŕnew_dirr. r. r/ rÜ Á s zWorkingDirectory.__init__c C s t †| j° d S )N)rA ŕchdirrT )rÖ r. r. r/ rČ Ž s zWorkingDirectory.__enter__c C s t †| j° d S )N)rA 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)rG ŕf)ŕinstancer. r/ ķ
<listcomp>Ű s zmethods.<locals>.<listcomp>)rT )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. )rG ŕkŕv)rį r. r/ ķ
<dictcomp>ż s z print_args.<locals>.<dictcomp>r& z: z, c s s | ]\}}|õ d |õ ĚV qdS )ŕ=Nr. )rG rĪ r≤ r. r. r/ rI 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į ŕ show_fileŕ show_funcrn rN ŕ_ŕfuncŕfrmr- 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:8ŕCUBLAS_WORKSPACE_CONFIGŕPYTHONHASHSEED)ŕrandomŕseedŕnprŹ ŕ manual_seedrÉ ŕmanual_seed_allŕ
check_versionr
ŕ use_deterministic_algorithmsŕbackendsŕcudnnŕ
deterministicrA rB 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. )rG rn )rĪ r. r/ rI s z-intersect_dicts.<locals>.<dictcomp>.<genexpr>)ŕallŕshape)rG 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)rG 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ŕmaxrA r| ŕgetctime)ŕ
search_dirŕ 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)r| 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)r| 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)rG r¨ r. r. r/ rI 9 s z file_size.<locals>.<genexpr>z**/*g )r rų rÓ rÝ ŕis_dirŕsumrś )r| ŕ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_connectionrM 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< )r| 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. )rG 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 rJ r r r3 ŕsplitr r, rL ŕindexŕstripra rņ rŃ ) r ŕbranchŕurlŕmsgr- ŕsplitsŕmatchesŕremoteŕ local_branchŕnŕpullr. )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 )Nŕ gitpythonr z.gitr& )r r ŕcommit)r ŕgitŕRepoŕremotesr
r ŕreplaceŕheadr ŕhexshaŕ
active_branchrh ŕ TypeErrorŕexcŕInvalidGitRepositoryError)r| 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)rh ŕhard)r— rw ŕ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)ŕpkgŕ
parse_version)rG rn r. r. r/ rI č s z check_version.<locals>.<genexpr>u WARNING ‚ö†ÔłŹ z is required by YOLOv5, but z is currently installed)r r rņ ŕwarning)ŕcurrentr+ rh ŕpinnedr) ri ŕ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_divisiblera )rG rn )ŕ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 )ŕ
isinstancera rÁ r6 ŕlistrņ r1 )ŕimgszr- r7 ŕnew_sizer. )r7 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 )NrU )r r ť r TuM WARNING ‚ö†ÔłŹ Environment does not support cv2.imshow() or PIL Image.show()
F) r> r rO ŕcv2ŕimshowrő ŕzerosŕwaitKeyŕdestroyAllWindowsr< rņ r1 )ŕwarnŕer. r. r/ ŕ check_imshow° s
rE ķ
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 ) r8 r) r9 ŕtupler ŕsuffixŕlowerr* r )rN rI r r¨ r- r. r. r/ ŕ check_suffixĪ s
rK ©z.yamlz.ymlc C s
t | |ÉS )N)ŕ
check_file)rN rI r. r. r/ ŕ
check_yamlľ s rN 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://ŕclearmlz[ClearML is not installed, so cannot use ClearML dataset. Try running 'pip install clearml'.)ŕdataŕmodelsŕutilsz**T)rš zFile not found: r zMultiple files match 'z', specify exact path: )rK r) rA r| ŕisfilerę r ŕurllibŕparseŕunquoter rh rņ rŃ rŹ ŕhubŕdownload_url_to_filerJ rÓ rÝ r r6 r7 ŕextendrś rľ r* )rN rI r ŕfilesŕdr. r. r/ rM Ń s,
,
("rM 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_DIRrh rJ rņ rŃ rŹ rX rY r) )ŕfontr] rN r r. r. r/ ŕ
check_font› s
r` 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 )rT ŕunzipŕdeleteŕcurlŕthreadsz*.yaml)ŕtrainŕvalŕnamesz data.yaml 'u :' field missing ‚ĚĆrg c s s | ]}t |tÉV qd S )N)r8 ra )rG rĪ r. r. r/ rI ķ s z check_dataset.<locals>.<genexpr>z2data.yaml names keys must be integers, i.e. 2: carŕncr| )re rf rU z../r= c s g | ]}t ą | †° ÉĎqS r. )r) r• )rG rn )r| r. r/ rģ
s z!check_dataset.<locals>.<listcomp>c 3 s | ]}ą † |°V qd S )N)rC )rG rn )rQ r. r/ rI
s )re rf rU ŕdownloadc S s g | ]}t |Ɇ° ĎqS r. )r r• )rG rn r. r. r/ rģ s c s s | ]}|† ° V qd S )N)rJ )rG rn r. r. r/ rI s u+
Dataset not found ‚ö†ÔłŹ, missing paths %sc S s g | ]}|† ° st|ÉĎqS r. )rJ r) )rG rn r. r. r/ rģ s u Dataset not found ‚ĚĆŕhttpz.zipz Downloading z to z...)ŕparentsru )r| zbash zRunning z ...)r ŕyamlķ(zs))r Nu success ‚úÖ z , saved to ŕboldzfailure u ‚ĚĆzDataset download z Arial.ttfzArial.Unicode.ttf)r] ).r8 r) r r r ri ŕ DATASETS_DIRrŅ ŕnextŕrglobŕparentŕ yaml_loadr r r9 rH ŕdictŕ enumerater◊ ŕkeysr* rC ŕ is_absoluterľ r• rJ rę rņ rŃ r< rá ŕendswithrh rŹ rX rY ry ŕ
unzip_filerS ŕ
subprocessŕrunŕexecŕroundr¬ r` r0 )
rQ ŕ autodownloadŕ extract_dirrĪ rn re rf rU r- rĄ r¨ ŕrrä r. )rQ r| r/ ŕ
check_datasetÁ sd $
"
, rĀ 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)ri r TgöôôôôôĻ?)ŕatol)ŕxywhnŕamprō rŹ ŕallclose)ŕmodelŕimŕmŕaŕb)rā r. r/ ŕ amp_allclose- s
z check_amp.<locals>.amp_allclosezAMP: )ŕcpuŕmpsFrQ ŕimageszbus.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 )ŕ
models.commonrā rÉ r¬ rp rŠ rĀ rč rľ rJ r rő ŕonesr r rņ rŃ r< r1 )rą rÉ rć ŕprefixrĀ r¨ rČ ŕhelp_urlr. )rā r/ ŕ check_amp) s
"rē ķ data.yamlc C s" t | ddćŹ }t†|°S Q R X d S )Nr( )ŕerrors)rK rl ŕ safe_load)rN r¨ r. r. r/ rs E s rs c C s8 t | dÉŹ$}tjddĄ |†° D É|ddć W d Q R X d S )NrP c S s( i | ] \}}t |tÉr t|Én||ďqS r. )r8 r r) )rG rĪ r≤ r. r. r/ r≥ N s z yaml_save.<locals>.<dictcomp>F)ŕ sort_keys)rK rl ŕ safe_dumprļ )rN rQ r¨ r. r. r/ ŕ yaml_saveK s rõ ©z .DS_Storeŕ__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. )rG rn )r¨ r. r/ rI W s z unzip_file.<locals>.<genexpr>)r| )r rr r
ŕnamelistr◊ ŕextract)rN r| rŕ ŕzipObjr. )r¨ r/ ry Q s
ry c C s0 t t| ÉɆdd°} ttj†| °Éj†d°d S )Nz:/z://rO r )r) r r rU rV rW rh 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 )ŕsilent)r] u" ‚ö†ÔłŹ Download failure, retrying r r
u ‚ĚĆ Failed to download z.gzz
Unzipping ŕtarŕxfz --directory)ŕcheckŕxfz)rA r| rT r rh rņ rŃ ŕranger rŹ rX rY rų r1 rI r r ry rz r{ rr rS )r rT ŕsuccessr¨ ŕi)rc rb ŕretryrd ra r. r/ ŕ download_onec s2
$(
z download.<locals>.download_oneT)rk ru r c s ą | é S )Nr. )rn )rę r. r/ ro á rp zdownload.<locals>.<lambda>)
r ry r ŕimapŕzipr ŕcloser√ r8 r) ) r rT ra rb rc rd r™ ŕpoolŕur. )rc rb rę r™ rd ra r/ ri a s
ri c C s* t |tjÉrt|†° É}t†| | °| S )N)r8 rŹ ŕTensorra rÁ ŕmathŕceil)rn ŕdivisorr. r. r/ r6 Ź s r6 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)rn )ŕstepsŕy1ŕy2r. r/ ro Ě rp 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 ŕbluern r z[30mz[31mz[32mz[33mz[34mz[35mz[36mz[37mz[90mz[91mz[92mz[93mz[94mz[95mz[96mz[97mz[0mz[1mz[4m)ŕblackŕredŕgreenŕyellowr¬ ŕmagentaŕcyanŕwhiteŕ bright_blackŕ
bright_redŕ bright_greenŕ
bright_yellowŕ bright_blueŕbright_magentaŕ bright_cyanŕ bright_whiteŕendrn ŕ underliner& c 3 s | ]}ą | V qd S )Nr. )rG rn )ŕcolorsr. r/ rI ∑ 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ŕastypera ŕbincountrķ ŕ
from_numpyŕfloat)ŕlabelsrh ŕ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Ŕ ra )rG rn )rh r. r/ rģ – s z+labels_to_image_weights.<locals>.<listcomp>r )rő ŕarrayŕreshaperķ )r› rh ŕ
class_weightsŕ class_countsr. )rh 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" ť ť
ť ť ť ť ť ť ť ť ť ť ť ť ť ť ť r5 ť! ť" ť# ť$ ť% ť& ť' ť( ť) ť* ť+ ť, ť. ť/ ť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= )r8 rŹ rĪ ŕclonerő ŕcopy)rn ŕyr. r. r/ ŕ xyxy2xywhŗ s r1 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= )r8 rŹ rĪ r. rő r/ )rn r0 r. r. r/ ŕ xywh2xyxyÍ s r2 ťÄ 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= )r8 rŹ rĪ r. rő r/ )rn rP ŕhŕpadwŕpadhr0 r. r. r/ ŕ
xywhn2xyxyŰ s r7 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_boxesr8 rŹ rĪ r. rő r/ )rn rP r4 ŕclipŕepsr0 r. r. r/ ŕ
xyxy2xywhnĢ s r; c C sJ t | tjÉr| †° nt†| °}|| d | |d< || d | |d< |S )N).r ).r )r8 rŹ rĪ r. rő r/ )rn rP r4 r5 r6 r0 r. r. r/ ŕxyn2xy
s r< c C sp | j \}}|dk|dk@ ||k@ ||k@ }|| || }}t|Érft†|†° |†° |†° |†° g°S t†d°S )Nr )r rā )ŕTrL rő rŠ ŕminrÁ r@ )ŕsegmentŕwidthŕheightrn r0 ŕinsider. r. r/ ŕ segment2box s
rC c C sL g }x8| D ]0}|j \}}|†|†° |†° |†° |†° g° q
W tt†|°ÉS )N)r= ŕappendr> rÁ r1 rő rŠ )ŕsegmentsŕboxesr- rn r0 r. r. r/ ŕsegments2boxes s
&rG ťŤ 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)rG r© )r- rn ŕxpr. r/ rģ ) s z%resample_segments.<locals>.<listcomp>rĽ r ) ru rő rō ŕlinspacer* ŕarangerß r‚ r= )rE r r© r. )r- rn rK r/ ŕresample_segments# s "2rN 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> r8 )ŕ
img1_shaperF ŕ
img0_shapeŕ ratio_padŕgainŕpadr. r. r/ ŕ scale_boxes- s "2
rT 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)rO rE rP rQ ŕ normalizerR rS r. r. r/ ŕscale_segments= s "2
rW 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= )r8 rŹ rĪ ŕclamp_r9 )rF rō r. r. r/ r8 P s (r8 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 )r8 rŹ rĪ rX r9 )rE rō r. r. r/ rU \ s
(rU Á –?ÁÕŐŐŐŐŐ‹?ť, 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 rŹ rĽ r ).rā i i0u g ŗ?göôôôôô©?TFrś )rĀ Nrā g ū?)ŕas_tuple)ŕkeepdimr )ŕ
descendingg pß@u WARNING ‚ö†ÔłŹ NMS time limit z.3fz
s exceeded)"r r8 r9 rH rĀ rč ré rō rá rŹ r@ ru r* rß r ŕcatr2 ŕnonzeror= r‹ rÁ ŕviewŕtensorrL ŕargsortŕ torchvisionŕopsŕnmsr ŕmmrķ ŕtorņ r1 )%ŕ
predictionŕ
conf_thresŕ iou_thresrř ŕagnosticŕ multi_labelr› ŕmax_detŕnmrĀ rŹ ŕbsrh ŕxcŕmax_whŕmax_nmsŕ
time_limitŕ redundantŕmergerĄ ŕmiŕoutputŕxirn ŕlbr≤ ŕboxŕmaskr© ŕjŕconfr ŕcrF ŕscoresŕiourŖ 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 )Nré )ŕ map_locationŕemarą )ŕ optimizerŕ best_fitnessrÖ ŕupdatesr ŕepochFg ÄĄ.AzOptimizer stripped from ķ,z
saved as r& r
z.1fŕMB)
rŹ ŕloadrĀ rC ŕhalfrŠ ŕ
requires_gradŕsaverA r| ŕgetsizerņ rŃ )r¨ r- rn rĪ ŕprŻ r. r. r/ ŕstrip_optimizer◊ s
rí 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 )rG rn r. r. r/ rI Ž s z!print_mutation.<locals>.<genexpr>zgs://z /evolve.csvr ŕgsutilŕcpr& z%20s,rä ŕ
rč z%20.5g,rP T)ŕskipinitialspacec S s | † ° S )N)r )rn r. r. r/ ro ż rp 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 )rG rn r. r. r/ rI s rÁ c s s | ]}|d õV qdS )z>20.5gNr. )rG rn r. r. r/ rI s z
F)rô z' generations finished, current result:
c s s | ]}|† ° d õV qdS )z>20sN)r )rG rn r. r. r/ rI s c s s | ]}|d õV qdS )z20.5gNr. )rG rn r. r. r/ rI s )rH rv ŕvaluesr* r rJ rÓ rÝ rz r{ ŕrstriprK ŕwriteŕpdŕread_csvŕrenamerő ŕargmaxr r√ rl rö ŕlocŕto_dictrņ rŃ )rv ŕresultsŕhypŕsave_dirŕbucketrď ŕ
evolve_csvŕ evolve_yamlŕvalsr r r- r¨ rQ r© ŕ generationsr. r. r/ ŕprint_mutationÁ s2 " * p,Hr© 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= )ťŗ rę r )ŕdtypeťˇ )r8 rő ŕndarrayru r* r. r1 rÁ ŕ unsqueezer2 r rT rō ra r> ŕresizeŕ transposeŕascontiguousarrayŕfloat32rD rŹ rĪ rh rĀ rě )
rn rą ŕimgŕim0r© r\ rĆ ŕ pred_cls1ŕimsrč ŕcutoutrČ ŕ pred_cls2r. r. r/ ŕapply_classifier s* 4, ,
8 & rļ c C sĆ t | É} | †° rv|sv| †° r,| †d°| jfn| df\} }x4tddÉD ]&}| õ |õ |õ |õ Ě}tj†|°sDP qDW t |É} |rą| jdddć | S )Nr& rĽ i' T)rk ru ) r rJ rų rĹ rI rß rA r| ry )r| ru ŕsepry rI r rĎ r. r. r/ ŕincrement_path1 s $ rľ c C s t †t†| tj°|°S )N)r> ŕimdecoderő ŕfromfileŕuint8)ŕfilenameŕflagsr. r. r/ ŕimreadO s r¬ c C s: y t †t| Éj|°d †| ° dS tk
r4 dS X d S )Nr TF)r> ŕimencoder rI ŕtofiler< )rņ rī r. r. r/ ŕimwriteS s
rŇ c C s t | †d°†° |É d S )Nŕunicode_escape)ŕimshow_r+ r, )r| rČ 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. )rG r© )ŕ feature_mapŕx_maxŕx_minŕy_maxŕy_minr. r/ rģ s s z1get_object_level_feature_maps.<locals>.<listcomp>)rō rŹ ŕclampra rß )r» ŕtargetsŕfeature_map_shapeŕx_centerŕy_centerr@ rA ŕ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. )rG r© )r» r… r rň rŐ r. r/ rģ á s z2get_object_level_feature_maps2.<locals>.<listcomp>)rō rŹ rÕ ra rß )r» rő rŌ r– r— r@ rA 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ŕmapra rD )ŕconcatenated_featuresŕ resize_boxesŕroi_features_listŕbox_idxŕ
box_coordsŕbox_x1ŕbox_y1ŕbox_x2ŕbox_y2ŕroi_x1ŕroi_y1ŕroi_x2ŕroi_y2ŕ 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 ŕviridisŕauto)ŕcmapŕaspectzChannel r zBounding BoxesrÄ ŕnone)r ŕ edgecolorŕ facecolorzImage saved at: )ŕpltŕsubplotsr* ru r? ré ŕdetachŕnumpyŕ set_titlerß rō ŕpatchesŕ Rectangleŕ add_patchŕsuptitleŕsavefigŕprintŕshow)r» rF ŕchannelsŕtitleŕ save_pathŕfigŕaxsr© ŕchannelr{ ŕ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Ź rb ) ŕxywhr– r— r@ rA 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Ľ )ra r’ )ŕnormalized_xyxyŕint_featr rŐ r… rň r. r. r/ ŕget_fixed_xyxyō s r )r& )r1 )F)rq rr )NTF)r F)r. )r„ )r r )r„ )r( )r- r- r. FFF)r5 r )F)rF rG r& )rL )r& )T)rĖ )Nrú )r„ TTFr r= )rł rĻ rļ )r÷ )r3 r3 r r )r3 r3 Frł )r3 r3 r r )r3 r3 )rH )N)NF)rY rZ NFFr. r[ r )rÉ r& )Fr& F)N)ļŕ__doc__r: rś rĶ rc ŕlogging.configr≤ rA rw rŐ r3 rĚ rz r6 rá rU r/ r r ŕ itertoolsr ŕmultiprocessing.poolr ŕpathlibr r ŕtarfiler ŕtypingr ŕzipfiler
r r> rÚ rő ŕpandasrõ ŕ
pkg_resourcesr/ rŹ rd rl ŕtorch.nnŕnnZtorchvision.opsr r ŕhasattrr ŕ ImportErrorrx ŕultralytics.utils.checksr rS r r ŕutils.downloadsr r ŕ
utils.metricsr r ŕ__file__r• ŕFILErk rľ ra rb r r> rÁ ŕ cpu_countŕ NUM_THREADSrr ro r) rJ ŕ AUTOINSTALLŕVERBOSEŕTQDM_BAR_FORMATŕFONTŕset_printoptionsrX ŕoptionsŕdisplayŕ max_columnsŕ
setNumThreadsrB r0 r5 r8 r> rD r2 rO rV ŕ LOGGING_NAMErk ŕ getLoggerrņ rŃ r1 rm ŕsetattrrĎ r~ r^ ŕContextDecoratorr rĒ r£ rĮ rt r… r÷ r‹ r‚ rŽ rŮ rŲ rŁ r r r r' r, r— r< rE rK rN rM r` rĀ rē rs rõ ry r° ri r6 r∑ rŃ r¬ rŗ rí rŚ r- r1 r2 r7 r; r< rC rG rN rT rW r8 rU rā rí r© rļ rľ r? r« ŕ IMREAD_COLORr¬ rŇ ŕstackrņ ŕas_posixr” r‘ rŚ ŕmatplotlib.pyplotŕpyplotrÔ ŕmatplotlib.patchesrŰ r r r r. r. r. r/ ŕ<module> s8
B
.
h '# *%