vato007/fast-depth-tf
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update warp test to verify output shape

Browse Source
This commit is contained in:
Piv
2021-08-24 21:39:10 +09:30
parent c164c9720a
commit 42fcf5554a
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
unsupervised/warp_tests.py
View File

@@ -56,7 +56,7 @@ class MyTestCase(unittest.TestCase):
disp = tf.random.uniform([1, height, width]) * 255 disp = tf.random.uniform([1, height, width]) * 255
pose = tf.random.uniform([1, 6]) pose = tf.random.uniform([1, 6])
warp.projective_inverse_warp(img, disp, pose, intrinsics, coords) self.assertEqual(warp.projective_inverse_warp(img, disp, pose, intrinsics, coords).shape, img.shape)
if __name__ == '__main__': if __name__ == '__main__':