Shawn87377/Instruct-X-Decoder
0
1U
2 E$�c�3 � @ sL d Z ddlZddlmZ dd� Zdd� Zdd � Zd4d � Zd d
� Zdd� Z dS )z35Utilities for bounding box manipulation and GIoU.6� N)�box_areac C sL | � d�\}}}}|d| |d| |d| |d| g}tj|dd�S )N�����g �?��dim��unbind�torch�stack)�x�x_c�y_c�w�h�b� r �//data/arXiv/demo/Demo/xdecoder/utils/box_ops.py�box_cxcywh_to_xyxy s 7 8�r c C sD | � d�\}}}}|| d || d || || g}tj|dd�S )Nr � r r �r9 �x0�y0�x1�y1r r r r �box_xyxy_to_cxcywh s �r c C s4 | � d�\}}}}|||| || g}tj|dd�S )Nr r r r r r r �box_xywh_to_xyxy s r c 10 C s� t | �}t |�}t�| d d �d d d�f |d d �d d�f �}t�| d d �d dd �f |d d �dd �f �}|| jdd�}|d d �d d �df |d d �d d �df }|d d �d f | | }|| } | |fS )Nr r ��min� )r r �maxr �clamp)11�boxes1�boxes2�area1�area2�lt�rb�wh�inter�union�iour r r �box_iou s ..,r* c C s | dd�dd�f | dd�dd�f k� � s0t�|dd�dd�f |dd�dd�f k� � s`t�t| |�\}}t�| dd�ddd�f |dd�dd�f �}t�| dd�ddd�f |dd�dd�f �}|| jdd�}|dd�dd�df |dd�dd�df }||| | S )z�12 Generalized IoU from https://giou.stanford.edu/13 14 The boxes should be in [x0, y0, x1, y1] format15 16 Returns a [N, M] pairwise matrix, where N = len(boxes1)17 and M = len(boxes2)18 Nr r r r )�all�AssertionErrorr* r r r r )r r! r) r( r$ r% r& �arear r r �generalized_box_iou- s 00..,r. c C s� | � � dkr tjd| jd�S | jdd� \}}tjd|tjd�}tjd|tjd�}t�||�\}}| |�d� }|� d��19d�d }|� | � � d �� d��
d�d }| |�d� }|� d��20d�d } |� | � � d �� d��
d�d }21t�||22|| gd�S )23z�Compute the bounding boxes around the provided masks24 25 The masks should be in format [N, H, W] where N is the number of masks, (H, W) are the spatial dimensions.26 27 Returns a [N, 4] tensors, with the boxes in xyxy format28 r )r � )�device�����N)�dtyper r g �חA)�numelr �zerosr0 �shape�arange�float�meshgrid� unsqueeze�flattenr � masked_fill�boolr r ) �masksr r
�yr29 Zx_mask�x_max�x_minZy_mask�y_max�y_minr r r �masks_to_boxesE s ""rC )30�__doc__r Ztorchvision.ops.boxesr r r r r* r. rC r r r r �<module> s