Grid Sensitive:
#Grid Sensitive
self.scale_x_y=1.05
th_xy = torch.sigmoid(ps[..., :2])
pxy = self.scale_x_y * th_xy - 0.5 * (self.scale_x_y -1.0)
Grid Sensitive:
#Grid Sensitive
self.scale_x_y=1.05
th_xy = torch.sigmoid(ps[..., :2])
pxy = self.scale_x_y * th_xy - 0.5 * (self.scale_x_y -1.0)