o V h@sndZddlZddlmZddlmZmZmZmZm Z ddl Z ddl m Z m Z ddl mZGdd d e jZdS) z, Implements the Generalized R-CNN framework N) OrderedDict)DictListOptionalTupleUnion)nnTensor)_log_api_usage_oncec sXeZdZdZdejdejdejdejddf fdd Zejj d d Z dd d Z Z 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 transformreturnNcs4tt|||_||_||_||_d|_dS)NF)super__init__r rr rr _has_warned)selfr rrr __class__a/var/www/vscode/kcb/lib/python3.10/site-packages/torchvision/models/detection/generalized_rcnn.pyrs  zGeneralizedRCNN.__init__cCs|jr|S|SN)training)rlosses detectionsrrr eager_outputs&szGeneralizedRCNN.eager_outputsc Cs*|jrA|durtddn3|D]0}|d}t|tjr4tt|jdko+|jddkd|jd qtdd t|d qg}|D]&}|jd d}tt|dkd |jd d||d |dfqE| ||\}}|durt |D]@\}}|d}|ddddf|ddddfk} | rt | j ddd d } ||  } tdd| d|d q|||j} t| tjrtd| fg} ||| |\} }|| | |j|\}}|j ||j|}i}||||tjr|js tdd|_||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 enumerateanywheretolistr tensorsrrr image_sizes postprocessupdatejit is_scriptingrwarningswarnr)rimagestargetstargetroriginal_image_sizesimgval target_idxdegenerate_boxesbb_idxdegen_bbfeatures proposalsproposal_lossesrdetector_lossesrrrrforward.sf   (        zGeneralizedRCNN.forwardr) __name__ __module__ __qualname____doc__rModulerr(r7unusedrrI __classcell__rrrrr s *  r )rMr9 collectionsrtypingrrrrrr(rr utilsr rNr rrrrs