def refine(inputs, num_steps=2):
base, offsets = inputs
# sample bilinearly
for _ in range(num_steps):
base = base + bilinear_sampler(offsets, base)
return base
def refine(base, offsets, num_steps=2):
for i in range(num_steps):
base = base + bilinear_sampler(offsets, base)
return base