Function coords_grid

evaluation_DVL.py:32–35

(batch, ht, wd)

from metrics import metrics

def coords_grid(batch, ht, wd):
    coords = torch.meshgrid(torch.arange(ht), torch.arange(wd))    
    coords = torch.stack(coords[::-1], dim=0).float()     
    return coords[None].repeat(batch, 1, 1, 1)        
def image_flow_warp(image, flow, padding_mode='zeros'):
    '''
    Input: