Join cameras as batch

Summary:
Function to join a list of cameras objects into a single batched object.

FB: In the next diff I will remove the `concatenate_cameras` function in implicitron and update the callsites.

Reviewed By: nikhilaravi

Differential Revision: D33198209

fbshipit-source-id: 0c9f5f5df498a0def9dba756c984e6a946618158
This commit is contained in:
Jeremy Reizenstein
2022-01-21 05:25:23 -08:00
committed by Facebook GitHub Bot
parent 9e2bc3a17f
commit 39bb2ce063
5 changed files with 187 additions and 9 deletions

View File

@@ -250,7 +250,4 @@ class TestPixels(TestCaseMixin, unittest.TestCase):
],
dim=1,
)
print(wanted)
print(camera_points[batch_idx])
self.assertClose(camera_points[batch_idx], wanted)