transforms.py

Go to the documentation of this file.

import copy
import torch
 
#numpy 타입, list 타입, 배치단위 처리 추가 필요
def xyxy_to_xywh(xyxy): 
    """
    xyxy 구조박스의 xywh 구조를 변경하는 기능
    xyxy: left, top, right, bottom
    xywh: left, top, width, height
    args:
        xyxy(list or np.array): xyxy 구조의 박스
    return (list or np.array): xywh 구조의 박스
    """
    xywh = copy.deepcopy(xyxy)
    xywh[2] = xyxy[2]-xyxy[0]
    xywh[3] = xyxy[3]-xyxy[1]
    return xywh
 
def xywh_to_xyxy(xywh):
    """
    xywh 구조박스의 xyxy 구조를 변경하는 기능
    xyxy: left, top, right, bottom
    xywh: left, top, width, height
    args:
        xywh(list or np.array): xywh 구조의 박스
    return (list or np.array): xyxy 구조의 박스
    """
    xyxy = copy.deepcopy(xywh)
    xyxy[2] = xywh[2]+xywh[0]
    xyxy[3] = xywh[3]+xywh[1]
    return xyxy
 
def xyxy_to_cwh(xyxy):
    """
    xywh 구조박스의 cwh 구조를 변경하는 기능
    xyxy: left, top, right, bottom
    cwh: 박스 중심점(cx, cy), width, height, 
    args:
        xyxy(list or np.array): xyxy 구조의 박스
    return (list or np.array): cwh 구조의 박스
    """
    cwh = copy.deepcopy(xyxy)
    cwh[2] = xyxy[2]-xyxy[0]
    cwh[3] = xyxy[3]-xyxy[1]
    cwh[0] = xyxy[0]+cwh[2]*0.5
    cwh[1] = xyxy[1]+cwh[3]*0.5
    return cwh
 
def cwh_to_xyxy(cwh):
    """
    cwh 구조박스의 xywh 구조를 변경하는 기능
    xyxy: left, top, right, bottom
    cwh: 박스 중심점(cx, cy), width, height, 
    args:
        cwh(list or np.array): cwh 구조의 박스
    return (list or np.array): xyxy 구조의 박스
    """
    xyxy = copy.deepcopy(cwh)
    xyxy[0] = cwh[0]-cwh[2]*0.5
    xyxy[1] = cwh[1]-cwh[3]*0.5
    xyxy[2] = cwh[2]+xyxy[0]
    xyxy[3] = cwh[3]+xyxy[1]
    return xyxy
 
def bbox_xyxy_to_cxcyah(bboxes):
    """
    xyxy 구조박스의 cxcyah 구조를 변경하는 기능
    xyxy: left, top, right, bottom
    cxcyah: 박스 중심점(cx, cy), ratio, height, 
    args:
        bboxes(np.array): xyxy 구조의 박스
    return (np.array): cxcyah 구조의 박스
    """
    cx = (bboxes[:, 2] + bboxes[:, 0]) / 2
    cy = (bboxes[:, 3] + bboxes[:, 1]) / 2
    w = bboxes[:, 2] - bboxes[:, 0]
    h = bboxes[:, 3] - bboxes[:, 1]
    xyah = torch.stack([cx, cy, w / h, h], -1)
    return xyah
 
 
def bbox_cxcyah_to_xyxy(bboxes):
    """
    cxcyah 구조박스의 xyxy 구조를 변경하는 기능
    xyxy: left, top, right, bottom
    cxcyah: 박스 중심점(cx, cy), ratio, height, 
    args:
        bboxes(np.array): cxcyah 구조의 박스
    return (np.array): xyxy 구조의 박스
    """
    cx, cy, ratio, h = bboxes.split((1, 1, 1, 1), dim=-1)
    w = ratio * h
    x1y1x2y2 = [cx - w / 2.0, cy - h / 2.0, cx + w / 2.0, cy + h / 2.0]
    return torch.cat(x1y1x2y2, dim=-1)