ҶfddlZddlZddlZddlZddlZddlZddlZddlZddlZddl m Z ddl Z ddl Z ddlZddlZddlZddlZddlmZddlmZddlmZejddd e jdd d ji d ejj_e jdee ej!dej"d<dOdZ#dPdZ$dQdZ%dZ&dRdZ'dZ(dZ)dSdZ*dTdZ+dZ,d Z-d!Z.d"Z/d#Z0dUd'Z1d(Z2dVd*Z3d)e j4d)fd+Z5d,Z6d-Z7d.Z8dWd0Z9dWd1Z:dXd2Z;d3Zd7Z?d[d;Z@d\d>ZAd?ZBd@ZCdAZDd]dBeEfdCZFd]dBeEfdDZG d^dGZH d_dHZId`dJZJdadLZKdMZLdbdNZMdS)cN)Path)gsutil_getsize)fitness)init_torch_seedsi@long) linewidth precisionprofile float_kindz{:11.5g})r formatter NUMEXPR_MAX_THREADScftjd|dvr tjn tjdS)Nz %(message)s)rr)formatlevel)logging basicConfigINFOWARN)ranks %/var/www/html/yolov7/utils/general.py set_loggingr!sB "googll7<AAAAAActj|tj|t|dSN)randomseednpr)r s r init_seedsr"'s: KINN4Tr.c|tj|dd}|r t|tjjndS)Nz /**/last*.ptT recursive)key)globmaxospathgetctime) search_dir last_lists rget_latest_runr0.sA Z555FFFI3< D3ybg. / / / /"DrcDtdS)Nz /workspace)rexistsrrisdockerr44s    $ $ & &&rr(ctjdkr(|ddn|S)NWindowsasciiignore)platformsystemencodedecode)strs remojisr>9s85=_5F5F)5S5S3::<<  w 1 1 1Y\\rc^ddl} |dddS#t$rYdSwxYw)Nr)z1.1.1.1irTF)socketcreate_connectionOSError)r@s r check_onlinerC>sNMMM  !11555t uus  ,,c ttdd tds Jdt r Jdt s Jdd}t j|d  d}t jd d  }tt jd |d d }|dkrd|dd|dkzd|d}nd|d}tt|dS#t$r}t|Yd}~dSd}~wwxYw)Nzgithub: r()endz.gitz%skipping check (not a git repository)zskipping check (Docker image)zskipping check (offline)z/git fetch && git config --get remote.origin.urlTshellzgit rev-parse --abbrev-ref HEADz git rev-list z..origin/master --countru'⚠️ WARNING: code is out of date by z commitsz). Use 'git pull' to update or 'git clone z' to download latest.zup to date with u ✅)printcolorstrrr2r4rC subprocess check_outputr<striprstripintr> Exception)cmdurlbranchnrHes rcheck_git_statusrWHs (:  B''''F||""$$MM&MMMM::>>>>>>~~999999?%c666==??EEGGNNvVV()JRVWWW^^``ffhh  '(W(W(W(W_cddd e e q55U!UUC1q5MUU:=UUUAA-3,,,A fQii  asEE)) F 3FF requirements.txtr3cddl}tddd}t|ttfrt |}|s)t |d|ddSfd|| D}nfd|D}d}|D]} | |#t$rg}|d z }t |d|j d t tjd |j d d Yd}~d}~wwxYw|rmdt!vr|n|}|d|dd|d kzd|d|dtddd } t t#| dSdS)Nrredboldz requirements: z not found, check failed.cBg|]}|jv |j|jSr3)name specifier.0xexcludes r z&check_requirements..gs7wwwQabagovavav1601;00avavavrcg|]}|v| Sr3r3r`s rrdz&check_requirements..is#DDDa1G3C3C3C3C3CrrIz> not found and is required by YOLOR, attempting auto-update...z pip install ''TrFfilez packagerHz updated per  u ⚠️ z;Restart runtime or rerun command for updates to take effect) pkg_resourcesrK isinstancer=rr2rJresolveparse_requirementsopenrequirerQreqrLrMr<localsr>) requirementsrcpkgprefixrgrUrrVsourcerHs ` rcheck_requirementsrv^sO eV_ 5 5F,d ,,EL!!{{}}  VGGdllnnGGG H H H Fwwww#:P:PQUQZQZQ\Q\:]:]www DDDD<DDD  A ZZ Z KKNNNN Z Z Z FA Vccaeccc d d d *)*B!%*B*B*B$OOOVVXX Y Y Y Y Y Y Y Y Z  #)VXX#5#5< s s s s3!a%= s sv s s s s'0mnn s s s fQii sC E(AE  E cvt|t|}||krtd|||fz|S)NzHWARNING: --img-size %g must be multiple of max stride %g, updating to %g)make_divisiblerPrJ)img_sizerHnew_sizes rcheck_img_sizer|{sEhA//H8 X\dfgiq[rrsss OrcP tr JdtjdtjdtjdtjtjddS#t$r}td|Yd}~dSd}~wwxYw)Nz/cv2.imshow() is disabled in Docker environmentstest)rIrIrITzVWARNING: Environment does not support cv2.imshow() or PIL Image.show() image displays F) r4cv2imshowr!zeroswaitKeydestroyAllWindowsrQrJ)rVs r check_imshowrs ::PPPPPP 628I../// A  At  khikkllluuuuusA:A>> B%B  B%ct|s|dkr|Stjd|zd}t|s Jd|t|dksJd|d||d S) Nr(z./**/Tr%zFile Not Found: rIzMultiple files match 'z', specify exact path: r)ris_filer)len)rgfiless r check_filers Dzztrzz  'D.D9995zz444d444445zzQ ] ] ]V[ ] ]Qxrc|d|d}}|rDt|r6dt|tr|n|gD}t d|Dst dd|Dz|rt|rt d|z|drd|d rOt|j }tj ||tjd |d |}ntj|}t d |d krdndzdStddSdSdS)NvaldownloadcPg|]#}t|$Sr3)rrkrarbs rrdz!check_dataset..s(TTTQtAww  TTTrc3>K|]}|VdSr)r2rs r z check_dataset..s*++!188::++++++rz2 WARNING: Dataset not found, nonexistent paths: %scTg|]%}|t|&Sr3)r2r=rs rrdz!check_dataset..s1JqJqJqVWfgfnfnfpfpJq3q66JqJqJqrzDownloading %s ...httpz.zipz unzip -q z -d ../ && rm zDataset autodownload %s rsuccessfailurezDataset not found.)getrrjlistallrJ startswithendswithrr^torchhubdownload_url_to_filer+r:rQ)dictrrHfrts r check_datasetrs XXe__dhhz22C 6s3xx6TT*S$2G2G+R33cUTTT++s+++++ 6 GJqJq[^JqJqJqq r r r 6SVV 6*Q.///<<''%AJJv,>,>%Q AI221a888 111aa"HIIAA ! A1!q&&YYiXYYYYY 45556666 6 6rc6tj||z |zSr)mathceil)rbdivisors rryrys 9Q[ ! !G ++rc0tjdd|S)Nu"[|@#!¡·$€%&()=?¿^*;:,¨´><+]_)patternreplstring)resub)rHs r clean_strrs 6>SQR S S SSr?dcfdS)Ncjdtj|tjzz z dz z zzS)NrI)rcospi)rbstepsy1y2s rzone_cycle..s4q48AK%$7888A="r'JROrr3)rrrs```r one_cyclers O O O O O OOrct|dkr|n dd|df^}}iddddd d d d dd ddddddddddddddddddd d!d"d#d$d%d&d'd(d)fd*|D|zd$zS)+NrIbluer[rblackzrZzgreenzyellowzzmagentazcyanzwhitez bright_blackz bright_redz bright_greenz bright_yellowz bright_bluezbright_magentaz bright_cyanz bright_whitezrEzzz)r[ underliner(c3(K|] }|V dSrr3)rarbcolorss rrzcolorstr..s'++6!9++++++r)rjoin)inputargsrrs @rrKrKsE ZZ!^^EE&&%(1KMT6&gz&Z&z& &j &  & j &z&j&J&j&z&Z& &Z&j& Y!&" $%&&&F& 77++++d+++ + +k 9F5M IIrPcR|dtjStj|d}|dddftj}tj||}d||dk<d|z }||z}tj|S)Nr minlengthrI) rTensorr! concatenateastypeint32bincountsum from_numpy)labelsncclassesweightss rlabels_to_class_weightsrs ay|~~ ^FA & &FQQQTl!!"(++Gk'R000G GGqL'kG w{{}}G  G $ $$rctjfd|D}|d|zd}|S)Ncg|]?}tj|dddftj@S)Nrr)r!rrr)rarbrs rrdz+labels_to_image_weights..sCaaaUVR[111a4)A)ARPPPaaarrI)r!arrayreshaper)rr class_weights class_counts image_weightss ` rlabels_to_image_weightsrsV8aaaaZ`aaabbL"**1b11L@EEaHHM rcgd}|S)N)PrIrrrr r rw!"#$%&'()*+,./0123456789:;<=>?@ACFHIJKLMNOrQRTUVWXYZr3)rbs rcoco80_to_coco91_classr1s  a a aA Hrct|tjr|nt j|}|dddf|dddfzdz |dddf<|dddf|dddfzdz |dddf<|dddf|dddfz |dddf<|dddf|dddfz |dddf<|SNrrrIrrjrrcloner!copyrbys r xyxy2xywhr9s5<00@ bgajjAAw111a4 A%AaaadGAw111a4 A%AaaadG1g!!!Q$AaaadG1g!!!Q$AaaadG Hrct|tjr|nt j|}|dddf|dddfdz z |dddf<|dddf|dddfdz z |dddf<|dddf|dddfdz z|dddf<|dddf|dddfdz z|dddf<|Sr3r4r7s r xywh2xyxyr; s5<00@ bgajjA1g!!!Q$! #AaaadG1g!!!Q$! #AaaadG1g!!!Q$! #AaaadG1g!!!Q$! #AaaadG Hrct|tjr|nt j|}||dddf|dddfdz z z|z|dddf<||dddf|dddfdz z z|z|dddf<||dddf|dddfdz zz|z|dddf<||dddf|dddfdz zz|z|dddf<|Sr3r4rbwhpadwpadhr8s r xywhn2xyxyrCs!5<00@ bgajjA1QQQT7Qqqq!tWq[()D0AaaadG1QQQT7Qqqq!tWq[()D0AaaadG1QQQT7Qqqq!tWq[()D0AaaadG1QQQT7Qqqq!tWq[()D0AaaadG Hrct|tjr|nt j|}||dddfz|z|dddf<||dddfz|z|dddf<|S)NrrIr4r>s rxyn2xyrEs}5<00@ bgajjA!AAAqD'kD AaaadG!AAAqD'kD AaaadG Hrcr|j\}}|dk|dkz||kz||kz}||||}}t|r`tj||||gntjdS)Nr)rIr)Tanyr!rminr*r)segmentwidthheightrbr8insides r segment2boxrN%s 9DAq1fa AJ /1; ?F fIqyqA=@VV Y28QUUWWaeeggquuww8 9 9 9RXIYIYYrc(g}|D]m}|j\}}|||||gnt t j|Sr)rGappendrIr*r9r!r)segmentsboxesrHrbr8s rsegments2boxesrS-su E ;;s1 aeeggquuww9:::: RXe__ % %%rct|D]\}tjddddffdtjdt dz |tjt tjfdt dDddj||<|S)NrrIaxisc Pg|]"}tjdd|f#Sr)r!interp)rairHrbxps rrdz%resample_segments..<s3%R%R%RAbi2qAw&?&?%R%R%Rrrr) enumerater!rlinspacerarangerangerrG)rQrUrZrHrbr[s @@@rresample_segmentsr`6s(##ee1 NAq1aaay> 2 2 2 K3q66A:q ) ) Ys1vv  n%R%R%R%R%R%Rq%R%R%RSS[[\]_abbd Orc|]t|d|dz |d|dz }|d|d|zz dz |d|d|zz dz f}n|dd}|d}|ddddgfxx|dzcc<|ddddgfxx|dzcc<|ddddfxx|zcc<t|||SNrrIrrr)rI clip_coords) img1_shapecoords img0_shape ratio_padgainpads r scale_coordsrj@s:a=:a=0*Q-*Q-2OPP!}z!}t33q8:a=:VW=[_K_;_cd:dd|Al 111q!f9Q 111q!f9Q 111bqb5MMMTMMM ### Mrc6|dddfd|d|dddfd|d|dddfd|d|dddfd|ddS)NrrIrr)clamp_)rR img_shapes rrcrcPs !!!Q$Kq)A,''' !!!Q$Kq)A,''' !!!Q$Kq)A,''' !!!Q$Kq)A,'''''rTFHz>c |j}|rG|d|d|d|df\}}} } |d|d|d|df\} } } }n|d|ddz z |d|ddz z} }|d|ddz z |d|ddz z} }|d|ddz z |d|ddz z} } |d|ddz z |d|ddz z}} tj| | tj|| z dtj| |tj|| z dz}| |z | |z |z}}| | z || z |z}}||z||zz|z |z}||z }|s|s|rKtj| | tj|| z }tj| |tj|| z }|s|r|dz|dzz|z}| | z|z | z dz| |z|z | z dzzdz }|r|||z z S|rdt jdzz tjtj|||zz tj|||zz z dz}tj 5|||z d|zzz }dddn #1swxYwY|||z ||zzz SdS||z|z}|||z |z z S|Srb) rGrrIr*clamprrpowatanno_grad)box1box2x1y1x2y2GIoUDIoUCIoUepsb1_x1b1_y1b1_x2b1_y2b2_x1b2_y1b2_x2b2_y2interw1h1w2h2unionioucwchc2rho2valphac_areas rbbox_iourXs 6DD%)!Wd1gtAwQ%G"ueU%)!Wd1gtAwQ%G"ueUUAwa1,d1gQ! .CuAwa1,d1gQ! .CuAwa1,d1gQ! .CuAwa1,d1gQ! .CuYue $ $uy'>'> > E Ea H H Yue $ $uy'>'> > E Ea H H IEU]EEMC/B U]EEMC/B Gb2g  % +E %-C tt Yue $ $uy'>'> > Yue $ $uy'>'> >  34 3q27"S(BU]U*U2q8U]U*U2q89<=>D 5TBY& 5A%5:bBHo3N3NQVQ[\^bdgjbj\kQlQl3lno)p)pp]__66SAG!45E666666666666666dRi!e)344  5 5 "Ws]F&5.F22 2 s'KK Kr& .>c|j}|rG|d|d|d|df\}} } } |d|d|d|df\} } }}n|d|ddz z |d|ddz z} }|d|ddz z |d|ddz z} } |d|ddz z |d|ddz z}} |d|ddz z |d|ddz z}} tj| |tj|| z dtj| |tj| | z dz}| |z | | z |z}}|| z || z |z}}||z||zz|z |z}tj||z |z|}|s|s|rtj| |tj|| z }tj| |tj| | z }|s|r|dz|dzz|z|z}tj| |z|z | z }tj| |z| z | z }|dz|dzzdz |z}|r|||z z S|rdtjdzz tjtj ||z tj ||z z dz}tj 5|d|z||z z |zz }dddn #1swxYwY|||z tj||z|z|zz SdStj||z|z|}|tj||z |z |z|z S|Srb) rGrrIr*rprqabsrrrrrs) rtrurvrwrxryrrzr{r|r}r~rrrrrrrrrrrrrrrho_xrho_yrr alpha_ciours rbbox_alpha_iours) 6DD%)!Wd1gtAwQ%G"ueU%)!Wd1gtAwQ%G"ueUUAwa1,d1gQ! .CuAwa1,d1gQ! .CuAwa1,d1gQ! .CuAwa1,d1gQ! .CuYue $ $uy'>'> > E Ea H H Yue $ $uy'>'> > E Ea H H IEU]EEMC/B U]EEMC/B Gb2g  % +E )E%K#%u - -C tt Yue $ $uy'>'> > Yue $ $uy'>'> >  K4 K'B!G#-3BIeeme3e;<.box_area%AQCFSVO44rNrrrGrrIr*rpprod)rtrurarea1area2rs rbox_iours555 HTV  E HTV  EYtAAAtQRRK($qqq!""u+ 6 644QSRSQS CTVZ[\[\[\^`_`^`[`Va9b9b b i ijk l l q qrs t tE E!!!T'NU*U2 33rc|dddf}|d}tj||d}||d|dz|z z S)Nr)rrIr)wh1wh2rs rwh_ioursd aaag,C d)C Ic3   $ $Q ' 'E CHHQKK#((1++-5 66rcd}||j}||j}tj|dddddf|ddddftj|dddddf|ddddfz dd}|dddf|z|z }||z }tj|dddddf|ddddf}tj|dddddf|ddddf} | |z d} | dddddf| dddddfz} || |z | z z S)a Return generalized intersection-over-union (Jaccard index) between two sets of boxes. Both sets of boxes are expected to be in ``(x1, y1, x2, y2)`` format with ``0 <= x1 < x2`` and ``0 <= y1 < y2``. Args: boxes1 (Tensor[N, 4]): first set of boxes boxes2 (Tensor[M, 4]): second set of boxes Returns: Tensor[N, M]: the NxM matrix containing the pairwise generalized IoU values for every element in boxes1 and boxes2 cH|d|dz |d|dz zSrr3rs rrzbox_giou..box_arearrNrrrIrIr) rtrurrrrrrltirbiwhiareais rbox_giours555 HTV  E HTV  E YtAAAtQRRK($qqq!""u+ 6 644QSRSQS CTVZ[\[\[\^`_`^`[`Va9b9b b i ijk l l q qrs t tE 111d7^e #e +E %-C )DD"1"%tAAArrE{ 3 3C )DD!""%tAAAqrrE{ 3 3C 9    " "C 111aL3qqq!!!Qw< 'E %%-5( ((rrzcd}||j}||j}tj|dddddf|ddddftj|dddddf|ddddfz dd}|dddf|z|z }||z }tj|dddddf|ddddf} tj|dddddf|ddddf} | | z d} | dddddfdz| dddddfdzz|z} |ddddf|ddddfzdz } |ddddf|ddddfzdz }|dddf|dddfzdz }|dddf|dddfzdz }| |z dz||z dzz}|ddddf|ddddfz }|ddddf|ddddfz }|dddf|dddfz }|dddf|dddfz }dtjdzz tjtj||z tj||z z dz}tj 5|d|z |z|zz }dddn #1swxYwY||| z z ||zz S) a( Return complete intersection-over-union (Jaccard index) between two sets of boxes. Both sets of boxes are expected to be in ``(x1, y1, x2, y2)`` format with ``0 <= x1 < x2`` and ``0 <= y1 < y2``. Args: boxes1 (Tensor[N, 4]): first set of boxes boxes2 (Tensor[M, 4]): second set of boxes eps (float, optional): small number to prevent division by zero. Default: 1e-7 Returns: Tensor[N, M]: the NxM matrix containing the pairwise complete IoU values for every element in boxes1 and boxes2 cH|d|dz |d|dz zSrr3rs rrzbox_ciou..box_area rrNrrrrIrr) rGrrIr*rprrrqrrrs)rtrurzrrrrrrrrrdiagonal_distance_squaredx_py_px_gy_gcenters_distance_squaredw_predh_predw_gth_gtrrs rbox_ciours555 HTV  E HTV  E YtAAAtQRRK($qqq!""u+ 6 644QSRSQS CTVZ[\[\[\^`_`^`[`Va9b9b b i ijk l l q qrs t tE 111d7^e #e +E %-C )DD"1"%tAAArrE{ 3 3C )DD!""%tAAAqrrE{ 3 3C 9    " "C!$QQQ1W!2s111aaa7|q7H IC O 4 d111dA:. .! 3C 4 d111dA:. .! 3C 1:QQQT "a 'C 1:QQQT "a 'C #c a/392BB !!!T1* QQQaZ 0 0F !!!T1* QQQaZ 0 0F 1:QQQT "D 1:QQQT "D eh!m  5:dTk+B+BUZPVY_P_E`E`+`cd e eeA ((QWq[3&'((((((((((((((( *-FF G%RS) SSs0K  KKcd}||j}||j}tj|dddddf|ddddftj|dddddf|ddddfz dd}|dddf|z|z }||z }tj|dddddf|ddddf} tj|dddddf|ddddf} | | z d} | dddddfdz| dddddfdzz|z} |ddddf|ddddfzdz } |ddddf|ddddfzdz }|dddf|dddfzdz }|dddf|dddfzdz }| |z dz||z dzz}||| z z S)a( Return distance intersection-over-union (Jaccard index) between two sets of boxes. Both sets of boxes are expected to be in ``(x1, y1, x2, y2)`` format with ``0 <= x1 < x2`` and ``0 <= y1 < y2``. Args: boxes1 (Tensor[N, 4]): first set of boxes boxes2 (Tensor[M, 4]): second set of boxes eps (float, optional): small number to prevent division by zero. Default: 1e-7 Returns: Tensor[N, M]: the NxM matrix containing the pairwise distance IoU values for every element in boxes1 and boxes2 cH|d|dz |d|dz zSrr3rs rrzbox_diou..box_areaArrNrrrrIrr)rtrurzrrrrrrrrrrrrrrrs rbox_diour3s555 HTV  E HTV  E YtAAAtQRRK($qqq!""u+ 6 644QSRSQS CTVZ[\[\[\^`_`^`[`Va9b9b b i ijk l l q qrs t tE 111d7^e #e +E %-C )DD"1"%tAAArrE{ 3 3C )DD!""%tAAAqrrE{ 3 3C 9    " "C!$QQQ1W!2s111aaa7|q7H IC O 4 d111dA:. .! 3C 4 d111dA:. .! 3C 1:QQQT "a 'C 1:QQQT "a 'C #c a/392BB *-FF GGr??c |jddz }|d|k}d\} } d} d} d} d}||d kz}d }tj}tjd |j g|jd z}t |D]\}}|||}|rt ||r||}tjt ||dzf|j }|ddd df|ddddf<d|dddf<d|tt ||ddd fdzf<tj ||fd }|jd s|d kr|ddddf|ddddf<n"|ddddfxx|ddddfzcc<t|ddddf}|ru|ddddf|k d j \}}tj |||||dzdf|dddf fd }nm|ddddfd d\}}tj ||| fd |d|k}|D||ddddftj||j kd }|jd }|s{|| kr.||dddfdd| }|ddddf|rd n| z}|ddddf|z|dddf}}t&j|||}|jd | kr |d| }|rd |cxkrdkrnnt-||||k}||dz}tj||ddddf |d dz ||ddf<|r||d d k}||||<tj|z | krt3d| dn|S)Runs Non-Maximum Suppression (NMS) on inference results Returns: list of detections, on (n,6) tensor per image [xyxy, conf, cls] rr.rri,0u$@TrIFrrdevicerNrras_tuplekeepdimrr descendingp@WARNING: NMS time limit s exceededshapetimerrrr\rr_rcatr;nonzerorGfloatr*viewtensorrHargsort torchvisionopsnmsrmmrrJ) prediction conf_thres iou_thresragnostic multi_labelrrxcmin_whmax_whmax_detmax_nms time_limit redundantmergetoutputxirblrrrZjconfrUcrRscoresrrs rnon_max_suppressionr`s  ! q B F j (BNFFGGJI26K E Ak&):;;; >>AAaCyAaaa!eHAaaadG36AeCFFmmQqqq!tW\\^^a// 0 1a&!$$Awqz   77AaCyAaaaeHH aaaeHHH!!!QqS& !HHH!!!RaR%!!  QaaaeHz)22E2BBDDAq 3q61QAt^#4a4j6F6F6H6HI1MMAA122hll1dl33GD! 3aggii0!44TYYr]]Z5OPA  1QQQ!V9 WQX F F FFKKANNOA GAJ >  [[!AAAqD'//T/228G8<=A aaa1fIh2F 3!!!RaR%1 a1gv O  vy 9 9 71:  (7( A  &a!kkkkckkkkk%(E**Y6CF4L(Gx111bqb52288::W[[TX[=Y=YYAa!eH &cggajj1n%qTr IKK!Oz ) ) CZCCC D D D E * Mrc  |"|s|jddz n|jddz }|d|k} d\} } d} d}d }d }||d kz}d }tj}tjd |jg|jdz}t |D]e\}}|| |}|rt ||r||}tjt ||dzf|j}|ddd df|ddddf<d|dddf<d|tt ||dddfdzf<tj ||fd}|jds|dddd|zfxx|ddddfzcc<t|ddddf}|ru|ddddf|k d j \}}tj |||||dzdf|dddf fd }n|sn|ddddfd d \}}tj ||| fd |d|k}n||ddddf}|ddddfd d \}}tj ||| |fd |d|k}|D||ddddftj||jkd }|jd}|s||kr.||dddfd d|}|ddddf|rdn| z}|ddddf|z|dddf} }t&j|| |}|jd| kr |d| }|rd |cxkrdkrnnt-||||k}!|!| dz}"tj|"|ddddf |"d d z ||ddf<|r||!d d k}||||<tj|z |krt3d|dng|S)rNrrrrrrrrTrIFrrrrrrrrrrrrrr)#rrrrrrr kpt_labelrnkptrrrrrrrrrrrrbrrrrZrrkptsrUrrRrrrs# rnon_max_suppression_kptrs} z-6 TZ a 1 $ $JNq>Q QF:&&FFA bfI  %c&*oo %r A SVVR!V,QX>>>AAaCyAaaa!eHAaaadG36AeCFFmmQqqq!tW\\^^a// 0 1a&!$$Awqz   !!!QqtV) !!!QqS& ! !!!RaR%!!  [aaaeHz)22E2BBDDAq 3q61QAt^#4a4j6F6F6H6HI1MMAA [AAAqrrE(,,q$,77aIsD!''))4a8829STABBxAAAqsF)--4-88aIsD!''))T:A>>tyy}}z?YZ  1QQQ!V9 WQX F F FFKKANNOA GAJ >  [[!AAAqD'//T/228G8<=A aaa1fIh2F 3!!!RaR%1 a1gv O  vy 9 9 71:  (7( A  &a!kkkkckkkkk%(E**Y6CF4L(Gx111bqb52288::W[[TX[=Y=YYAa!eH &cggajj1n%qTr IKK!Oz ) ) CZCCC D D D E * Mrbest.ptc tj|tjd}|dr |d|d<dD]}d||<d|d<|d|dD] }d|_ tj||p|tj |p|d z }td |d |rd |znd d|dddS)Ncpu) map_locationemamodel) optimizertraining_resultswandb_idrupdatesrepochFg.AzOptimizer stripped from ,z saved as %s,r(r\z.1fMB) rloadrrhalf parameters requires_gradsaver+r,getsizerJ)rrHrbkpmbs rstrip_optimizerr s 15<#6#6777AuuU||uX' J!AgJgJOO wZ " " $ $   Jq!&q a 3 &B XQ X X!)K/A*=*= X Xb X X X XYYYYYrhyp_evolved.yamlc Zdt|zt|z}dt|zt|z}dt|z|z}t d|d|d|d|rmd|z}t |t jdrt j dndkrt j d |ztdd 5}| ||zdzdddn #1swxYwYtjtjdd d } | tjt#|  } tjd| dt'|D]"\} } t)| d| dzf|| <#t|d5}t| dddf}dt|z|z}| dt| z|zdzt+j||ddddn #1swxYwY|rt j d|d|dSdS)Nz%10sz%10.3gz%10.4grhz Evolved fitness: zgs://%s/evolve.txtz evolve.txtrzgsutil cp %s .ar)ndminrVrr?z@# Hyperparameter Evolution Results # Generations: %g # Metrics: z F) sort_keyszgsutil cp evolve.txt z gs://)rtuplekeysvaluesrJrr+r,r2rr:rmwriter!uniqueloadtxtrrsavetxtr\ryamldump) hypresults yaml_filebucketrbrrSrrbrZrs rprint_mutationr(0sSE#((**---A3s88eCJJLL111A3w<<')A Eqqq!!!QQQ 7888 ."V+ #  27>>R^C_C_"f"'//,"?"?"?ef g g I&, - - - lC A A  "*\333!<<>EEA1JJ Jc Dt|tjr|gn|}t|D]o\}}|ft |rV|}t |ddddf}|ddddfddd|ddddf<|ddddfdzdz|ddddf<t| |ddddf<t|j dd|ddddf||j |dddf }g}t|D]\} } ||t| dt| dt| dt| df} tj| d } | ddddddd fddd} tj| tj } | d z} || |t)j||jd} |||| k||<q|S) NrrrIrg?rr)r+r)dtypego@)rjr!ndarrayr\rr5r9r* unsqueezer;rrjrrPrresize transposeascontiguousarrayfloat32rPrrtorargmax)rbrimgim0rZdr' pred_cls1imsrrcutoutim pred_cls2s rapply_classifierr=Osc2:.. 73%%CC! 001 =SVV= A!AAArrE(##AABBx||Aq)33A66AaaaeHABBx#~*AaaaeH ||((**Aaaa!eH 122!!!RaR%#a&, ? ? ?!!!Q$ IC!!   1QAaD #ad)) 3S1YYs1Q4yy5H HIZ 33111ddd ^--aA66)"BJ???e  2el3//2218<<==DDQGGIQ4 Y./AaD Hrc:tr|sstStjd}fd|D}d|D}|rt |dznd}|S)N*cRg|]#}tjddjz|$S)z%sz(\d+))rsearchstem)rar7r,seps rrdz"increment_path..ys8LLL!29-3--- 91==LLLrc`g|]+}|t|d,Sr)rPgroups)rams rrdz"increment_path..zs1 6 6 6AA 6SA   6 6 6rrIr)rr2r=r)r*)r,exist_okrCdirsmatchesrZrUs` ` rincrement_pathrKrs ::D !(!DKKMM!4yyyD#))LLLLLtLLL 6 6 6 6 6 "CFFQJJ  Q   r)rrE)r#)r()rXr3)rw)rrr)r)r<r<rr)r<r<)rTr)TFFFrn)FFFFrr)rn)rrNFFr3) rrNFFr3FNN)rr()rr()Tr()Nr)rrr+r9rrrLrpathlibrrnumpyr!pandaspdrrr!utils.google_utilsr utils.metricsrutils.torch_utilsrset_printoptionsroptionsdisplay max_columns setNumThreadsr=rI cpu_countenvironrr"r0r4r>rCrWrvr|rrrryrrrKronesrr1r9r;rCrErNrSr`rjrcrrrrrrrrrrrr(r=rKr3rrr[s       ------!!!!!!......6BBBBclJ(>$?$? !AAAA EEEE ''' ]]]] ,:   666(,,, TTT PPPP JJJ2%%%%&(*                  ZZZZ&&&     (((++++`1111h4442777)))D1T1Te1T1T1T1Th*H*He*H*H*H*HZpu!\\\\~tyBF^^^^B Z Z Z Z KKKK>    F ! ! ! ! ! !r