423A35C7/pytorch3d
1
0
Fork 0
mirror of https://github.com/facebookresearch/pytorch3d.git synced 2025-12-21 14:50:36 +08:00
Code Issues Packages Projects Releases Wiki Activity

More type annotations

Browse Source 
Summary: More type annotations: device, shaders, pluggable I/O, stats in NeRF project, cameras, textures, etc...

Reviewed By: nikhilaravi

Differential Revision: D29327396

fbshipit-source-id: cdf0ceaaa010e22423088752688c8dd81f1acc3c
This commit is contained in:
Patrick Labatut
2021-06-25 19:55:26 -07:00
committed by Facebook GitHub Bot
parent 542e2e7c07
commit f593bfd3c2
15 changed files with 196 additions and 148 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
pytorch3d/transforms/rotation_conversions.py
View File

@@ -10,6 +10,8 @@ from typing import Optional
import torch
import torch.nn.functional as F
from ..common.types import Device
"""
The transformation matrices returned from the functions in this file assume
@@ -286,7 +288,9 @@ def matrix_to_euler_angles(matrix, convention: str):
return torch.stack(o, -1)
def random_quaternions(n: int, dtype: Optional[torch.dtype] = None, device=None):
def random_quaternions(
n: int, dtype: Optional[torch.dtype] = None, device: Optional[Device] = None
):
"""
Generate random quaternions representing rotations,
i.e. versors with nonnegative real part.
@@ -306,7 +310,9 @@ def random_quaternions(n: int, dtype: Optional[torch.dtype] = None, device=None)
return o
def random_rotations(n: int, dtype: Optional[torch.dtype] = None, device=None):
def random_rotations(
n: int, dtype: Optional[torch.dtype] = None, device: Optional[Device] = None
):
"""
Generate random rotations as 3x3 rotation matrices.
@@ -323,7 +329,9 @@ def random_rotations(n: int, dtype: Optional[torch.dtype] = None, device=None):
return quaternion_to_matrix(quaternions)
def random_rotation(dtype: Optional[torch.dtype] = None, device=None):
def random_rotation(
dtype: Optional[torch.dtype] = None, device: Optional[Device] = None
):
"""
Generate a single random 3x3 rotation matrix.