Comments (10)
We've experienced this too, it is because of a bug in the indexing operation when exporting to onnx. Since the model uses all channels, this is virtually a no op, so it works anyway
from lidar-bonnetal.
FWIW I opened a fork that contains code to export to ONNX: https://github.com/andrewkouri/lidar-bonnetal/blob/master/train/tasks/semantic/create_onnx.py
Unfortunately I am still having problems opening the model in TensorRT because it is exported with an "ir_version" that is too new... will update if I figure out how to deal with it.
from lidar-bonnetal.
@Chen-Xieyuanli can you comment on this?
from lidar-bonnetal.
Hi @leo94-chen, here is my hack, which is not well-programmed, but you could take it as an example.
https://gist.github.com/Chen-Xieyuanli/980cc0a2c9b664b9cc279b19d61aa898#file-make_onnx-py
from lidar-bonnetal.
FWIW I opened a fork that contains code to export to ONNX: https://github.com/andrewkouri/lidar-bonnetal/blob/master/train/tasks/semantic/create_onnx.py
Unfortunately I am still having problems opening the model in TensorRT because it is exported with an "ir_version" that is too new... will update if I figure out how to deal with it.
This script works for me. Able to inference tensorRt without problem.
I have tried with cuda 10.1, tensorrt.5.1 and cuda 10.2, tensorrt 6.0
Sorry forget to mention. I modified backbone. https://github.com/PRBonn/lidar-bonnetal/blob/master/train/backbones/darknet.py
commented line 159.
#x = x[:, self.input_idxs]
If i dont comment i got something aten error
from lidar-bonnetal.
@akouri-dd @balajiravichandiran
hi,when i create onnx file using this script ,i meet a problem.
RuntimeError: ONNX export failed: Couldn't export operator aten::index
from lidar-bonnetal.
@caoyifeng001 you need to comment the line 159 in below file.
I modified backbone. https://github.com/PRBonn/lidar-bonnetal/blob/master/train/backbones/darknet.py
commented line 159.
#x = x[:, self.input_idxs]
from lidar-bonnetal.
@balajiravichandiran
thank you, I solved my problem, but I don't understand why this works.
from lidar-bonnetal.
I think it seems resolved. If you still have questions, please reopen the issue or comment on this issue.
from lidar-bonnetal.
I can successfully run this script file, but no. Onnx file is generated. Who can help me?Here are the results of my run,Thanks.
from lidar-bonnetal.
Related Issues (20)
- Expected object of scalar type Long but got scalar type Int for argument #2 'target'
- Question about speed after tensorrt of rangenet++ HOT 2
- Possibility of rosifying the process HOT 1
- No error or no visualizations when executing the viusalization.py file
- Questions about the learning setting HOT 1
- Where are the pretrained models HOT 4
- Inference, and training not working on Semantic KITTI dataset. HOT 2
- index the unused points HOT 1
- KNN HOT 1
- Module not found error when running the infer function to get the prediction.
- RangeNet++
- tensorflow / cuda / python versions for 22.04/20.04 HOT 2
- Where is the rangenet? HOT 1
- about labels/semantic-kitti yaml
- About train my dataset result HOT 2
- IndexError: index -2147483648 is out of bounds for axis 0 with size 64 HOT 6
- About learning_map in semantic-kitti.yaml file HOT 3
- Where do the values in arch_cfg.yaml file come from, especially fov_up, fov_down, img_means, img_stds? HOT 4
- IndexError HOT 2
- Stop-gradients in skip-connections HOT 2
Recommend Projects
-
React
A declarative, efficient, and flexible JavaScript library for building user interfaces.
-
Vue.js
🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.
-
Typescript
TypeScript is a superset of JavaScript that compiles to clean JavaScript output.
-
TensorFlow
An Open Source Machine Learning Framework for Everyone
-
Django
The Web framework for perfectionists with deadlines.
-
Laravel
A PHP framework for web artisans
-
D3
Bring data to life with SVG, Canvas and HTML. 📊📈🎉
-
Recommend Topics
-
javascript
JavaScript (JS) is a lightweight interpreted programming language with first-class functions.
-
web
Some thing interesting about web. New door for the world.
-
server
A server is a program made to process requests and deliver data to clients.
-
Machine learning
Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.
-
Visualization
Some thing interesting about visualization, use data art
-
Game
Some thing interesting about game, make everyone happy.
Recommend Org
-
Facebook
We are working to build community through open source technology. NB: members must have two-factor auth.
-
Microsoft
Open source projects and samples from Microsoft.
-
Google
Google ❤️ Open Source for everyone.
-
Alibaba
Alibaba Open Source for everyone
-
D3
Data-Driven Documents codes.
-
Tencent
China tencent open source team.
from lidar-bonnetal.