"fmoe/git@developer.sourcefind.cn:OpenDAS/fastmoe.git" did not exist on "df715c9ff4963dfd69d1ddd6366d5ff1cc4cf743"
Commit 731ea53c authored by Jeremy Reizenstein's avatar Jeremy Reizenstein Committed by Facebook GitHub Bot
Browse files

Llff & blender convention fix

Summary: Images were coming out in the wrong format.

Reviewed By: shapovalov

Differential Revision: D37291278

fbshipit-source-id: c10871c37dd186982e7abf2071ac66ed583df2e6
parent 2e42ef79
...@@ -42,7 +42,7 @@ class BlenderDatasetMapProvider(SingleSceneDatasetMapProviderBase): ...@@ -42,7 +42,7 @@ class BlenderDatasetMapProvider(SingleSceneDatasetMapProviderBase):
) )
H, W, focal = hwf H, W, focal = hwf
H, W = int(H), int(W) H, W = int(H), int(W)
images = torch.from_numpy(images) images = torch.from_numpy(images).permute(0, 3, 1, 2)[:, :3]
# pyre-ignore[16] # pyre-ignore[16]
self.poses = _interpret_blender_cameras(poses, H, W, focal) self.poses = _interpret_blender_cameras(poses, H, W, focal)
......
...@@ -50,7 +50,7 @@ class LlffDatasetMapProvider(SingleSceneDatasetMapProviderBase): ...@@ -50,7 +50,7 @@ class LlffDatasetMapProvider(SingleSceneDatasetMapProviderBase):
i_split = (i_train, i_test, i_test) i_split = (i_train, i_test, i_test)
H, W, focal = hwf H, W, focal = hwf
H, W = int(H), int(W) H, W = int(H), int(W)
images = torch.from_numpy(images) images = torch.from_numpy(images).permute(0, 3, 1, 2)
poses = torch.from_numpy(poses) poses = torch.from_numpy(poses)
# pyre-ignore[16] # pyre-ignore[16]
......
...@@ -40,6 +40,7 @@ class TestDataLlff(TestCaseMixin, unittest.TestCase): ...@@ -40,6 +40,7 @@ class TestDataLlff(TestCaseMixin, unittest.TestCase):
self.assertEqual(len(dataset), length) self.assertEqual(len(dataset), length)
# try getting a value # try getting a value
value = dataset[0] value = dataset[0]
self.assertEqual(value.image_rgb.shape, (3, 800, 800))
self.assertIsInstance(value, FrameData) self.assertIsInstance(value, FrameData)
def test_llff(self): def test_llff(self):
...@@ -62,6 +63,7 @@ class TestDataLlff(TestCaseMixin, unittest.TestCase): ...@@ -62,6 +63,7 @@ class TestDataLlff(TestCaseMixin, unittest.TestCase):
value = dataset[0] value = dataset[0]
self.assertIsInstance(value, FrameData) self.assertIsInstance(value, FrameData)
self.assertEqual(value.frame_type, frame_type) self.assertEqual(value.frame_type, frame_type)
self.assertEqual(value.image_rgb.shape, (3, 378, 504))
self.assertEqual(len(dataset_map.test.get_eval_batches()), 3) self.assertEqual(len(dataset_map.test.get_eval_batches()), 3)
for batch in dataset_map.test.get_eval_batches(): for batch in dataset_map.test.get_eval_batches():
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment