安装pytorch3d的一把辛酸泪

啥也不说了,直接去官网看
已经成功安装了,网上其他杂七杂八,真的是半天解决不了问题

1、Install PyTorch3D (following the instructions here)
2、Try a few 3D operators

from pytorch3d.utils import ico_sphere
from pytorch3d.io import load_obj
from pytorch3d.structures import Meshes
from pytorch3d.ops import sample_points_from_meshes
from pytorch3d.loss import chamfer_distance

# Use an ico_sphere mesh and load a mesh from an .obj e.g. model.obj
sphere_mesh = ico_sphere(level=3)
verts, faces, _ = load_obj("model.obj")
test_mesh = Meshes(verts=[verts], faces=[faces.verts_idx])

# Differentiably sample 5k points from the surface of each mesh and then compute the loss.
sample_sphere = sample_points_from_meshes(sphere_mesh, 5000)
sample_test = sample_points_from_meshes(test_mesh, 5000)
loss_chamfer, _ = chamfer_distance(sample_sphere, sample_test)

注:你需要哪个版本只要点上面的tag切换就可以了

你可能感兴趣的:(python,pytorch)