导出onnx模型
import torch
import numpy as np
import pointnet2_utilsclass CustomModel(torch.nn.Module):def __init__(self):super(CustomModel, self).__init__()def forward(self, unknown, known):return pointnet2_utils.three_nn(unknown, known)model CustomModel…