жfXddlZddlZddlZddlmZddlmZddlmZdZ ddZ dd Z dS)N)kmeans)tqdm)colorstrc|jdd}|d|dz }|jd|jdz }||krYt d|jd|jdd<|jd|jdd<dSdS)NrzReversing anchor order) anchor_gridprodviewstridesignprintanchorsflip)madadss (/var/www/html/yolov7/utils/autoanchor.pycheck_anchor_orderr s 2##B''A 21B " #B wwyyBGGII &'''y~~a(( !!! =--a00 aaa@c >td}td|ddt|dr|jjdn |jd}||jz|jdd z }tj d d |j d df}tj tj dt||z|jDfd}|jdd} || \} } td| dd| dd| dkrtd|jdz} t-|| |dd} n+#t.$r} t|d| Yd} ~ nd} ~ wwxYw|| d }|| krtj | |jj|j} | |j|jdd<t9|| |j|j|jjdddz |jdd<t|dnt|dtddS)N autoanchor:  zAnalyzing anchors... endmodulerTkeepdims?g?r)sizec6g|]\}}|ddddf|zSN.0sls r z!check_anchors..s0%d%d%d1a1Q3i!m%d%d%drcdddf|dz }tj|d|z dd}|dd}|dz kd}|dz k}||fSN?rr )torchminmaxfloatsummean)krxbestaatbprthrwhs rmetriczcheck_anchors..metric s qqq$wK!D' ! Iaa $ $Q ' ' *uuQxx{28|""$$((++0022b3h%%'',,..Cxrr2zanchors/target = .2fz, Best Possible Recall (BPR) = .4fg\(\?z/. Attempting to improve anchors, please wait...F)nimg_sizer?genverbosezERROR: )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 hasattrrmodelshapesr5nprandomuniformshaper3tensor concatenateziplabelsr6rclonecpur numel kmean_anchors ExceptionrrItype_asview_asrr to)datasetrKr?imgszprefixrrLscalerArr>r=naenew_bprr@s ` @r check_anchorsrds" n % %F ,v , , ,"5555")%":":O 2 BA W^ #gn&8&8T&8&J&J JF I  c3fl1oq-A  B BE bn%d%dFUNT[Tb@c@c%d%d%dee f f l l n nBm!!##''))..r155GvgHC Oc O O Oc O O OUWXXXX Tzz ?@@@ ] " "a ' )#GrEsPT^cdddGG ) ) ) V''A'' ( ( ( ( ( ( ( ( )&//!$ S==l7193CDDDLLQYWWG&}}66q}EEAM!!!  q ! ! !"==??2219== AIL\@]@]@b@bceghjk@l@llAIaaaL Vppp q q q q Vhhh i i i "IIIIIs#F99 G!GG!./data/coco.yaml rDTc \dz tddfd}fd}t|trbt|5}t j|tj} dddn #1swxYwYdd lm} | | d d d } n|} | j z| j d d z } tj dt| | jDdkd } | r%t#d| dt%ddkd t#ddt%dd}t)|z d\}}t%|ks,Jt#ddt%|||z}t+jt*jt+jt*j||}tj}|||jddf\}}}}t5t7|d }|D]}tj|}|d krk|||k|z|j|z|zd zd!d}|d kk| |zd"}||}||kr0|| }}d#|d$|_!|r ||||S)%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() r1rc|dddf|dz }tj|d|z dd}||ddfSr0)r3r4r5)r9r@r:r;s rrAzkmean_anchors..metricRs\ qqq$wK!D' ! Iaa $ $Q ' ' *!%%((1+~rctj|tj\}}||kzS)Ndtype)r3rQfloat32r6r8)r9_r<rAr?r@s ranchor_fitnessz%kmean_anchors..anchor_fitnessXsP&au}===rBB4s ))+++11333rc|tj|d}| \}}| k| kz}}t d dd|dd|ddt dd d |d d |d d || kd d dt |D]Y\}}t dt|dt|dfz|t|dz krdndZ|S)Nr zthr=rBz: rCz 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) rMargsortr r6r8r enumerateroundlen) r9r;r<r>r=irFrArEr_r?wh0s r print_resultsz$kmean_anchors..print_results\s bj## $&C..43J%%'',,..S0A0A0F0F0H0H10LS bbSbbbbbbbbbbccc 9919999l99tyy{{l99AG*//++8999>@ B B B BaLL _ _DAq 'U1Q4[[%!++66QQRS^^EEY] ^ ^ ^ ^ ^r)LoaderNr)LoadImagesAndLabelstrainT)augmentrectr r!c6g|]\}}|ddddf|zSr&r)r*s rr.z!kmean_anchors..qs0PPPDAq!AAAqsF)a-PPPrg@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 rjr#g?z(Evolving anchors with Genetic Algorithm:)descg333333?)r4z3Evolving anchors with Genetic Algorithm: fitness = rC)"r isinstancestropenyamlload SafeLoaderutils.datasetsryrLr5rMrRrSrTanyr7r rtstdrr3rQrlrNrPrrangeonesallrandnclipcopyr)pathrErFr?rGrHrnrwf data_dictryr]rLrur,r9distnprshmppbarrmvkgfgrAr_r@rvs ``` @@@@rrXrX>sT" s(C n % %F 4444444          $ $ZZ =1 !DO<<!$$agsC7LAr2q c F!T!T!T U U UD  ! ! GBKKAvllnn _**R..2%5  2FJQNTTUXZ]^^AAvllnn _ffhhl S ) ) ^B   66rwwyyqA!]]VW]]]DI ! a   =  s!BBB)rr)rerfrrrDT) numpyrMr3rscipy.cluster.vqrr utils.generalrrrdrXr)rrrs ######""""""111$$$$Nbbbbbbr