Raysampler as pluggable

Summary:
This converts raysamplers to ReplaceableBase so that users can hack their own raysampling impls.

Context: Andrea tried to implement TensoRF within implicitron but could not due to the need to implement his own raysampler.

Reviewed By: shapovalov

Differential Revision: D36016318

fbshipit-source-id: ef746f3365282bdfa9c15f7b371090a5aae7f8da
This commit is contained in:
David Novotny
2022-05-12 15:39:35 -07:00
committed by Facebook GitHub Bot
parent e85fa03c5a
commit e767c4b548
16 changed files with 185 additions and 96 deletions

View File

@@ -49,10 +49,8 @@ generic_model_args:
append_xyz:
- 5
latent_dim: 0
raysampler_args:
raysampler_AdaptiveRaySampler_args:
n_rays_per_image_sampled_from_mask: 1024
min_depth: 0.0
max_depth: 0.0
scene_extent: 8.0
n_pts_per_ray_training: 64
n_pts_per_ray_evaluation: 64