khdZddlZddlmZddlmZmZmZmZm Z ddl Z ddl m Z m Z ddl mZGdd e jZy) z, Implements the Generalized R-CNN framework N) OrderedDict)DictListOptionalTupleUnion)nnTensor)_log_api_usage_oncec eZdZdZdej dej dej dej ddf fd Zejjd 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 transformreturnNc~t|t|||_||_||_||_d|_y)NF)super__init__r rrrr _has_warned)selfrrrr __class__s i/var/www/teggl/fontify/venv/lib/python3.12/site-packages/torchvision/models/detection/generalized_rcnn.pyrzGeneralizedRCNN.__init__s: D!"  " c"|jr|S|SN)training)rlosses detectionss r eager_outputszGeneralizedRCNN.eager_outputs&s ==Mrc |jr|tjddn|D]}|d}t|tjrOtjt |j dk(xr|j ddk(d|j d qtjdd t|d g}|D]\}|j d d}tjt |dk(d |j d d|j|d |df^|j||\}}|t|D]\}}|d}|ddddf|ddddfk} | js3tj| jdd d } || j} tjdd| d|d |j|j} t| tjrt!d| fg} |j#|| |\} }|j%| | |j&|\}}|jj)||j&|}i}|j+||j+|tj,j/r,|j0st3j4dd|_||fS|j7||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.s == e%WX%pF"7OE!%6  ,1Jekk"o6JXY^YdYdXeefg  e/_`dej`k_llm-nop79 :C))BC.C MMCA \]`]f]fgigj]k\lm  ! ' 'QQ(8 9  :..9  &/&8 " Fw#(AB<5BQB<#? #'')"[[)9)=)=!)=)DEaHKF,1&M,@,@,BHMM..6Z7LZLXY[ ==0 h -"S(O#45H%)XXfh%H" ?&*nnXy&J\J\^e&f# O^^// FrYs3#55 (g:biig:r