"f"ddlZddlZddlZddlZddlZddlZddlZddlZddlZddl m Z ddl Z ddl Z ddlZddlZddlZddlZddlmZddlmZddlmZej2ddd e j2dd d j4i d ej6j8_e j<dee ejBdejDd<d;dZ#ddZ'dZ(dZ)d?dZ*d@dZ+dZ,dZ-dZ.dZ/dZ0dAdZ1dZ2dBd!Z3d e jhd fd"Z5d#Z6d$Z7d%Z8dCd&Z9dCd'Z:dDd(Z;d)Zd,Z?dGd-Z@dHd.ZAd/ZBd0ZCd1ZDdId2eEfd3ZFdId2eEfd4ZG dJd5ZH dKd6ZIdLd7ZJdMd8ZKd9ZLdNd:ZMy)ON)Path)gsutil_getsize)fitness)init_torch_seedsi@long) linewidth precisionprofile float_kindz{:11.5g})r formatter NUMEXPR_MAX_THREADSctjd|dvrtjytjy)Nz %(message)s)r)formatlevel)logging basicConfigINFOWARN)ranks 1C:\work2\python\work_new2\yolov7\utils\general.py set_loggingr!s. "gogllA3:<<Actj|tjj|t|yN)randomseednpr)r s r init_seedsr"'s& KKIINN4Trctj|dd}|r%t|tjjSdS)Nz /**/last*.ptT recursive)key)globmaxospathgetctime) search_dir last_lists rget_latest_runr/.s7 ZL 5FI3<3ybgg.. /D"Drc4tdjS)Nz /workspace)rexistsrrisdockerr34s   $ $ &&rcttjdk(r |jjddS|S)NWindowsasciiignore)platformsystemencodedecode)strs remojisr=9s/5=__5F)5S3::<  w 1\Y\\rcPddl} |jddy#t$rYywxYw)Nr)z1.1.1.1irTF)socketcreate_connectionOSError)r?s r check_onlinerB>s0  !115 s  %%cttdd tdjsJdt rJdt sJdd}t j|d jjjd}t jd d jj}tt jd |d d }|dkDrd|dd|dkDzd|d}nd|d}tt|y#t$r}t|Yd}~yd}~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 ✅)printcolorstrrr1r3rB subprocess check_outputr;striprstripintr= Exception)cmdurlbranchnrGes rcheck_git_statusrVHs5 (: B'F|""$M&MM$:>>>~~999~?%%c6==?EEGNNvV(()JRVW^^`ffh  ''-x?V(W_cd e q59!GC1q5M?S::=>SUA#3%t,A fQi  asDD(( E1 EEcxddl}tddd}t|ttfrt |}|j s t |d|jdy|j|jDcgc]+}|j|vs|j|j-}}n|Dcgc] }||vs| }}d}|D]} |j||rWd t%vrjn|} |d|dd|dkDzd| d|dtddd } t t'| yycc}wcc}w#t$rf}|dz }t |d|jdt tj d |jd d j#Yd}~d}~wwxYw)Nrredboldz requirements: z not found, check failed.rHz> not found and is required by YOLOR, attempting auto-update...z pip install ''TrEfilez packagerGz updated per  u ⚠️ z;Restart runtime or rerun command for updates to take effect) pkg_resourcesrJ isinstancer<rr1rIresolveparse_requirementsopenname specifierrequirerPreqrKrLr;localsr=) requirementsexcludepkgprefixr\xrTrrUsourcerGs rcheck_requirementsro^s eV_ 5F,d ,L!{{} VHAdlln--FG H :=:P:PQUQZQZQ\:]w:]Qabagagovav166(1;;-0:] w#/D trJdtjdtjdtj dtj tj dy#t$r}td|Yd}~yd}~wwxYw)Nz/cv2.imshow() is disabled in Docker environmentstest)rHrHrHTzVWARNING: Environment does not support cv2.imshow() or PIL Image.show() image displays F) r3cv2imshowr!zeroswaitKeydestroyAllWindowsrPrI)rUs r check_imshowr}sx :PPP~ 6288I./ A  A  ghigjklsA8A;; BBBct|js|dk(r|Stjd|zd}t|s Jd|t|dk(s Jd|d||d S) Nr'z./**/Tr$zFile Not Found: rHzMultiple files match 'z', specify exact path: r)ris_filer(len)r\filess r check_filersy Dztrz  'D.D95z4-dV44z5zQ]"8>UV[U\ ]]Qxrc |jd|jd}}|r[t|rNt|tr|n|gDcgc]}t |j }}t d|Dstd|Dcgc]}|jrt| c}z|rt|rtd|z|jdrb|jdrQt |j}tjj||t!j"d|d |}nt!j"|}td |d k(r d zyd zyt%dyyycc}wcc}w)Nvaldownloadc3<K|]}|jywr)r1).0rls r z check_dataset..s+s!188:ssz2 WARNING: Dataset not found, nonexistent paths: %szDownloading %s ...httpz.zipz unzip -q z -d ../ && rm zDataset autodownload %s rsuccessfailurezDataset not found.)getrr_listrr`allrIr1r< startswithendswithrctorchhubdownload_url_to_filer*r9rP)dictrrGrlfrms r check_datasetrs> XXe_dhhz2C s3x2Q AII221a8 1a"HIA ! A1!q&YXYiXY 455,sTJqs FF  F c8tj||z |zSr)mathceil)rldivisors rrqrqs 99Q[ !G ++rc2tjdd|S)Nu"[|@#!¡·$€%&()=?¿^*;:,¨´><+]_)patternreplstring)resub)rGs r clean_strrs 66>SQR SSrcfdS)Nctdtj|tjzz z dz z zzS)NrH)rcospi)rlstepsy1y2s rzone_cycle..s3q488AK%$788A="r'JROrr2)rrrs```r one_cyclers OOrct|dkDr|ndd|df^}}iddddd d d d dd ddddddddddddddddddd d!d"d#d$d%d&d'd(d)jfd*|D|zd$zS)+NrHbluerYrblackzrXzgreenzyellowzzmagentazcyanzwhitez bright_blackz bright_redz bright_greenz bright_yellowz bright_bluezbright_magentaz bright_cyanz bright_whitezrDzzz)rY underliner'c3(K|] }| ywrr2)rrlcolorss rrzcolorstr..s+d6!9ds)rjoin)inputargsrrs @rrJrJs# Z!^E&&%(1KMT6&gz&Z&z& &j &  & j &z&j&J&j&z&Z& &Z&j& Y!&" $%&F& 77+d+ + 9F5M IIrPcD|dtjStj|d}|dddfj tj }tj ||}d||dk(<d|z }||jz}tj|SNr) minlengthrH) rTensorr! concatenateastypeint32bincountsum from_numpy)labelsncclassesweightss rlabels_to_class_weightsrs ay||~ ^^FA &FQTl!!"((+Gkk'R0G GGqL'kG w{{}G   G $$rc tj|Dcgc]=}tj|dddfjtj|?c}}|j d||zj d}|Scc}wr)r!arrayrrrreshaper)rr class_weightsrl class_counts image_weightss rlabels_to_image_weightsrsq88Z`aZ`UVR[[1a4)ARPZ`abL"**1b1L@EEaHM bsABcgd}|S)N)PrHrrwrr r  !"#$%&'()*+,./0123456789:;<=>?@ACFHIJKLMNOrQRTUVWXYZr2)rls rcoco80_to_coco91_classr$s  aA Hrc`t|tjr|jnt j |}|dddf|dddfzdz |dddf<|dddf|dddfzdz |dddf<|dddf|dddfz |dddf<|dddf|dddfz |dddf<|SNrrrHrwr_rrcloner!copyrlys r xyxy2xywhr,s5<<0 bggajAAw1a4 A%AadGAw1a4 A%AadG1g!Q$AadG1g!Q$AadG Hrclt|tjr|jnt j |}|dddf|dddfdz z |dddf<|dddf|dddfdz z |dddf<|dddf|dddfdz z|dddf<|dddf|dddfdz z|dddf<|Sr&r'r*s r xywh2xyxyr. s5<<0 bggajA1g!Q$! #AadG1g!Q$! #AadG1g!Q$! #AadG1g!Q$! #AadG Hrct|tjr|jnt j |}||dddf|dddfdz z z|z|dddf<||dddf|dddfdz z z|z|dddf<||dddf|dddfdz zz|z|dddf<||dddf|dddfdz zz|z|dddf<|Sr&r'rlwhpadwpadhr+s r xywhn2xyxyr5s5<<0 bggajA1QT7Qq!tWq[()D0AadG1QT7Qq!tWq[()D0AadG1QT7Qq!tWq[()D0AadG1QT7Qq!tWq[()D0AadG Hrct|tjr|jnt j |}||dddfz|z|dddf<||dddfz|z|dddf<|S)NrrHr'r0s rxyn2xyr7se5<<0 bggajA!AqD'kD AadG!AqD'kD AadG HrcD|j\}}|dk\|dk\z||kz||kz}||||}}t|rQtj|j |j |j |j gStj dS)Nr)rHr)Tanyr!rminr)rz)segmentwidthheightrlr+insides r segment2boxr@%s 99DAq1fa AJ /1; ?F fIqyqA=@V288QUUWaeegquuw8 9YRXIYYrcg}|D]^}|j\}}|j|j|j|j|jg`t t j |Sr)r9appendr;r)r,r!r)segmentsboxesrGrlr+s rsegments2boxesrE-s` E ss1 aeegquuw9: RXXe_ %%rc t|D]\}}tj||ddddffd}tjdt |dz |}tj t |}tjt dDcgc] }tj|||dd|f"c}jddj|<|Scc}w)NrrHaxisrr) enumerater!rlinspacerarangerangeinterprr9)rCrTirGrlxps rresample_segmentsrP6s(#1 NNAq1ay> 2 KK3q6A:q ) YYs1v nnq%RAbii2qAw&?%RS[[\]_abdd $ O&Ss%C c<|At|d|dz |d|dz }|d|d|zz dz |d|d|zz dz f}n |dd}|d}|ddddgfxx|dzcc<|ddddgfxx|dzcc<|ddddfxx|zcc<t|||SNrrHrrwr)r; clip_coords) img1_shapecoords img0_shape ratio_padgainpads r scale_coordsrZ@s:a=:a=0*Q-*Q-2OP!}z!}t33q8:a=:VW=[_K_;_cd:dd|Al 1q!f9Q 1q!f9Q 1bqb5MTM # Mrc|dddfjd|d|dddfjd|d|dddfjd|d|dddfjd|dy)NrrHrrw)clamp_)rD img_shapes rrSrSPsr !Q$Kq)A,' !Q$Kq)A,' !Q$Kq)A,' !Q$Kq)A,'rc f|j}|r/|d|d|d|df\}}} } |d|d|d|df\} } } }np|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 j dtj| |tj|| z j dz}| |z | |z |z}}| | z || z |z}}||z||zz|z |z}||z }|s|s|rGtj| | tj|| z }tj| |tj|| z }|s|r|dz|dzz|z}| | z|z | z dz| |z|z | z dzzdz }|r|||z z S|rdt j dzz tjtj|||zz tj|||zz z dz}tj5|||z d|zzz }ddd|||z |zzz S||z|z}|||z |z z Sy|S#1swY-xYwrR) r9rr;r)clamprrpowatanno_grad)box1box2x1y1x2y2GIoUDIoUCIoUepsb1_x1b1_y1b1_x2b1_y2b2_x1b2_y1b2_x2b2_y2interw1h1w2h2unionioucwchc2rho2valphac_areas rbbox_iourXs/ 66D%)!Wd1gtAwQ%G"ueU%)!Wd1gtAwQ%G"ueUAwa1,d1gQ! .CuAwa1,d1gQ! .CuAwa1,d1gQ! .CuAwa1,d1gQ! .CuYYue $uyy'> > E Ea H YYue $uyy'> > E Ea H IEU]EEMC/B U]EEMC/B Gb2g  % +E %-C tt YYue $uyy'> > YYue $uyy'> > 4q27"S(BU]U*U2q8U]U*U2q89<=>DTBY&A%5::bBHo3NQVQ[Q[\^bdgjbj\kQl3lno)pp]]_SAG!45E%dRi!e)344"Ws]F&5.F22 2 %_s ,J''J0ct|j}|r/|d|d|d|df\}} } } |d|d|d|df\} } }}np|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 j dtj| |tj| | z j 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 tj tj||z tj||z z dz}tj5|d|z||z z |zz }ddd|||z tj |z|z|zz Stj||z|z|}|tj ||z |z |z|z Sy|S#1swYoxYwrR) r9rr;r)r_r`absrrrarb) rcrdrerfrgrhr~rirjrkrlrmrnrorprqrrrsrtrurvrwrxryrzr{rho_xrho_yr|r} alpha_ciours rbbox_alpha_iours 66D%)!Wd1gtAwQ%G"ueU%)!Wd1gtAwQ%G"ueUAwa1,d1gQ! .CuAwa1,d1gQ! .CuAwa1,d1gQ! .CuAwa1,d1gQ! .CuYYue $uyy'> > E Ea H YYue $uyy'> > E Ea H IEU]EEMC/B U]EEMC/B Gb2g  % +E ))E%K#%u -C tt YYue $uyy'> > YYue $uyy'> > 4'B!G#-3BIIeeme3e;.box_area%AQCFSVO44rNrrr9rr;r)r_prod)rcrdrarea1area2rrs rbox_iours5 TVV E TVV EYYtAtQRK($q!"u+ 644QSRSQS CTVZ[\^`_`^`[`Va9b b i ijk l q qrs tE E!T'NU*U2 33rc|dddf}|d}tj||jd}||jd|jdz|z z S)Nr)rr;r)wh1wh2rrs rwh_ioursX ag,C d)C IIc3  $ $Q 'E CHHQK#((1+-5 66rcpd}||j}||j}tj|dddddf|ddddftj|dddddf|ddddfz j dj d}|dddf|z|z }||z }tj|dddddf|ddddf}tj|dddddf|ddddf} | |z j 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 c0|d|dz |d|dz zSrr2rs rrzbox_giou..box_arearrNrrr;rHr) rcrdrrrrrrwrxltirbiwhiareais rbox_gioursT5 TVV E TVV E YYtAtQRK($q!"u+ 644QSRSQS CTVZ[\^`_`^`[`Va9b b i ijk l q qrs tE 1d7^e #e +E %-C ))DD"1"%tArrE{ 3C ))DD!"%tAqrE{ 3C 9    "C 1aL3q!Qw< 'E %%-5( ((rric(d}||j}||j}tj|dddddf|ddddftj|dddddf|ddddfz j dj d}|dddf|z|z }||z }tj|dddddf|ddddf} tj|dddddf|ddddf} | | z j d} | dddddfdz| dddddfdzz|z} |ddddf|ddddfzdz } |ddddf|ddddfzdz }|dddf|dddfzdz }|dddf|dddfzdz }| |z dz||z dzz}|ddddf|ddddfz }|ddddf|ddddfz }|dddf|dddfz }|dddf|dddfz }dtj dzz tjtj||z tj||z z dz}tj5|d|z |z|zz }ddd||| z z |zz S#1swYxYw) 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 c0|d|dz |d|dz zSrr2rs rrzbox_ciou..box_area rrNrrrrHrwr) r9rr;r)r_rrr`rarb)rcrdrirrrrrrwrxrrrdiagonal_distance_squaredx_py_px_gy_gcenters_distance_squaredw_predh_predw_gth_gtr}r~s rbox_ciours5 TVV E TVV E YYtAtQRK($q!"u+ 644QSRSQS CTVZ[\^`_`^`[`Va9b b i ijk l q qrs tE 1d7^e #e +E %-C ))DD"1"%tArrE{ 3C ))DD!"%tAqrE{ 3C 9    "C!$Q1W!2s1a7|q7H IC O 4 d1dA:. .! 3C 4 d1dA:. .! 3C 1:QT "a 'C 1:QT "a 'C #c a/392BB !T1* QaZ 0 0F !T1* QaZ 0 0F 1:QT "D 1:QT "D ehh!m  5::dTk+BUZZPVY_P_E`+`cd eeA QWq[3&'  *-FF G%RS) SS s #JJcVd}||j}||j}tj|dddddf|ddddftj|dddddf|ddddfz j dj d}|dddf|z|z }||z }tj|dddddf|ddddf} tj|dddddf|ddddf} | | z j 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 c0|d|dz |d|dz zSrr2rs rrzbox_diou..box_areaArrNrrrrHrwr)rcrdrirrrrrrwrxrrrrrrrrrs rbox_diour3s 5 TVV E TVV E YYtAtQRK($q!"u+ 644QSRSQS CTVZ[\^`_`^`[`Va9b b i ijk l q qrs tE 1d7^e #e +E %-C ))DD"1"%tArrE{ 3C ))DD!"%tAqrE{ 3C 9    "C!$Q1W!2s1a7|q7H IC O 4 d1dA:. .! 3C 4 d1dA:. .! 3C 1:QT "a 'C 1:QT "a 'C #c a/392BB *-FF GGrc|jddz }|d|kD}d\} } d} d} d} d}||d kDz}d }tj}tjd |j g|jd z}t |D]m\}}|||}|rt ||r||}tjt ||dzf|j }|ddd df|ddddf<d|dddf<d|tt ||ddd fjdzf<tj||fd }|jd s|d k(r|ddddf|ddddf<n|ddddfxx|ddddfzcc<t|ddddf}|rf|ddddf|kDjd j\}}tj|||||dzdf|dddfjfd }n[|ddddfjd d\}}tj|||jfd |jd|kD}|?||ddddftj ||j k(j#d }|jd }|s*|| kDr||dddfj%dd| }|ddddf|rd n| z}|ddddf|z|dddf}}t&j(j+|||}|jd | kDr|d| }|rd |cxkrdkr~nn{t-||||kD}||dz}tj.||ddddfj|j1d dz ||ddf<|r||j1d d kD}||||<tj|z | kDs^t3d| d|S|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$@TrHFrrdevicerNr?as_tuplekeepdimrr descendingp@WARNING: NMS time limit s exceededshapetimerrzrrIrrLrcatr.nonzeror9floatr)viewtensorr:argsort torchvisionopsnmsrmmrrI) prediction conf_thres iou_thresragnostic multi_labelrrxcmin_whmax_whmax_detmax_nms time_limit redundantmergetoutputxirllr}rrNjconfrTcrDscoresrxrs rnon_max_suppressionr`s7   ! q B F j (BNFFGGJI26K E Akk&):):; AAaCyAa!eHAadG36AeCFmQq!tW\\^a// 0 1a&!$Awwqz  7AaCyAaeH aeH!QqS& !H!RaR%! aeHz)22E2BDDDAq 3q61QAt^#4a4j6F6F6HI1MA12hll1dl3GD! 3aggi0!4TYYr]Z5OPA  1Q!V9 WQXX FFKKANOA GGAJ  [!AqD'//T/28G<=A a1fIhF 3!RaR%1 a1gv OO  vy 9 771: (7 A a!kck%(E*Y6CF4L(Gxx1bqb5288:W[[TX[=YYAa!eHcggaj1n%qTr IIK!Oz ) ,ZL C D  MM'L Mrc |&|s|jddz n|jddz }|d|kD} d\} } d} d}d }d }||d kDz}d }tj}tjd |jg|jdz}t |D]\}}|| |}|rt ||r||}tjt ||dzf|j}|ddd df|ddddf<d|dddf<d|tt ||dddfjdzf<tj||fd}|jds|dddd|zfxx|ddddfzcc<t|ddddf}|rf|ddddf|kDjd j\}}tj|||||dzdf|dddfjfd }n|s\|ddddfjd d \}}tj|||jfd |jd|kD}ng|ddddf}|ddddfjd d \}}tj|||j|fd |jd|kD}|?||ddddftj ||jk(j#d }|jd}|s}||kDr||dddfj%d d|}|ddddf|rdn| z}|ddddf|z|dddf} }t&j(j+|| |}|jd| kDr|d| }|rd |cxkrdkr~nn{t-||||kD}!|!| dz}"tj.|"|ddddfj|"j1d d z ||ddf<|r||!j1d d kD}||||<tj|z |kDst3d|d|S|S)rNrrrrrrrrTrHFrrrrrrrrrrrrrr)#rrrrrrr kpt_labelrnkptrrrrrrrrrrrrlrr}rrNrrkptsrTrrDrrxrs# rnon_max_suppression_kptrs z-6Z  a 1 $JN>Nq>Q QF:&A bfI c&*or A SVR!V,QXX>AAaCyAa!eHAadG36AeCFmQq!tW\\^a// 0 1a&!$Awwqz  !QqtV) !QqS& ! !RaR%! aeHz)22E2BDDDAq 3q61QAt^#4a4j6F6F6HI1MAAqrE(,,q$,7aIIsD!'')4a829STABxAqsF)--4-8aIIsD!'')T:A>tyy}z?YZ  1Q!V9 WQXX FFKKANOA GGAJ  [!AqD'//T/28G<=A a1fIhF 3!RaR%1 a1gv OO  vy 9 771: (7 A a!kck%(E*Y6CF4L(Gxx1bqb5288:W[[TX[=YYAa!eHcggaj1n%qTr IIK!Oz ) ,ZL C D  MQ'P Mrc tj|tjd}|jdr|d|d<dD]}d||< d|d<|dj |dj D] }d|_ tj||xs|tjj|xs|d z }td |d |rd |znd d|ddy)Ncpu) map_locationemamodel) optimizertraining_resultswandb_idrupdatesrepochFg.AzOptimizer stripped from ,z saved as %s,r'rZz.1fMB) rloadrrhalf parameters requires_gradsaver*r+getsizerI)rrGrlkpmbs rstrip_optimizerr s 15<<#67AuuU|uX' J!KAgJgJOO wZ " " $% JJq!&q a 3 &B $QCq!/A*=(LAbQTXUW XYrc dt|zt|jz}dt|zt|jz}dt|z|z}t d|d|d|d|rid|z}t |t jjdrt jjdndkDrt jd |ztdd 5}|j||zdzdddtjtjdd d } | tj t#|  } tj$d| dt'|jD]\} } t)| d| dzf|| <t|d5}t| dddf}dt|z|z}|jdt| z|zdzt+j,||dddd|rt jd|d|yy#1swY8xYw#1swY5xYw)Nz%10sz%10.3gz%10.4gr]z Evolved fitness: zgs://%s/evolve.txtz evolve.txtrzgsutil cp %s .ar)ndminrGrr1z@# Hyperparameter Evolution Results # Generations: %g # Metrics: z F) sort_keyszgsutil cp evolve.txt z gs://)rtuplekeysvaluesrIrr*r+r1rr9rbwriter!uniqueloadtxtrrsavetxtrIryamldump) hypresults yaml_filebucketrbrrRrrlrNrs rprint_mutationr0sSE#((*--A3s8eCJJL11A3w<')A q!Q 78 "V+ # 277>>R^C_"''//,"?ef g II&, - lC A A  ! "**\3!)FFFFrg& .>)r7)??NFFr2) r8r9NFFr2FNN)zbest.ptr')zhyp_evolved.yamlr')Tr')Nr(rrr*r8rrrKrpathlibrrxnumpyr!pandaspdrrr utils.google_utilsr utils.metricsrutils.torch_utilsrset_printoptionsroptionsdisplay max_columns setNumThreadsr<r; cpu_countenvironrr"r/r3r=rBrVrortr}rrrqrrrJronesrr$r,r.r5r7r@rErPrZrSrrrrrrrrrrrrr(r3r2rrrIs       -!.6BclJ$? !A E ' ] ,: 6(, T P J2%&(*      Z&  (+`1h427)D1Te1Th*He*HZpu!\~tyBF^B Z K>  F !r