RandLA-Net 实现 Tensorrt 部署
Code: https://github.com/QingyongHu/RandLA-Net
RandLA-Net tensorrt C++
-
tensorflow 转 onnx
python -m tf2onnx.convert --graphdef ./results/pb_convert/board.pb --output ./results/pb_convert/board.onnx --inputs xyz_0:0,xyz_1:0,xyz_2:0,xyz_3:0,xyz_4:0,neigh_idx_0:0,neigh_idx_1:0,neigh_idx_2:0,neigh_idx_3:0,neigh_idx_4:0,sub_idx_0:0,sub_idx_1:0,sub_