QЦiSrSSKrSSKJr SSKJrJrJrJrJ r SSK r SSK J r J r SSK Jr "SS \ R5rg) z, Implements the Generalized R-CNN framework N) OrderedDict)DictListOptionalTupleUnion)nnTensor)_log_api_usage_oncec ^\rSrSrSrS\R S\R S\R S\R SS4 U4S jjr\RRS 5r S S jr S r U=r$)GeneralizedRCNNa@ 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 transformreturnNcv>[TU]5 [U5 X@lXlX lX0lSUlg)NF)super__init__r rrrr _has_warned)selfrrrr __class__s l/var/www/html/ai-image-ml/venv/lib/python3.13/site-packages/torchvision/models/detection/generalized_rcnn.pyrGeneralizedRCNN.__init__s2 D!" " c,UR(aU$U$N)training)rlosses detectionss r eager_outputsGeneralizedRCNN.eager_outputs&s ==Mrc :UR(aUc[R"SS5 OUHnUSn[U[R5(aV[R"[ UR 5S:H=(a UR SS:HSUR S 35 M}[R"SS [U5S 35 M /nUH^nUR S Sn[R"[ U5S:HS UR S S35 URUS US45 M` URX5upUb[U5HupUSnUSS2SS24USS2SS24:*n U R5(dM8[R"U RSS95S S n XJR5n [R"SSU SUS 35 M URUR5n [U [R5(a[!SU 4/5n UR#XU5upUR%XUR&U5unnURR)XR&U5n0nUR+U5 UR+U5 [R,R/5(a2UR0(d[2R4"S5 SUlUU4$UR7UU5$)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)r torch_assert isinstancer lenshapetypeappendr enumerateanywheretolistrtensorsrrr image_sizes postprocessupdatejit is_scriptingrwarningswarnr#)rimagestargetstargetr&original_image_sizesimgval target_idxdegenerate_boxesbb_idxdegen_bbfeatures proposalsproposal_lossesr"detector_lossesr!s rforwardGeneralizedRCNN.forward.s == e%WX%F"7OE!%66  ,1Jekk"o6JXY^YdYdXeefg  e/_`dej`k_llm-no&79C))BC.C MMCA \]`]f]fgigj]k\lm  ! ' 'QQ(8 9 ..9  &/&8" w#(AB<5BQB<#? #''))"[[)9)=)=!)=)DEaHKF,1M,@,@,BHMM..6Z7LZLXY['9==0 h - -"S(O#45H%)XXf%H" &*nnX&J\J\^e&f# O^^// unusedr#rP__static_attributes__ __classcell__)rs@rrrsn !!!ryy!]_]f]f!ko! YYH:H:rr)rVr@ collectionsrtypingrrrrrr/r r utilsr rWrrrr_s3#55 (g:biig:r