423A35C7/pytorch3d
1
0
Fork 0
mirror of https://github.com/facebookresearch/pytorch3d.git synced 2025-08-02 03:42:50 +08:00
Code Issues Packages Projects Releases Wiki Activity
pytorch3d/tests
History
Tristan Rice 7e0146ece4 shader: add SoftDepthShader and HardDepthShader for rendering depth maps (#36) 
Summary:
X-link: https://github.com/fairinternal/pytorch3d/pull/36

This adds two shaders for rendering depth maps for meshes. This is useful for structure from motion applications that learn depths based off of camera pair disparities.

There's two shaders, one hard which just returns the distances and then a second that does a cumsum on the probabilities of the points with a weighted sum. Areas that don't have any z faces are set to the zfar distance.

Output from this renderer is `[N, H, W]` since it's just depth no need for channels.

I haven't tested this in an ML model yet just in a notebook.

hard:
![hardzshader](https://user-images.githubusercontent.com/909104/170190363-ef662c97-0bd2-488c-8675-0557a3c7dd06.png)

soft:
![softzshader](https://user-images.githubusercontent.com/909104/170190365-65b08cd7-0c49-4119-803e-d33c1d8c676e.png)

Pull Request resolved: https://github.com/facebookresearch/pytorch3d/pull/1208

Reviewed By: bottler

Differential Revision: D36682194

Pulled By: d4l3k

fbshipit-source-id: 5d4e10c6fb0fff5427be4ddd3bd76305a7ccc1e2
2022-06-26 04:01:29 -07:00
..
benchmarks
move targets
2022-05-25 06:16:03 -07:00
data
Importing obj files without usemtl
2022-02-10 09:39:44 -08:00
implicitron
omegaconf 2.2.2 compatibility
2022-06-24 04:18:01 -07:00
pulsar
move targets
2022-05-25 06:16:03 -07:00
__init__.py
Update license for company name
2022-01-04 11:43:38 -08:00
common_testing.py
test runner for experiment.py
2022-05-26 05:33:03 -07:00
test_acos_linear_extrapolation.py
move targets
2022-05-25 06:16:03 -07:00
test_ball_query.py
move targets
2022-05-25 06:16:03 -07:00
test_blending.py
move targets
2022-05-25 06:16:03 -07:00
test_build.py
move targets
2022-05-25 06:16:03 -07:00
test_camera_conversions.py
move targets
2022-05-25 06:16:03 -07:00
test_camera_pixels.py
move targets
2022-05-25 06:16:03 -07:00
test_camera_utils.py
move targets
2022-05-25 06:16:03 -07:00
test_cameras_alignment.py
move targets
2022-05-25 06:16:03 -07:00
test_cameras.py
move targets
2022-05-25 06:16:03 -07:00
test_chamfer.py
chamfer for empty pointclouds #1174
2022-05-26 14:56:22 -07:00
test_common_linear_with_repeat.py
move targets
2022-05-25 06:16:03 -07:00
test_common_testing.py
move targets
2022-05-25 06:16:03 -07:00
test_common_workaround.py
move targets
2022-05-25 06:16:03 -07:00
test_compositing.py
move targets
2022-05-25 06:16:03 -07:00
test_cubify.py
move targets
2022-05-25 06:16:03 -07:00
test_face_areas_normals.py
move targets
2022-05-25 06:16:03 -07:00
test_graph_conv.py
move targets
2022-05-25 06:16:03 -07:00
test_harmonic_embedding.py
move targets
2022-05-25 06:16:03 -07:00
test_interpolate_face_attributes.py
move targets
2022-05-25 06:16:03 -07:00
test_io_gltf.py
move targets
2022-05-25 06:16:03 -07:00
test_io_obj.py
move targets
2022-05-25 06:16:03 -07:00
test_io_off.py
move targets
2022-05-25 06:16:03 -07:00
test_io_ply.py
move targets
2022-05-25 06:16:03 -07:00
test_iou_box3d.py
move targets
2022-05-25 06:16:03 -07:00
test_knn.py
move targets
2022-05-25 06:16:03 -07:00
test_laplacian_matrices.py
move targets
2022-05-25 06:16:03 -07:00
test_lighting.py
move targets
2022-05-25 06:16:03 -07:00
test_marching_cubes.py
move targets
2022-05-25 06:16:03 -07:00
test_materials.py
move targets
2022-05-25 06:16:03 -07:00
test_mesh_edge_loss.py
move targets
2022-05-25 06:16:03 -07:00
test_mesh_filtering.py
move targets
2022-05-25 06:16:03 -07:00
test_mesh_laplacian_smoothing.py
Update license for company name
2022-01-04 11:43:38 -08:00
test_mesh_normal_consistency.py
Update license for company name
2022-01-04 11:43:38 -08:00
test_mesh_rendering_utils.py
Update license for company name
2022-01-04 11:43:38 -08:00
test_meshes.py
move targets
2022-05-25 06:16:03 -07:00
test_ops_utils.py
move targets
2022-05-25 06:16:03 -07:00
test_packed_to_padded.py
move targets
2022-05-25 06:16:03 -07:00
test_perspective_n_points.py
move targets
2022-05-25 06:16:03 -07:00
test_point_mesh_distance.py
move targets
2022-05-25 06:16:03 -07:00
test_pointclouds.py
move targets
2022-05-25 06:16:03 -07:00
test_points_alignment.py
move targets
2022-05-25 06:16:03 -07:00
test_points_normals.py
move targets
2022-05-25 06:16:03 -07:00
test_points_to_volumes.py
move targets
2022-05-25 06:16:03 -07:00
test_r2n2.py
move targets
2022-05-25 06:16:03 -07:00
test_rasterize_meshes.py
move targets
2022-05-25 06:16:03 -07:00
test_rasterize_points.py
move targets
2022-05-25 06:16:03 -07:00
test_rasterize_rectangle_images.py
move targets
2022-05-25 06:16:03 -07:00
test_rasterizer.py
move targets
2022-05-25 06:16:03 -07:00
test_raymarching.py
move targets
2022-05-25 06:16:03 -07:00
test_raysampling.py
move targets
2022-05-25 06:16:03 -07:00
test_render_implicit.py
move targets
2022-05-25 06:16:03 -07:00
test_render_meshes_clipped.py
move targets
2022-05-25 06:16:03 -07:00
test_render_meshes.py
move targets
2022-05-25 06:16:03 -07:00
test_render_multigpu.py
move targets
2022-05-25 06:16:03 -07:00
test_render_points.py
move targets
2022-05-25 06:16:03 -07:00
test_render_volumes.py
move targets
2022-05-25 06:16:03 -07:00
test_rendering_utils.py
move targets
2022-05-25 06:16:03 -07:00
test_rotation_conversions.py
move targets
2022-05-25 06:16:03 -07:00
test_sample_farthest_points.py
move targets
2022-05-25 06:16:03 -07:00
test_sample_pdf.py
move targets
2022-05-25 06:16:03 -07:00
test_sample_points_from_meshes.py
move targets
2022-05-25 06:16:03 -07:00
test_se3.py
move targets
2022-05-25 06:16:03 -07:00
test_shader.py
shader: add SoftDepthShader and HardDepthShader for rendering depth maps (#36)
2022-06-26 04:01:29 -07:00
test_shapenet_core.py
move targets
2022-05-25 06:16:03 -07:00
test_so3.py
move targets
2022-05-25 06:16:03 -07:00
test_splatter_blend.py
Remove use of torch.tile to fix CI
2022-06-01 08:47:26 -07:00
test_struct_utils.py
move targets
2022-05-25 06:16:03 -07:00
test_subdivide_meshes.py
move targets
2022-05-25 06:16:03 -07:00
test_symeig3x3.py
move targets
2022-05-25 06:16:03 -07:00
test_texturing.py
move targets
2022-05-25 06:16:03 -07:00
test_transforms.py
move targets
2022-05-25 06:16:03 -07:00
test_vert_align.py
move targets
2022-05-25 06:16:03 -07:00
test_volumes.py
move targets
2022-05-25 06:16:03 -07:00
weights.pt
Initial commit
2020-01-23 11:53:46 -08:00