o hA@s^dZddlZddlmZddlmZmZddlZddlmZddl m Z Gdd d ej Z dS) z, Implements the Generalized R-CNN framework N) OrderedDict)OptionalUnion)nn)_log_api_usage_oncec seZdZdZdejdejdejdejddf fdd Zejj d e e ej fd e e e ej fdee e ej fe e e ej fffd d Z dde ej dee e e ej fdee e ej fe e e ej fffddZZS)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__rr r r r _has_warned)selfr r r r  __class__s/home/www/facesmatcher.com/frenv_anti/lib/python3.10/site-packages/torchvision/models/detection/generalized_rcnn.pyrs  zGeneralizedRCNN.__init__losses detectionscCs|jr|S|SN)training)rrrrrr eager_outputs,szGeneralizedRCNN.eager_outputsimagestargetsc 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)rtorchZ_assert isinstanceTensorlenshapetypeappendr enumerateanywheretolistr Ztensorsrr r Z image_sizes postprocessupdatejitZ is_scriptingrwarningswarnr)rrrtargetrZoriginal_image_sizesZimgvalZ target_idxZdegenerate_boxesZbb_idxZdegen_bbfeaturesZ proposalsZproposal_lossesrZdetector_lossesrrrrforward5sp   (        zGeneralizedRCNN.forwardr)__name__ __module__ __qualname____doc__rModulerr&r3Zunuseddictstrr(listrrrtupler9 __classcell__rrrrrs:  "  "r) r=r4 collectionsrtypingrrr&rutilsrr>rrrrrs