a 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 seZdZdZejejejejddfdd Zejj e e ej fe e e ej fee e ej fe e e ej ffdddZd e ej ee e e ej fee e ej fe e e ej ffd d d ZZS) 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 N)backbonerpn roi_heads transformreturncs4tt|||_||_||_||_d|_dS)NF)super__init__rr r r r _has_warned)selfr r r r  __class__[/var/www/auris/lib/python3.9/site-packages/torchvision/models/detection/generalized_rcnn.pyrs zGeneralizedRCNN.__init__)losses detectionsr cCs|jr |S|S)N)training)rrrrrr eager_outputs,szGeneralizedRCNN.eager_outputs)imagestargetsr c Cs2|jr|durtddnf|D]`}|d}t|tjrhtt|jdkoV|jddkd|jd q tdd t|d q g}|D]L}|jd d}tt|dkd |jd d||d |dfq| ||\}}|dur|t |D]\}}|d}|ddddf|ddddfk} | rt | j ddd d } ||  } tdd| d|d q||j} t| tjrtd| fg} ||| |\} }|| | |j|\}}|j ||j|}i}||||tjr"|jstdd|_||fS|||SdS)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)ZdimzLAll 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_sizesimgvalZ target_idxZdegenerate_boxesZbb_idxZdegen_bbfeaturesZ proposalsZproposal_lossesrZdetector_lossesrrrrforward5sn    (      zGeneralizedRCNN.forward)N)__name__ __module__ __qualname____doc__rModulerr$r1Zunuseddictstrr&listrrrtupler8 __classcell__rrrrrs"" "r) r<r2 collectionsrtypingrrr$rutilsrr=rrrrrs