423A35C7/pytorch3d
1
0
Fork 0
mirror of https://github.com/facebookresearch/pytorch3d.git synced 2025-12-19 05:40:34 +08:00
Code Issues Packages Projects Releases Wiki Activity

Llff & blender convention fix

Browse Source 
Summary: Images were coming out in the wrong format.

Reviewed By: shapovalov

Differential Revision: D37291278

fbshipit-source-id: c10871c37dd186982e7abf2071ac66ed583df2e6
This commit is contained in:
Jeremy Reizenstein
2022-06-22 05:54:54 -07:00
committed by Facebook GitHub Bot
parent 2e42ef793f
commit 731ea53c80
3 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tests/implicitron/test_data_llff.py
View File

@@ -40,6 +40,7 @@ class TestDataLlff(TestCaseMixin, unittest.TestCase):
self.assertEqual(len(dataset), length)
# try getting a value
value = dataset[0]
self.assertEqual(value.image_rgb.shape, (3, 800, 800))
self.assertIsInstance(value, FrameData)
def test_llff(self):
@@ -62,6 +63,7 @@ class TestDataLlff(TestCaseMixin, unittest.TestCase):
value = dataset[0]
self.assertIsInstance(value, FrameData)
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)
for batch in dataset_map.test.get_eval_batches():