%`iAtdZddlZddlmZddlmZmZddlZddlmZddl m Z Gdd ej Z dS) z, Implements the Generalized R-CNN framework N) OrderedDict)OptionalUnion)nn)_log_api_usage_oncec eZdZdZdejdejdejdejddf fd Zejj d e e ej fd e e e ej fdee e ej fe e e ej fffd Z dd e ej d ee e e ej fdee e ej fe e e ej fffdZxZS)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 transformreturnNctt|||_||_||_||_d|_dS)NF)super__init__rrr r r _has_warned)selfr r r r __class__s /home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/torchvision/models/detection/generalized_rcnn.pyrzGeneralizedRCNN.__init__sS D!!!"  " losses detectionsc|jr|S|SN)training)rrrs r eager_outputszGeneralizedRCNN.eager_outputs,s = Mrimagestargetsc :|jr|tjddn|D]}|d}t|tjrFtjt |jdko|jddkd|jd jtjdd t|d g}|D]j}|jd d}tjt |dkd |jd d||d |dfk| ||\}}|t|D]\}}|d}|ddddf|ddddfk} | rjtj | dd d } ||  } tjdd| d|d ||j} t| tjrt!d| fg} ||| |\} }|| | |j|\}}|j ||j|}i}||||tjr&|jst3jdd|_||fS|||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%WXXXX%  F"7OE!%66   ,,1Jek"o6JgY^Ydggg  !]tTY{{]]] 79 : :C)BCC.C MCA m]`]fgigjgj]kmm    ! ' 'QQ(8 9 9 9 9..99  &/&8&8  " Fw#(ABB<5BQB<#? #''))"[)9)=)=!)=)D)DEEaHKF,1&M,@,@,B,BHM[.6[[MW[[[ ==00 h - - 6"S(O#455H%)XXfh%H%H" ?&*nnXy&J\^e&f&f# O^// *,@   o&&& o&&& 9 ! ! # # :# ( ]^^^#' :% %%%fj99 9rr)__name__ __module__ __qualname____doc__rModulerr*r:unuseddictstrr-listrrrtuplerJ __classcell__)rs@rr r s  !)!Y!9 ! 9 !  !!!!!!  Y3 ,-;?S%,EV@W;X tC%&T#u|2C-D(EE F<@P:P:U\"P:$tC$5678P: tC%&T#u|2C-D(EE F P:P:P:P:P:P:P:P:rr ) rNr< collectionsrtypingrrr*rutilsrrOr rrrZs######"""""""" ((((((v:v:v:v:v:biv:v:v:v:v:r