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

New raysamplers

Browse Source 
Summary: New MultinomialRaysampler succeeds GridRaysampler bringing masking and subsampling. Correspondingly, NDCMultinomialRaysampler succeeds NDCGridRaysampler.

Reviewed By: nikhilaravi, shapovalov

Differential Revision: D33256897

fbshipit-source-id: cd80ec6f35b110d1d20a75c62f4e889ba8fa5d45
This commit is contained in:
Jeremy Reizenstein
2022-01-24 10:51:03 -08:00
committed by Facebook GitHub Bot
parent 174738c33e
commit 3eb4233844
7 changed files with 412 additions and 61 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
tests/benchmarks/bm_raysampling.py
View File

@@ -10,9 +10,9 @@ from fvcore.common.benchmark import benchmark
from pytorch3d.renderer import (
FoVOrthographicCameras,
FoVPerspectiveCameras,
GridRaysampler,
MonteCarloRaysampler,
NDCGridRaysampler,
MultinomialRaysampler,
NDCMultinomialRaysampler,
OrthographicCameras,
PerspectiveCameras,
)
@@ -21,7 +21,11 @@ from test_raysampling import TestRaysampling
def bm_raysampling() -> None:
case_grid = {
"raysampler_type": [GridRaysampler, NDCGridRaysampler, MonteCarloRaysampler],
"raysampler_type": [
MultinomialRaysampler,
NDCMultinomialRaysampler,
MonteCarloRaysampler,
],
"camera_type": [
PerspectiveCameras,
OrthographicCameras,