show_images.py 文件源码

def rotate_3d_ski(im, gt):
    im = np.transpose(im, (1, 2, 0))
    gt = np.transpose(gt, (1, 2, 0))

    ang = np.random.uniform(0, 360)
    r_im = rotate(im , ang, order=3)
    r_gt = rotate(gt, ang, order=3)

    return np.transpose(r_im, (2, 0, 1)), np.transpose(r_gt, (2, 0, 1))