jdZddlZddlmZddlmZmZmZmZm Z ddl Z ddl m Z m Z ddl mZGdd e jZdS) z, Implements the Generalized R-CNN framework N) OrderedDict)DictListOptionalTupleUnion)nnTensor)_log_api_usage_oncec eZdZdZdejdejdejdejddf fd Zejj d Z d d Z xZ S) 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__r rrrr _has_warned)selfrrrr __class__s ^C:\Projects\Historical\venv\Lib\site-packages\torchvision/models/detection/generalized_rcnn.pyrzGeneralizedRCNN.__init__sQ D!!!"  " c|jr|S|SN)training)rlosses detectionss r eager_outputszGeneralizedRCNN.eager_outputs&s = Mrc :|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 isinstancer lenshapetypeappendr enumerateanywheretolistrtensorsrrr image_sizes postprocessupdatejit is_scriptingrwarningswarnr!)rimagestargetstargetr#original_image_sizesimgval target_idxdegenerate_boxesbb_idxdegen_bbfeatures proposalsproposal_lossesr detector_lossesrs rforwardzGeneralizedRCNN.forward.sv = p e%WXXXX%ppF"7OE!%66p  ,,1Jek"o6JgY^Ydggg  e-n`dej`k`k-n-n-noooo68 : :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^// FrYs######55555555555555 ((((((g:g:g:g:g:big:g:g:g:g:r