mirror of
https://github.com/facebookresearch/pytorch3d.git
synced 2025-08-02 11:52:50 +08:00
4f78af6170
Summary: Fixing several unused imports and pyre/linter warnings. Reviewed By: bottler Differential Revision: D22592491 fbshipit-source-id: 463383b9b73a545949475044fb5c531712f8482c
52 lines
1.7 KiB
Python
52 lines
1.7 KiB
Python
# Copyright (c) Facebook, Inc. and its affiliates. All rights reserved.
|
|
|
|
|
|
import torch
|
|
import torch.nn as nn
|
|
|
|
|
|
# A renderer class should be initialized with a
|
|
# function for rasterization and a function for shading.
|
|
# The rasterizer should:
|
|
# - transform inputs from world -> screen space
|
|
# - rasterize inputs
|
|
# - return fragments
|
|
# The shader can take fragments as input along with any other properties of
|
|
# the scene and generate images.
|
|
|
|
# E.g. rasterize inputs and then shade
|
|
#
|
|
# fragments = self.rasterize(meshes)
|
|
# images = self.shader(fragments, meshes)
|
|
# return images
|
|
|
|
|
|
class MeshRenderer(nn.Module):
|
|
"""
|
|
A class for rendering a batch of heterogeneous meshes. The class should
|
|
be initialized with a rasterizer and shader class which each have a forward
|
|
function.
|
|
"""
|
|
|
|
def __init__(self, rasterizer, shader):
|
|
super().__init__()
|
|
self.rasterizer = rasterizer
|
|
self.shader = shader
|
|
|
|
def forward(self, meshes_world, **kwargs) -> torch.Tensor:
|
|
"""
|
|
Render a batch of images from a batch of meshes by rasterizing and then
|
|
shading.
|
|
|
|
NOTE: If the blur radius for rasterization is > 0.0, some pixels can
|
|
have one or more barycentric coordinates lying outside the range [0, 1].
|
|
For a pixel with out of bounds barycentric coordinates with respect to a
|
|
face f, clipping is required before interpolating the texture uv
|
|
coordinates and z buffer so that the colors and depths are limited to
|
|
the range for the corresponding face.
|
|
"""
|
|
fragments = self.rasterizer(meshes_world, **kwargs)
|
|
images = self.shader(fragments, meshes_world, **kwargs)
|
|
|
|
return images
|