官网介绍:
输入是两个点云P1,P2(也可以是同一个,比如都是P1),
举个例子:
import torch
from pytorch3d.ops import ball_query
# 示例点云数据
N = 2 # 批次大小
P1 = 5 # 源点云中的点数
P2 = 3 # 查询点云中的点数
p1 = torch.rand(N, P1, 3) # 形状为 (N, P1, 3) 的随机源点云
p2 = torch.rand
官网介绍:
输入是两个点云P1,P2(也可以是同一个,比如都是P1),
举个例子:
import torch
from pytorch3d.ops import ball_query
# 示例点云数据
N = 2 # 批次大小
P1 = 5 # 源点云中的点数
P2 = 3 # 查询点云中的点数
p1 = torch.rand(N, P1, 3) # 形状为 (N, P1, 3) 的随机源点云
p2 = torch.rand