search

ethnhe / PVN3D

Code for "PVN3D: A Deep Point-wise 3D Keypoints Hough Voting Network for 6DoF Pose Estimation", CVPR 2020
MIT License
482 stars 105 forks source link

cuda extension & onnx #92

Open Stone-flying opened 2 years ago

Stone-flying commented 2 years ago

作者，您好，非常感谢您的工作，我现在想把这个模型转到tensorRT上，需要先把模型转成onnx，但是在进行转换的时候调用cuda extension 的最远点采样的部分会出现报错，我请教一下您有这方面的经验吗，谢谢O(∩_∩)O！

ethnhe commented 2 years ago

抱歉，我们没有尝试过把模型转换成onnx，最远点采样部分你可以找找pointnet++有没有onnx版本可以借鉴，那个操作主要是pointnet++在用。如果有什么进展也欢迎在这个issue里和大家分享。

Stone-flying commented 2 years ago

抱歉，我们没有尝试过把模型转换成onnx，最远点采样部分你可以找找pointnet++有没有onnx版本可以借鉴，那个操作主要是pointnet++在用。如果有什么进展也欢迎在这个issue里和大家分享。

谢谢你的回复和建议，我后边自己再摸索一下，最近刚开始接触onnx