Add pixel loss functions, move warp

This commit is contained in:
Piv
2021-07-13 19:20:24 +09:30
parent 6f7da21977
commit b9457f17fe
2 changed files with 15 additions and 19 deletions

15
unsupervised/warp.py Normal file
View File

@@ -0,0 +1,15 @@
def projective_inverse_warp(target_img, source_img, depth, pose, intrinsics):
"""
Calculate the reprojected image from the source to the target, based on the given depth, pose and intrinsics
SFM Learner inverse warp step
ps ~ K.T(t->s).Dt(pt).K^-1.pt
:param target_img: Tensor (batch, height, width, 3)
:param source_img: Tensor, same shape as target_img
:param depth: Tensor, (batch, height, width, 1)
:param pose: (batch, 3, 3)
:param intrinsics: (batch, 3, 3)
:return: The source image reprojected to the target
"""
pass