convert Pytorch mobileNet to tensorRt if you want to use c++ code,please follow:
- cd build
- cmake ..
- make
then you can run compiled program which can generate .onnx Then you could run mobileNetTrt.py which can use tensorRt inference.
-
Notifications
You must be signed in to change notification settings - Fork 0
Zhongan-Wang/mobileNet2tensorRT
Folders and files
| Name | Name | Last commit message | Last commit date | |
|---|---|---|---|---|
Repository files navigation
About
convert Pytorch mobileNet to tensorRt
Resources
Stars
Watchers
Forks
Releases
No releases published
Packages 0
No packages published