Source code for pytorch3d.renderer.materials

# Copyright (c) Meta Platforms, Inc. and affiliates.
# All rights reserved.
# This source code is licensed under the BSD-style license found in the
# LICENSE file in the root directory of this source tree.

# pyre-unsafe

import torch

from ..common.datatypes import Device
from .utils import TensorProperties

[docs] class Materials(TensorProperties): """ A class for storing a batch of material properties. Currently only one material per batch element is supported. """
[docs] def __init__( self, ambient_color=((1, 1, 1),), diffuse_color=((1, 1, 1),), specular_color=((1, 1, 1),), shininess=64, device: Device = "cpu", ) -> None: """ Args: ambient_color: ambient reflectivity of the material diffuse_color: diffuse reflectivity of the material specular_color: specular reflectivity of the material shininess: The specular exponent for the material. This defines the focus of the specular highlight with a high value resulting in a concentrated highlight. Shininess values can range from 0-1000. device: Device (as str or torch.device) on which the tensors should be located ambient_color, diffuse_color and specular_color can be of shape (1, C) or (N, C) where C is typically 3 (for RGB). shininess can be of shape (1,) or (N,). The colors and shininess are broadcast against each other so need to have either the same batch dimension or batch dimension = 1. """ super().__init__( device=device, diffuse_color=diffuse_color, ambient_color=ambient_color, specular_color=specular_color, shininess=shininess, ) C = self.ambient_color.shape[-1] for n in ["ambient_color", "diffuse_color", "specular_color"]: t = getattr(self, n) if t.shape[-1] != C: msg = "Expected %s to have shape (N, %d); got %r" raise ValueError(msg % (n, C, t.shape)) if self.shininess.shape != torch.Size([self._N]): msg = "shininess should have shape (N); got %r" raise ValueError(msg % repr(self.shininess.shape))
[docs] def clone(self): other = Materials(device=self.device) return super().clone(other)