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

Use sample_pdf from PyTorch3D in NeRF

Browse Source 
Summary:
Use PyTorch3D's new faster sample_pdf function instead of local Python implementation.

Also clarify deps for the Python implementation.

Reviewed By: gkioxari

Differential Revision: D30512109

fbshipit-source-id: 84cfdc00313fada37a6b29837de96f6a4646434f
This commit is contained in:
Jeremy Reizenstein
2021-08-31 10:38:58 -07:00
committed by Facebook GitHub Bot
parent d2bbd0cdb7
commit 1251446383
5 changed files with 5 additions and 70 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
tests/test_sample_pdf.py
View File

@@ -12,6 +12,7 @@ from common_testing import TestCaseMixin
from pytorch3d.renderer.implicit.sample_pdf import sample_pdf, sample_pdf_python
@unittest.skipIf(torch.__version__[:4] == "1.5.", "searchsorted needs PyTorch 1.6")
class TestSamplePDF(TestCaseMixin, unittest.TestCase):
def setUp(self) -> None:
super().setUp()