"fVddlZddlZddlZddlmZddlmZddlmZdZ ddZ d dZ y) N)kmeans)tqdm)colorstrc|jjdjd}|d|dz }|jd|jdz }|j |j k7r\t d|j jd|j dd|jjd|jddyy)NrzReversing anchor order) anchor_gridprodviewstridesignprintanchorsflip)madadss 4C:\work2\python\work_new2\yolov7\utils\autoanchor.pycheck_anchor_orderr s 2##B'A 21B " #B wwyBGGI &'yy~~a( ! ==--a0 ac <td}td|ddt|dr|jjdn|jd}||j z|j j dd z }tjjd d |jd df}tjtjt||z|jD cgc]\}} | ddddf|zc} }j!fd} |j"j%j'j)dd} | | \} } td| dd| dd| dkrdtd|j"j+dz} t-|||dd} | | d }|| kDr tj| |j0j2j5|j0} | j%j7|j"|j"ddt9|| j%j7|j0|j:j=|j0j2j)dddz |j0ddt|dnt|dtdycc} }w#t.$r}t|d|Yd}~[d}~wwxYw) N autoanchor:  zAnalyzing anchors... endmodulerTkeepdims?g?r)sizecVdddf|dz }tj|d|z jdd}|jdd}|dz kDjj dj }|dz kDjj }||fSN?rr)torchminmaxfloatsummean)krxbestaatbprthrwhs rmetriczcheck_anchors..metric s q$wK!D' ! IIaa $ $Q ' *uuQx{28|""$((+002b3h%%',,.Cxrr(zanchors/target = .2fz, Best Possible Recall (BPR) = .4fg\(\?z/. Attempting to improve anchors, please wait...F)nimg_sizer5genverbosezERROR: )devicezSNew anchors saved to model. Update model *.yaml to use these anchors in the future.zKOriginal anchors better than new anchors. Proceeding with original anchors.)rr hasattrrmodelshapesr+nprandomuniformshaper)tensor concatenateziplabelsr,rclonecpur numel kmean_anchors Exceptionrr?type_asview_asrr to)datasetrAr5imgszprefixrrBscaleslr7rr4r3naenew_bprr6s ` @r check_anchorsr\s n %F Bvh+ ,"5")%": 2 BA W^^ #gnn&8&8T&8&J JF II  c3fll1oq-A  BE bnnFUNT[TbTb@c%d@c1a1Q3i!m@c%de f l l nBmm!!#'')..r15GgHC c#Y&Ec#Y OUWX Tz ?@ ]] "a ' )#GrEsPT^cdG/!$ S=ll71993C3CDLLQYYWG&}}66q}}EAMM!  q !"==?22199= AIIL\L\@]@b@bceghjk@llAIIaL VHop q VHgh i "I;&e$ ) VHGA3' ( ( )s+K1*K77 LLLc dz tddfd}fd}t|trNt|5}t j |tj } ddddd lm} |  d d d } n|} | jz| jjd d z } tjt| | jD cgc]\} }|ddddf| zc}} dkjd j!}|rt#d|dt%ddk\jd t#ddt%dj'd} t)| z d\}}t%|k(s!Jt#ddt%||| z}t+j,t*j.t+j,t*j.||}tj0}|||j2ddf\}}}} t5t7|d !}|D]}tj8|}|d k(j;r_|j1||k|j1z|j<|z| zd zj?d"d}|d k(j;r_|jA|zj?d#}||}||kDs||jA}}d$|d%|_!|s||||S#1swYxYwcc}} w)&aD Creates kmeans-evolved anchors from training dataset Arguments: path: path to dataset *.yaml, or a loaded dataset n: number of anchors img_size: image size used for training thr: anchor-label wh ratio threshold hyperparameter hyp['anchor_t'] used for training, default=4.0 gen: generations to evolve anchors using genetic algorithm verbose: print all results Return: k: kmeans evolved anchors Usage: from utils.autoanchor import *; _ = kmean_anchors() r'rc|dddf|dz }tj|d|z jdd}||jddfSr&)r)r*r+)r/r6r0r1s rr7zkmean_anchors..metricRsR q$wK!D' ! IIaa $ $Q ' *!%%(1+~rctj|tj\}}||kDjzj S)Ndtype)r)rGfloat32r,r.)r/_r2r7r5r6s ranchor_fitnessz%kmean_anchors..anchor_fitnessXsCau}}=rB4s ))++1133rcp|tj|jd}| \}}| kDjj | kDjj z}}t d dd|dd|ddt dd d |j d d |j d d || kDj d d dt |D]B\}}t dt|dt|dfz|t|dz krdndD|S)Nrzthr=r8z: r9z best possible recall, z anchors past thrzn=z , img_size=z , metric_all=z.3f/z-mean/best, past_thr=z-mean: rrz%i,%irz, r) rCargsortr r,r.r enumerateroundlen) r/r1r2r4r3ir<r7r;rUr5wh0s r print_resultsz$kmean_anchors..print_results\s9 bjj# $C.43J%%',,.S0A0F0F0H10LS SIRCy0GCyPabc 1#[ -~Qtyy{[^N_`AG*//+C09>@ BaLDAq 'U1Q4[%!+66QQRS^EY] ^!r)LoaderNr)LoadImagesAndLabelstrainT)augmentrectrrr#r$g@z(WARNING: Extremely small objects found. z of z labels are < 3 pixels in size.g@zRunning kmeans for z anchors on z points...)iterz)ERROR: scipy.cluster.vq.kmeans requested z points but returned only r`r!g?z(Evolving anchors with Genetic Algorithm:)descg333333?)r*z3Evolving anchors with Genetic Algorithm: fitness = r9)"r isinstancestropenyamlload SafeLoaderutils.datasetsrorBr+rCrHrIrJanyr-r rjstdrr)rGrbrDrFrrangeonesallrandnclipcopyru)pathr;r<r5r=r>rdrmf data_dictrorSrBrWrXrkr/distnprshmppbarrcvkgfgr7rUr6rls ``` @@@@rrNrN>s"" s(C n %F 4  $ $Z1 !DOOrs* #"1$Nbr