Function image_forward_warp

core/utils/utils.py:366–380

Input: image: HxWx3 numpy flow: HxWx2 torch.Tensor Output: outImg: HxWx3 numpy

(image, flow, padding_mode='zeros')

    return outImg.astype(np.uint8)

def image_forward_warp(image, flow, padding_mode='zeros'):
    '''
    Input:
        image: HxWx3 numpy
        flow: HxWx2 torch.Tensor
    Output:
        outImg: HxWx3 numpy
    '''
    forward_warp = ForwardWarp()
    image = torch.from_numpy(image).permute([2,0,1])[None].cuda()
    flow = torch.from_numpy(flow).permute([2,0,1])[None].cuda()
    out = forward_warp(image, flow)
    image = (out[0][0] / (out[1][0] + 1e-6)).cpu().permute([1,2,0]).numpy().astype(np.uint8)

    return image