^jN.ddlZddlmZddlZddlmZmZddlmZGddejZ Gdd ejZ y) N)Optional)nnTensor) ImageListc peZdZdZdeej iZ dfd Zejejdfdee dee dejdejd ef d Zdejdejfd Zd ee fd Zd eee deeed eefdZdedeed eefdZxZS)AnchorGeneratora Module that generates anchors for a set of feature maps and image sizes. The module supports 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_anchorscXt|t|dttfst d|D}t|dttfs|ft |z}||_||_t||Dcgc]\}}|j||c}}|_ ycc}}w)Nrc3"K|]}|f ywN).0ss t/var/www/ramen.bs-engineer-server.com/venv/lib/python3.12/site-packages/torchvision/models/detection/anchor_utils.py z+AnchorGenerator.__init__..,s.11$.s ) super__init__ isinstancelisttuplelensizes aspect_ratioszipgenerate_anchorsr )selfrrsize aspect_ratio __class__s rrzAnchorGenerator.__init__#s %(T5M2...E-*T5M:*,s5z9M *PSTY[hPi :L$ D ! !$ 5  sB&cpuscalesrdtypedevicereturnc~tj|||}tj|||}tj|}d|z }|dddf|dddfzjd}|dddf|dddfzjd}tj| | ||gddz } | j S)Nr#r$rdim)torch as_tensorsqrtviewstackround) rr"rr#r$h_ratiosw_ratioswshs base_anchorss rrz AnchorGenerator.generate_anchors:suVD U6R ::m,x<q$w&q/1 7 7 ;q$w&q/1 7 7 ;{{RC"b"#51=A !!##cn|jDcgc]}|j||c}|_ycc}w)Nr')r to)rr#r$ cell_anchors rset_cell_anchorsz AnchorGenerator.set_cell_anchorsLs+[_[l[lmK[^^%^Gmms2ct|j|jDcgc]\}}t|t|zc}}Scc}}wr )rrrr)rras rnum_anchors_per_locationz(AnchorGenerator.num_anchors_per_locationOs7,/ DrWrrl __classcell__r s@rr r s * U\\*O !( 6#]]+u||E2 $S $E{${{ $  $  $$nekkn5<<nQ$s)Q tDId6lASX\]cXd>)4<DQWLr7r c eZdZdZ ddeeedededeeedeeedef fd Z e je jd fd ed e jd e jd eefdZd eefdZe jfdeeedeed e jd efdZd efdZdedeed eefdZxZS)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)``. r min_ratio max_ratior"stepsclipct ||"t|t|k7r td||_||_||_t|}|Z|dkDrK||z }t|D cgc]} ||| z|dz z zc} |_|jjdn||g|_n||_|j||_ ycc} w)Nz3aspect_ratios and steps should have the same lengthrrp) rrr ValueErrorrrrrar"rK_generate_wh_pairs _wh_pairs) rrr}r~r"rr num_outputs range_ratiokr s rrzDefaultBoxGenerator.__init__s   ]!3s5z!ARS S*  -(  >Q')3 Z_`kZlmUVy;?kC>O+PPm  ""3'()4  DK00=ns-Cr!rr#r$r%cg}t|D]}|j|}tj|j||j|dzz}||g||gg}|j|D]R} tj| } |j|| z} |j|| z } |j | | g| | ggT|j tj||||S)Nrr') rar"mathr.rextendrKr,r-) rrr#r$rrs_k s_prime_kwh_pairsarsq_arwhs rrz&DefaultBoxGenerator._generate_wh_pairss#% {# TA++a.C $++a.4;;q1u3E"EFIc Y $:;H((+ 2 " KKNU*KKNU*!Q!Q 01  2   U__XU6R S Tr7c^|jDcgc]}ddt|zzc}Scc}w)Nr+)rr)rrs rr>z,DefaultBoxGenerator.num_anchors_per_locations((,(:(:;1AAJ;;;s*r?recg}t|D]\}}|j+|d|j|z }|d|j|z }n|\}}tjd|ddz|z j |} tjd|ddz|z j |} tj | | d\} } | j d} | j d} tj| | ft|j|zdj dd } |jr |j|jdd n|j|}|j|d|dzd}tj| |fd}|j|tj|dS) Nrrror#rBrCr(r)r+)minmax) enumeraterr,rGr9rIrJr0rrrclamprepeatrcrK)rr?rer# default_boxesrf_kx_f_ky_f_krRrSrTrUrV_wh_pairr default_boxs r_grid_default_boxesz'DefaultBoxGenerator._grid_default_boxess  + .FAszz%"1  1 5"1  1 5" uaQ036%?CC%CPHaQ036%?CC%CPH$~~h4P GWoob)Goob)G[['7!3c$..:K6L!LRTU]]^`bcdF@D t~~a(..1!.K   -+ ..yyA..r7c |jjd|jd|jd|jd|j d }|S)Nz(aspect_ratios=z, clip=z , scales=z, steps=))r rqrrr"r)rrs r__repr__zDefaultBoxGenerator.__repr__sX~~&&'(!//0dii[ }tzzl    r7rXrYc :|Dcgc]}|jdd}}|jjdd}|dj|dj}}|j |||}|j |}g} t j|d|dg|j} |jD]c} |} t j| ddddfd| ddddfzz | z| ddddfd| ddddfzz| zgd} | j| e| Scc}w) Nr[rrr)r$r+ror() r\r]r#r$rr9r,tensorrbrcrK) rrXrYrdr?rer#r$rdboxesx_y_sizerhdboxes_in_images rrlzDefaultBoxGenerator.forwardsG@LMk'',M M''--bc2 $Q--|A/E/Ev00Zu0U %((0 <<A 1 >}G[G[\'' +A+O#ii$QU+cOAqrE4J.JJhV$QU+cOAqrE4J.JJhV O MM/ * + %NsD)g333333?g?NNT)rqrrrsrtrrwrxrboolrr,rvr$r#rrr>rstrrrrlryrzs@rr|r|s?( (,%)>DI>> > e% > S " >>>6;]][g[`[g[ghm[n',{{LQLL f(<$s)< X]WdWd/tCy//7;Cy/IN/ /: # )4<DQWLr7r|) rtypingrr,rrrXrModuler r|rr7rrs9  !{bii{|D"))Dr7