\hM.SSKrSSKJr SSKrSSKJrJr SSKJr "SS\R5r "SS \R5r g) N)Optional)nnTensor) ImageListc ^\rSrSrSrS\\R0rSU4Sjjr \R\R"S54S\\ S\\ S\RS \RS \4 S jjrS\RS \R4S jrS \\ 4S jrS\\\ S\\\S \\4SjrS\S\\S \\4SjrSrU=r$)AnchorGenerator ao 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_anchorsch>[TU]5 [US[[45(d[ SU55n[US[[45(dU4[ U5-nXlX l[X5VVs/sHup4URX45PM snnUl gs snnf)Nrc3&# UHo4v M g7fN).0ss c/var/www/fran/franai/venv/lib/python3.13/site-packages/torchvision/models/detection/anchor_utils.py +AnchorGenerator.__init__..,s.1$s) super__init__ isinstancelisttuplelensizes aspect_ratioszipgenerate_anchorsr )selfrrsize aspect_ratio __class__s rrAnchorGenerator.__init__#s %(T5M22...E-*T5M::*,s5z9M *PSTYPi Pi:L$D ! !$ 5Pi  sB.cpuscalesrdtypedevicereturncn[R"XUS9n[R"X#US9n[R"U5nSU- nUSS2S4USSS24-RS5nUSS2S4USSS24-RS5n[R"U*U*Xx/SS9S- n U R 5$)Nr&r'rdim)torch as_tensorsqrtviewstackround) rr%rr&r'h_ratiosw_ratioswshs base_anchorss rr AnchorGenerator.generate_anchors:sVD 6R ::m,x<q$w&q/1 7 7 ;q$w&q/1 7 7 ;{{RC"b#51=A !!##cjURVs/sHo3RXS9PM snUlgs snf)Nr*)r to)rr&r' cell_anchors rset_cell_anchors AnchorGenerator.set_cell_anchorsLs-[_[l[lm[lK^^%^G[lmms0c[URUR5VVs/sHup[U5[U5-PM snn$s snnfr)rrrr)rras rnum_anchors_per_location(AnchorGenerator.num_anchors_per_locationOs;,/ D)4<DQWLr;r c ^\rSrSrSrSS\\\S\S\S\\\S\\\S\ 4 U4S jjjr \ R\ R"S 54S \S \ RS \ RS\\4SjjrS\\4Sjr\ R4S\\\S\\S \ RS\4SjjrS\4SjrS\S\\S\\4SjrSrU=r$)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%stepsclipc>[T U]5 Ub#[U5[U5:wa [S5eXlXPlX`l[U5nUcZUS:aKX2- n[U5V s/sHoX-US- - -PM sn UlURRS5 OX#/UlOX@lURU5Ul gs sn f)Nz3aspect_ratios and steps should have the same lengthrrx) rrr ValueErrorrrrrhr%rQ_generate_wh_pairs _wh_pairs) rrrrr%rr num_outputs range_ratiokr"s rrDefaultBoxGenerator.__init__s   ]!3s5z!ARS S*  -(  >Q'3 Z_`kZlmZlUV;?kC>O+PPZlm  ""3'(4  K00=ns-C r$rr&r'r(c /n[U5HnURUn[R"URUURUS--5nXf/Xw//nURUHRn [R"U 5n URUU -n URUU - n UR X/X//5 MT UR [R"XUS95 M U$)Nrr*) rhr%mathr1rextendrQr/r0) rrr&r'rrs_k s_prime_kwh_pairsarsq_arwhs rr&DefaultBoxGenerator._generate_wh_pairss#% {#A++a.C $++a.4;;q1u3E"EFI Y$:;H((+ " KKNU*KKNU*!! 01 ,   U__X6R S$r;cdURVs/sHnSS[U5--PM sn$s snf)Nr.)rr)rrs rrC,DefaultBoxGenerator.num_anchors_per_locations.(,(:(:;(:1AAJ(:;;;s-rErlcv/n[U5GHupVURb+USURU- nUSURU- nOUup[R"SUS5S-U- R US9n [R"SUS5S-U- R US9n [R "XSS9upU R S5n U R S5n [R"X4[URU5-SS9R SS 5n UR(aURURSSS 9OURUnURUSUS-S5n[R"X4SS9nURU5 GM [R"USS9$) Nrrrwr&rHrIr+r,r.)minmax) enumeraterr/rMr=rOrPr3rrrclamprepeatrjrQ)rrErlr& default_boxesrf_kx_f_ky_f_krXrYrZr[r\_wh_pairr default_boxs r_grid_default_boxes'DefaultBoxGenerator._grid_default_boxess  +FAzz%"1  1 5"1  1 5" aQ036%?CC%CPHaQ036%?CC%CPH$~~h4P Goob)Goob)G[['!3c$..:K6L!LRTU]]^`bcdF@D t~~a(..1!.K   -+,.yyA..r;c URRSURSURSURSUR S3 nU$)Nz(aspect_ratios=z, clip=z , scales=z, steps=))r"ryrrr%r)rrs r__repr__DefaultBoxGenerator.__repr__sX~~&&'(!//0dii[ }tzzl    r;r_r`c 6UVs/sHo3RSSPM nnURRSSnUSRUSRpvUR XEUS9nUR U5n/n [ R"USUS/URS9n URHen Un [ R"U SS2SS24SU SS2SS24-- U -U SS2SS24SU SS2SS24--U -/S5n U RU 5 Mg U $s snf) Nrbrrr)r'r.rwr+) rcrdr&r'rr=r/tensorrirjrQ) rr_r`rkrErlr&r'rdboxesx_y_sizerodboxes_in_images rrsDefaultBoxGenerator.forwards=@LM '', M''--bc2 $Q--|A/E/Ev00u0U %((0 <<A 1 >}G[G[\''A+O#ii$QU+cOAqrE4J.JJhV$QU+cOAqrE4J.JJhV O MM/ *( %NsD)rrrr%r)g333333?g?NNT)ryrzr{r|r}rrrrboolrr/rr'r&rrrCrstrrrrsrrrs@rrrsF( (,%)>DI>> > e% > S " >>>>6;]][`[g[ghm[n',{{LQLL f(<$s)< X]WdWd/tCy//7;Cy/IN/ /: # )4<DQWLr;r) rtypingrr/rrr_rModuler rrr;rrs9  !{bii{|D"))Dr;