U hS.@sdddlZddlmZmZddlZddlmZmZddlmZGdddej Z Gdd d ej Z dS) N)ListOptional)nnTensor) ImageListcseZdZdZdeejiZdfdd Zej e dfee ee ej ej edd d Zej ej d d d Zee dddZeee eeeeedddZeeeeedddZZS)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@csxtt|dttfs.tdd|D}t|dttfsN|ft|}|_|_fddt||D_ dS)Nrcss|] }|fVqdS)N).0srr[/var/www/html/venv/lib/python3.8/site-packages/torchvision/models/detection/anchor_utils.py ,sz+AnchorGenerator.__init__..csg|]\}}||qSr)generate_anchors)rsizeZ aspect_ratioselfrr 2sz,AnchorGenerator.__init__..) super__init__ isinstancelisttuplelensizes aspect_ratioszipr )rr r! __class__rrr#s  zAnchorGenerator.__init__cpu)scalesr!dtypedevicereturnc Cstj|||d}tj|||d}t|}d|}|dddf|dddfd}|dddf|dddfd}tj| | ||gddd} | S)Nr'r(rdim)torch as_tensorsqrtviewstackround) rr&r!r'r(Zh_ratiosZw_ratioswshs base_anchorsrrrr:s &&z AnchorGenerator.generate_anchorsr*csfdd|jD|_dS)Ncsg|]}|jdqS)r*)to)rZ cell_anchorr(r'rrrMsz4AnchorGenerator.set_cell_anchors..)r )rr'r(rr9rset_cell_anchorsLsz AnchorGenerator.set_cell_anchorsr)cCsddt|j|jDS)NcSs g|]\}}t|t|qSrr)rrarrrrPsz.)r"r r!rrrrnum_anchors_per_locationOsz(AnchorGenerator.num_anchors_per_location) grid_sizesstridesr)c Csg}|j}t|dk dtt|t|ko.rSrcsTg|]L}tjdtjdd|dtjdtjdd|dgqS)rr*rr)r/emptyZint64Zfill_)rgr( image_sizerrrws""cSsg|]}|qSrr)rZanchors_per_feature_maprrrrscSsg|]}t|qSr)r/cat)rZanchors_per_imagerrrrs) tensorsrUr'r(r:rNranger image_sizesrG) rrPrQr?r'r@Zanchors_over_all_feature_mapsrH_Zanchors_in_imagerrYrforwardss    zAnchorGenerator.forward)r r )__name__ __module__ __qualname____doc__rr/r__annotations__rfloat32r(intfloatr'rr:r>rNrr` __classcell__rrr#rr s( &rcseZdZdZdeeeeeeeeeeeedfdd Z e j e d fee j e j eed d d Zeed ddZe j feeeeee j edddZed ddZeeeeedddZZS)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??NT)r! min_ratio max_ratior&stepsclipcst|dk r*t|t|kr*td||_||_||_t||dkrdkr|fddtD|_|j dq|g|_n||_| |_ dS)Nz3aspect_ratios and steps should have the same lengthrcs g|]}|dqS)rr)rkrm num_outputsZ range_ratiorrrsz0DefaultBoxGenerator.__init__..r) rrr ValueErrorr!rorpr]r&rG_generate_wh_pairs _wh_pairs)rr!rmrnr&rorpr#rrrrs  zDefaultBoxGenerator.__init__r%)rsr'r(r)c Csg}t|D]}|j|}t|j||j|d}||g||gg}|j|D]@} t| } |j|| } |j|| } || | g| | ggqR|tj|||dq |S)Nrr*) r]r&mathr1r!extendrGr/r0) rrsr'r(rvrqZs_kZ s_prime_kwh_pairsarZsq_arwhrrrrus   z&DefaultBoxGenerator._generate_wh_pairsr;cCsdd|jDS)NcSsg|]}ddt|qS)r.r<)rrrrrrsz@DefaultBoxGenerator.num_anchors_per_location..)r!rrrrr>sz,DefaultBoxGenerator.num_anchors_per_location)r?rZr'r)cCsFg}t|D](\}}|jdk rF|d|j|}|d|j|}n|\}}td|dd|j|d} td|dd|j|d} tj| | dd\} } | d} | d} tj| | ft|j |dddd } |j r|j |j ddd n|j |}| |d|dd}tj | |fdd}||q tj |ddS) Nrrrr'rArBr+r,r.)minmax) enumerateror/rDr8rErFr3rrvrpclamprepeatr[rG)rr?rZr' default_boxesrqZf_kZx_f_kZy_f_krIrJrKrLrMZ_wh_pairryZ default_boxrrr_grid_default_boxess"     ($ z'DefaultBoxGenerator._grid_default_boxesc Cs2|jjd|jd|jd|jd|jd }|S)Nz(aspect_ratios=z, clip=z , scales=z, steps=))r$rar!rpr&ro)rrrrr__repr__s,zDefaultBoxGenerator.__repr__rOc Csdd|D}|jjdd}|dj|dj}}|j|||d}||}g}tj|d|dg|jd} |jD]z} |} t | ddddfd | ddddf| | ddddfd | ddddf| gd } | | qt|S) NcSsg|]}|jddqSrRrTrVrrrrsz/DefaultBoxGenerator.forward..rSrr~r)r(r.rr+) r\rUr'r(rr8r/Ztensorr^r[rG) rrPrQr?rZr'r(rZdboxesZx_y_sizer_Zdboxes_in_imagerrrr`s"  .. zDefaultBoxGenerator.forward)rkrlNNT)rarbrcrdrrgrhrboolrr/rfr(r'rrur>rstrrrr`rirrr#rrjs@       rj) rwtypingrrr/rrrPrModulerrjrrrrs  ~