%`iM.ddlZddlmZddlZddlmZmZddlmZGddejZ Gdd ejZ dS) N)Optional)nnTensor) ImageListc reZdZdZdeejiZ dfd Zej ej dfdee dee d ej d ej d ef d Zd ej d ej fd Zd ee fdZdeee deeed eefdZdedeed eefdZxZS)AnchorGeneratora Module that generates anchors for a set of feature maps and image sizes. The module support computing anchors at multiple sizes and aspect ratios per feature map. This module assumes aspect ratio = height / width for each anchor. sizes and aspect_ratios should have the same number of elements, and it should correspond to the number of feature maps. sizes[i] and aspect_ratios[i] can have an arbitrary number of elements, and AnchorGenerator will output a set of sizes[i] * aspect_ratios[i] anchors per spatial location for feature map i. Args: sizes (Tuple[Tuple[int]]): aspect_ratios (Tuple[Tuple[float]]): cell_anchors)i)??g@ctt|dttfst d|D}t|dttfs|ft |z}|_|_fdt||D_ dS)Nrc3K|]}|fVdS)N).0ss }/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/torchvision/models/detection/anchor_utils.py z+AnchorGenerator.__init__..,s$..11$......cBg|]\}}||Sr)generate_anchors)rsize aspect_ratioselfs r z,AnchorGenerator.__init__..2s:   :L$ D ! !$ 5 5   r) super__init__ isinstancelisttuplelensizes aspect_ratioszipr )rr%r& __class__s` rr zAnchorGenerator.__init__#s %(T5M22 /.......E-*T5M:: :*,s5zz9M *    PSTY[hPiPi   rcpuscalesr&dtypedevicereturnctj|||}tj|||}tj|}d|z }|dddf|dddfzd}|dddf|dddfzd}tj| | ||gddz } | S)Nr+r,rdim)torch as_tensorsqrtviewstackround) rr*r&r+r,h_ratiosw_ratioswshs base_anchorss rrz AnchorGenerator.generate_anchors:suVDDD U6RRR :m,,x<qqq$w&qqq/1 7 7 ; ;qqq$w&qqq/1 7 7 ; ;{RC"b"#51===A !!###rc<fd|jD|_dS)Nc>g|]}|S)r/)to)r cell_anchorr,r+s rrz4AnchorGenerator.set_cell_anchors..Ms)mmmK[^^%^GGmmmr)r )rr+r,s ``rset_cell_anchorsz AnchorGenerator.set_cell_anchorsLs-mmmmm[_[lmmmrcJdt|j|jDS)NcRg|]$\}}t|t|z%Srr$)rras rrz.Ps,PPPDAqAQPPPr)r'r%r&rs rnum_anchors_per_locationz(AnchorGenerator.num_anchors_per_locationOs%PPC D grid_height grid_width stride_height stride_widthr,shifts_xshifts_yshift_yshift_xshiftss r grid_anchorszAnchorGenerator.grid_anchorsTs(  l$.0QRRR  OOs7|| @ @ @ @s.t$MMMk',MMMrrhrcg|]}tjdtjd|dztjdtjd|dzgS)rr/rr)r4emptyint64fill_)rgr, image_sizes rrz+AnchorGenerator.forward..ws    Bek&AAAGG ST YZ[\Y]H]^^ Bek&AAAGG ST YZ[\Y]H]^^    rcg|]}|Srr)ranchors_per_feature_maps rrz+AnchorGenerator.forward..suuu.s$QQQ4E59.//QQQr) tensorsrjr+r,rCrbranger$ image_sizesrV) rrcrdrJr+rKanchors_over_all_feature_mapsrW_anchors_in_imager,rss @@rforwardzAnchorGenerator.forwardss MM MMM '-bcc2 $Q-|A/E v          eV,,,(,(9(9*g(N(N%,.s:12233 - -AuuWtuuu  NN+ , , , ,QQQQQr)r r)__name__ __module__ __qualname____doc__r"r4r__annotations__r float32r,intfloatr+rrCrIrbrr __classcell__r(s@rr r s* U\*O !(      6#]+u|E22 $$S $E{${ $  $  $$$$$nekn5<nnnnQ$s)QQQQ tDId6lASX\]cXd>)4<DQWLrr c eZdZdZ ddeeededed eeed eeed ef fd Z e j e j d fdede j de j deefdZdeefdZe j fdeeedeede j defdZdefdZdedeedeefdZxZS)DefaultBoxGeneratora This module generates the default boxes of SSD for a set of feature maps and image sizes. Args: aspect_ratios (List[List[int]]): A list with all the aspect ratios used in each feature map. min_ratio (float): The minimum scale :math:` ext{s}_{ ext{min}}` of the default boxes used in the estimation of the scales of each feature map. It is used only if the ``scales`` parameter is not provided. max_ratio (float): The maximum scale :math:` ext{s}_{ ext{max}}` of the default boxes used in the estimation of the scales of each feature map. It is used only if the ``scales`` parameter is not provided. scales (List[float]], optional): The scales of the default boxes. If not provided it will be estimated using the ``min_ratio`` and ``max_ratio`` parameters. steps (List[int]], optional): It's a hyper-parameter that affects the tiling of default boxes. If not provided it will be estimated from the data. clip (bool): Whether the standardized values of default boxes should be clipped between 0 and 1. The clipping is applied while the boxes are encoded in format ``(cx, cy, w, h)``. 333333??NTr& min_ratio max_ratior*stepsclipct|/t|t|krtd||_||_||_t||RdkrB|z fdtD|_|j dn|g|_n||_| |_ dS)Nz3aspect_ratios and steps should have the same lengthrc,g|]}|zdz z zS)rr)rkr num_outputs range_ratios rrz0DefaultBoxGenerator.__init__..s-mmmUVy;?kC>O+PPmmmrr) rr r$ ValueErrorr&rrrzr*rV_generate_wh_pairs _wh_pairs) rr&rrr*rrrrr(s ` @@rr zDefaultBoxGenerator.__init__s   ]!3!3s5zz!A!ARSS S*  -((  >Q')3 mmmmmmZ_`kZlZlmmm  ""3''''()4  DK00==rr)rr+r,r-cg}t|D]}|j|}tj|j||j|dzz}||g||gg}|j|D]Q} tj| } |j|| z} |j|| z } || | g| | ggR|tj||||S)Nrr/) rzr*mathr6r&extendrVr4r5) rrr+r,rrs_k s_prime_kwh_pairsarsq_arwhs rrz&DefaultBoxGenerator._generate_wh_pairss#% {## T TA+a.C $+a.4;q1u3E"EFFIc Y $:;H(+ 2 2 " KNU*KNU*!Q!Q 01111   U_XU6RRR S S S Src$d|jDS)Nc8g|]}ddt|zzS)r3rF)rrs rrz@DefaultBoxGenerator.num_anchors_per_location..s&;;;1AAJ;;;r)r&rHs rrIz,DefaultBoxGenerator.num_anchors_per_locations;;(:;;;;rrJrscg}t|D]\}}|j-|d|j|z }|d|j|z }n|\}}tjd|ddz|z |} tjd|ddz|z |} tj| | d\} } | d} | d} tj| | ft|j |zddd } |j r"|j | dd n |j |}| |d|dzd}tj | |fd}||tj |dS) Nrrrr+rMrNr0r1r3)minmax) enumeraterr4rRrArTrUr8r$rrclamprepeatrwrV)rrJrsr+ default_boxesrf_kx_f_ky_f_kr]r^r_r`ra_wh_pairr default_boxs r_grid_default_boxesz'DefaultBoxGenerator._grid_default_boxess  ++ . .FAsz%"1  1 5"1  1 5" uaQ0036%?CC%CPPHaQ0036%?CC%CPPH$~h4PPP GWoob))Goob))G['7!3c$.:K6L6L!LRTUUU]]^`bcddF@D `t~a(..1!.<<>>K   - - - -yA....rc d|jjd|jd|jd|jd|jd }|S)Nz(aspect_ratios=z, clip=z , scales=z, steps=))r(rr&rr*r)rrs r__repr__zDefaultBoxGenerator.__repr__sf~&  !/  i     z     rrcrdc d|D}|jjdd}|dj|dj}}||||}||}g}t j|d|dg|j} |jD]r} |} t j | ddddfd| ddddfzz | z| ddddfd| ddddfzz| zgd } | | s|S) Nc,g|]}|jddSrgrirks rrz/DefaultBoxGenerator.forward..rmrrhrrr)r,r3rr0) ryrjr+r,rrAr4tensorr{rwrV) rrcrdrJrsr+r,rdboxesx_y_sizer}dboxes_in_images rrzDefaultBoxGenerator.forwardsLMM MMM '-bcc2 $Q-|A/Ev00Zu0UU %((00 <A 1 >}G[\\\' + +A+O#i$QQQU+cOAAAqrrE4J.JJhV$QQQU+cOAAAqrrE4J.JJhV O MM/ * * * * r)rrNNT)rrrrr"rrrboolr r4rr,r+rrrIrstrrrrrrs@rrrs( (,%)>>DI>> > e% > S " >>>>>>>>6;][g[`[ghm[n[n',{LQL f(<$s)<<<< X]Wd//tCy//7;Cy/IN/ ////: #    )4<DQWLrr) rtypingrr4rrrcrModuler rrrrrs  !!!!!!{{{{{bi{{{|DDDDD")DDDDDr