bbox鏡像數據增強

def random_flip(img):
    img = img[:, ::-1, :]
    return img


def flip_bbox(bbox, size):
    H, W = size
    bbox = bbox.copy()
    x_max = W - bbox[:, 0]
    x_min = W - bbox[:, 2]
    bbox[:, 0] = x_min
    bbox[:, 2] = x_max
    return bbox

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章