mirror of
https://github.com/facebookresearch/pytorch3d.git
synced 2025-08-03 20:32:51 +08:00
47c0997227
Summary: D33970393 (e9fb6c27e3) ran an inference to add some typing. Remove some where it was a bit too confident. (Also fix some pyre errors in plotly_vis caused by new mismatch.)
Reviewed By: patricklabatut
Differential Revision: D34004689
fbshipit-source-id: 430182b0ff0b91be542a3120da6d6b1d2b247c59
111 lines
3.7 KiB
Python
111 lines
3.7 KiB
Python
# 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.
|
|
|
|
from typing import Optional
|
|
|
|
import numpy as np
|
|
from PIL import Image, ImageDraw
|
|
from pytorch3d.renderer.mesh import TexturesUV
|
|
|
|
|
|
def texturesuv_image_matplotlib(
|
|
texture: TexturesUV,
|
|
*,
|
|
texture_index: int = 0,
|
|
radius: float = 1,
|
|
color=(1.0, 0.0, 0.0),
|
|
subsample: Optional[int] = 10000,
|
|
origin: str = "upper",
|
|
) -> None: # pragma: no cover
|
|
"""
|
|
Plot the texture image for one element of a TexturesUV with
|
|
matplotlib together with verts_uvs positions circled.
|
|
In particular a value in verts_uvs which is never referenced
|
|
in faces_uvs will still be plotted.
|
|
This is for debugging purposes, e.g. to align the map with
|
|
the uv coordinates. In particular, matplotlib
|
|
is used which is not an official dependency of PyTorch3D.
|
|
|
|
Args:
|
|
texture: a TexturesUV object with one mesh
|
|
texture_index: index in the batch to plot
|
|
radius: plotted circle radius in pixels
|
|
color: any matplotlib-understood color for the circles.
|
|
subsample: if not None, number of points to plot.
|
|
Otherwise all points are plotted.
|
|
origin: "upper" or "lower" like matplotlib.imshow .
|
|
upper (the default) matches texturesuv_image_PIL.
|
|
"""
|
|
|
|
import matplotlib.pyplot as plt
|
|
from matplotlib.patches import Circle
|
|
|
|
texture_image = texture.maps_padded()
|
|
centers = texture.centers_for_image(index=texture_index).numpy()
|
|
|
|
ax = plt.gca()
|
|
ax.imshow(texture_image[texture_index].detach().cpu().numpy(), origin=origin)
|
|
|
|
n_points = centers.shape[0]
|
|
if subsample is None or n_points <= subsample:
|
|
indices = range(n_points)
|
|
else:
|
|
indices = np.random.choice(n_points, subsample, replace=False)
|
|
for i in indices:
|
|
# setting clip_on=False makes it obvious when
|
|
# we have UV coordinates outside the correct range
|
|
ax.add_patch(Circle(centers[i], radius, color=color, clip_on=False))
|
|
|
|
|
|
def texturesuv_image_PIL(
|
|
texture: TexturesUV,
|
|
*,
|
|
texture_index: int = 0,
|
|
radius: float = 1,
|
|
color="red",
|
|
subsample: Optional[int] = 10000,
|
|
): # pragma: no cover
|
|
"""
|
|
Return a PIL image of the texture image of one element of the batch
|
|
from a TexturesUV, together with the verts_uvs positions circled.
|
|
In particular a value in verts_uvs which is never referenced
|
|
in faces_uvs will still be plotted.
|
|
This is for debugging purposes, e.g. to align the map with
|
|
the uv coordinates. In particular, matplotlib
|
|
is used which is not an official dependency of PyTorch3D.
|
|
|
|
Args:
|
|
texture: a TexturesUV object with one mesh
|
|
texture_index: index in the batch to plot
|
|
radius: plotted circle radius in pixels
|
|
color: any PIL-understood color for the circles.
|
|
subsample: if not None, number of points to plot.
|
|
Otherwise all points are plotted.
|
|
|
|
Returns:
|
|
PIL Image object.
|
|
"""
|
|
|
|
centers = texture.centers_for_image(index=texture_index).numpy()
|
|
texture_image = texture.maps_padded()
|
|
texture_array = (texture_image[texture_index] * 255).cpu().numpy().astype(np.uint8)
|
|
|
|
image = Image.fromarray(texture_array)
|
|
draw = ImageDraw.Draw(image)
|
|
|
|
n_points = centers.shape[0]
|
|
if subsample is None or n_points <= subsample:
|
|
indices = range(n_points)
|
|
else:
|
|
indices = np.random.choice(n_points, subsample, replace=False)
|
|
|
|
for i in indices:
|
|
x = centers[i][0]
|
|
y = centers[i][1]
|
|
draw.ellipse([(x - radius, y - radius), (x + radius, y + radius)], fill=color)
|
|
|
|
return image
|