fix for camera tangential distortion

This commit is contained in:
chunyuwang 2019-08-30 10:46:58 +08:00
Родитель a5fde6b597
Коммит 9c059a02bf
2 изменённых файлов: 2 добавлений и 2 удалений

Просмотреть файл

@ -38,7 +38,7 @@ def project_point_radial(x, R, T, f, c, k, p):
r2 = np.sum(y**2, axis=0)
radial = 1 + np.einsum('ij,ij->j', np.tile(k, (1, n)),
np.array([r2, r2**2, r2**3]))
tan = p[0] * y[1] + p[1] * y[0]
tan = 2 * p[0] * y[1] + 2 * p[1] * y[0]
y = y * np.tile(radial + tan,
(2, 1)) + np.outer(np.array([p[1], p[0]]).reshape(-1), r2)
ypixel = (f * y) + c

Просмотреть файл

@ -46,7 +46,7 @@ def project_point_radial(x, R, T, f, c, k, p):
r2exp = torch.cat([r2, r2**2, r2**3], 0)
radial = 1 + torch.einsum('ij,ij->j', kexp, r2exp)
tan = p[0] * y[1] + p[1] * y[0]
tan = 2 * p[0] * y[1] + 2 * p[1] * y[0]
corr = (radial + tan).repeat((2, 1))
y = y * corr + torch.ger(torch.cat([p[1], p[0]]).view(-1), r2.view(-1))