^jJJdZddlmZddlmZGddej Zy)zEDetection head: bounding-box regression + classification projections.N)MLPc6eZdZdZdededdffd ZdZxZS) DetectionHeadaUProjection head for object detection outputs. Wraps the classification linear layer and bounding-box MLP used by the LWDETR decoder to produce final detection predictions. Args: hidden_dim: Feature dimension coming from the transformer decoder. num_classes: Number of object classes (excluding background). hidden_dim num_classesreturnNc~t|tj|||_t ||dd|_y)N)super__init__nnLinear class_embedr bbox_embed)selfrr __class__s h/var/www/ramen.bs-engineer-server.com/venv/lib/python3.12/site-packages/rfdetr/models/heads/detection.pyr zDetectionHead.__init__s3 99Z=j*a;cj|j|}|j|j}||fS)aProject decoder hidden states to class logits and box coordinates. Args: hs: Decoder output tensor of shape ``(B, N, hidden_dim)``. Returns: Tuple of ``(outputs_class, outputs_coord)`` where ``outputs_class`` has shape ``(B, N, num_classes)`` and ``outputs_coord`` has shape ``(B, N, 4)`` in ``[cx, cy, w, h]`` normalised to ``[0, 1]``. )rrsigmoid)rhs outputs_class outputs_coords rforwardzDetectionHead.forwards6((, +335 m++r)__name__ __module__ __qualname____doc__intr r __classcell__)rs@rrr s'<3r&s! L",BII,r