GuanLianzheng/pytorch_to_TensorRT5
通过pytorch搭建卷积神经网络完成手写识别任务,并将训练好的模型以多种方式部署到TensorRT中加速。(1)通过加载权重。(2)通过加载onnx。(3)通过加载engine。
Python
Stargazers
- 250zhanghu
- ahangchenDJI
- Allenstin
- BobZhang321BJTU
- ByronnarECNU
- chenjun2haomicroBT
- chl916185
- dilligencer-zrjXi'an Jiaotong University
- fishWhy
- float123
- foreverowl
- fusimengsitonholy(北京思腾合力)
- GitHubLuGeng
- HedlenMEGVII
- Hellcat1005
- Jasonmes
- Learningm
- Leo-xxx
- liangzimei
- longnanxi
- Lvqin001NanJing
- musicbeer
- pecanjkBeijing, China
- pengyang1225
- qinchaopypy
- qingyuanchen1997
- reborm
- Serge-weihaoSJTU
- tinaq45
- toyotomihiro
- ttssky
- XrosLiang
- YLONl
- ZhangMiaoZJU
- zhouqzzw
- Zuojianhao