943753c968fe7c0cbafc6e44d60f1b28b42e997b,torchsample/utils.py,,th_nearest_interp2d,#,114

Before Change

    
    // take clamp of coords so they"re in the image bounds
    coords[:,0] = torch.clamp(coords[:,0], 0, input.size(1)-1).round()
    coords[:,1] = torch.clamp(coords[:,1], 0, input.size(2)-1).round()

    stride = torch.LongTensor(input.stride())[1:].float()
    idx = coords.mv(stride).long()

After Change

    x_ix = x.mul(stride[1]).long()
    y_ix = y.mul(stride[2]).long()

    input_flat = input.view(input.size(0),-1).contiguous()

    mapped_vals = input_flat.gather(1, x_ix.add(y_ix))

    return mapped_vals.view_as(input)

Instances