U (dh @sXddlZddlmZeedddZeedddZeeddd Zeedd d ZdS) N)Tensor)boxesreturnc Cs\|d\}}}}|d|}|d|}|d|}|d|}tj||||fdd}|S)ad Converts bounding boxes from (cx, cy, w, h) format to (x1, y1, x2, y2) format. (cx, cy) refers to center of bounding box (w, h) are width and height of bounding box Args: boxes (Tensor[N, 4]): boxes in (cx, cy, w, h) format which will be converted. Returns: boxes (Tensor(N, 4)): boxes in (x1, y1, x2, y2) format. g?ZdimZunbindtorchstack) rcxcywhx1y1x2y2r@/tmp/pip-unpacked-wheel-vx7f76es/torchvision/ops/_box_convert.py_box_cxcywh_to_xyxys     rc CsT|d\}}}}||d}||d}||}||}tj||||fdd}|S)ah Converts bounding boxes from (x1, y1, x2, y2) format to (cx, cy, w, h) format. (x1, y1) refer to top left of bounding box (x2, y2) refer to bottom right of bounding box Args: boxes (Tensor[N, 4]): boxes in (x1, y1, x2, y2) format which will be converted. Returns: boxes (Tensor(N, 4)): boxes in (cx, cy, w, h) format. rrr) rrrrrr r r r rrr_box_xyxy_to_cxcywhs   rcCs4|d\}}}}tj||||||gdd}|S)aW Converts bounding boxes from (x, y, w, h) format to (x1, y1, x2, y2) format. (x, y) refers to top left of bouding box. (w, h) refers to width and height of box. Args: boxes (Tensor[N, 4]): boxes in (x, y, w, h) which will be converted. Returns: boxes (Tensor[N, 4]): boxes in (x1, y1, x2, y2) format. rrr)rxyr r rrr_box_xywh_to_xyxy2s rcCs<|d\}}}}||}||}tj||||fdd}|S)a] Converts bounding boxes from (x1, y1, x2, y2) format to (x, y, w, h) format. (x1, y1) refer to top left of bounding box (x2, y2) refer to bottom right of bounding box Args: boxes (Tensor[N, 4]): boxes in (x1, y1, x2, y2) which will be converted. Returns: boxes (Tensor[N, 4]): boxes in (x, y, w, h) format. rrr)rrrrrr r rrr_box_xyxy_to_xywhBs r)rrrrrrrrrrs