[jAvdZddlZddlmZddlmZmZddlZddlmZddl m Z Gdd ejZ y) z, Implements the Generalized R-CNN framework N) OrderedDict)OptionalUnion)nn)_log_api_usage_oncec eZdZdZdej dej dej dej ddf fd Zejjd e e ejfd e e e ejfdee e ejfe e e ejfffd Z dd e ejd ee e e ejfdee e ejfe e e ejfffdZxZS)GeneralizedRCNNad Main class for Generalized R-CNN. Args: backbone (nn.Module): rpn (nn.Module): roi_heads (nn.Module): takes the features + the proposals from the RPN and computes detections / masks from it. transform (nn.Module): performs the data transformation from the inputs to feed into the model backbonerpn roi_heads transformreturnNc~t|t|||_||_||_||_d|_y)NF)super__init__rrr r r _has_warned)selfr r r r __class__s l/var/www/html/ia-enhanced/venv/lib/python3.12/site-packages/torchvision/models/detection/generalized_rcnn.pyrzGeneralizedRCNN.__init__s< D!"  " losses detectionsc"|jr|S|SN)training)rrrs r eager_outputszGeneralizedRCNN.eager_outputs,s ==Mrimagestargetsc |jr|tjddn|D]}|d}t|tjrOtjt |j dk(xr|j ddk(d|j d qtjdd t|d g}|D]\}|j d d}tjt |dk(d |j d d|j|d |df^|j||\}}|t|D]\}}|d}|ddddf|ddddfk} | js3tj| jdd d } || j} tjdd| d|d |j|j} t| tjrt!d| fg} |j#|| |\} }|j%| | |j&|\}}|jj)||j&|}i}|j+||j+|tj,j/r,|j0st3j4dd|_||fS|j7||S)a  Args: images (list[Tensor]): images to be processed targets (list[dict[str, tensor]]): ground-truth boxes present in the image (optional) Returns: result (list[BoxList] or dict[Tensor]): the output from the model. During training, it returns a dict[Tensor] which contains the losses. During testing, it returns list[BoxList] contains additional fields like `scores`, `labels` and `mask` (for Mask R-CNN models). NFz0targets should not be none when in training modeboxesz:Expected target boxes to be a tensor of shape [N, 4], got .z0Expected target boxes to be of type Tensor, got zJexpecting the last two dimensions of the Tensor to be H and W instead got r)dimzLAll bounding boxes should have positive height and width. Found invalid box z for target at index 0z=RCNN always returns a (Losses, Detections) tuple in scriptingT)rtorch_assert isinstanceTensorlenshapetypeappendr enumerateanywheretolistr tensorsrr r image_sizes postprocessupdatejit is_scriptingrwarningswarnr)rrrtargetr!original_image_sizesimgval target_idxdegenerate_boxesbb_idxdegen_bbfeatures proposalsproposal_lossesrdetector_lossesrs rforwardzGeneralizedRCNN.forward5s" == e%WX% F"7OE!%6  ,1Jekk"o6JXY^YdYdXeefg  !NtTY{m[\] 79 :C))BC.C MMCA \]`]f]fgigj]k\lm  ! ' 'QQ(8 9  :..9  &/&8 " Fw#(AB<5BQB<#? #'')"[[)9)=)=!)=)DEaHKF,1&M,@,@,BHMM..6Z7LZLXY[ ==0 h -"S(O#45H%)XXfh%H" ?&*nnXy&J\J\^e&f# O^^// **,@  o& o& 99 ! ! ### ]^#' :% %%%fj9 9rr)__name__ __module__ __qualname____doc__rModulerr*r:unuseddictstrr-listrrrtuplerJ __classcell__)rs@rr r sF !))!YY!99 ! 99 !  !  YY3 ,-;?S%,,EV@W;X tC%&T#u||2C-D(EE F<@P:U\\"P:$tC$5678P: tC%&T#u||2C-D(EE F P:rr ) rNr< collectionsrtypingrrr*rutilsrrOr rrrZs0#" (v:biiv:r