RizhaoCai / PyTorch_ONNX_TensorRT
A tutorial about how to build a TensorRT Engine from a PyTorch Model with the help of ONNX
☆243Updated 4 years ago
Alternatives and similar repositories for PyTorch_ONNX_TensorRT:
Users that are interested in PyTorch_ONNX_TensorRT are comparing it to the libraries listed below
- Pytorch-->onnx-->TensorRT; CUDA11, CUDNN8, TensorRT8☆201Updated last year
- A pytorch to tensorrt convert with dynamic shape support☆258Updated 11 months ago
- Implement popular deep learning networks in pytorch, used by tensorrtx.☆191Updated 3 years ago
- TensorRT-7 Network Lib 包括常用目标检测、关键点检测、人脸检测、OCR等 可训练自己数据☆528Updated 3 years ago
- TensorRT ONNX Plugin、Inference、Compile☆466Updated 3 years ago
- Useful tensorrt plugin. For pytorch and mmdetection model conversion.☆161Updated 3 months ago
- convert mmdetection model to tensorrt, support fp16, int8, batch input, dynamic shape etc.☆594Updated 11 months ago
- model compression based on pytorch (1、quantization: 8/4/2bits(dorefa)、ternary/binary value(twn/bnn/xnor-net);2、 pruning: normal、regular a…☆170Updated 4 years ago
- ☆78Updated 4 years ago
- Deploy your model with TensorRT quickly.☆763Updated last year
- ⚡ Useful scripts when using TensorRT☆240Updated 4 years ago
- 服务侧深度学习部署案例☆451Updated 4 years ago
- pytorch -> onnx -> caffe, pytorch to caffe, or other deep learning framework to onnx and onnx to caffe.☆158Updated 4 years ago
- volksdep is an open-source toolbox for deploying and accelerating PyTorch, ONNX and TensorFlow models with TensorRT.☆286Updated 3 years ago
- NVIDIA-阿里2021 TRT比赛 `二等奖` 代码提交 团队:美迪康 AI Lab☆166Updated 2 years ago
- This is a C++ implementation of CenterNet using TensorRT and CUDA☆148Updated last year
- tensorrt wheels (API)☆77Updated 4 years ago
- NVIDIA TensorRT 加速推断教程!☆133Updated 3 years ago
- tensorrt5, yolov4, yolov3,yolov3-tniy,yolov3-tniy-prn☆265Updated 4 years ago
- This project aims to explore the deployment of Swin-Transformer based on TensorRT, including the test results of FP16 and INT8.☆164Updated 2 years ago
- Code for some onnxruntime projects☆122Updated 3 years ago
- tensorrt5 , centernet , centerface, deform conv, int8☆768Updated 2 years ago
- Implement yolov5 with Tensorrt C++ api, and integrate batchedNMSPlugin. A Python wrapper is also provided.☆50Updated 3 years ago
- A nnie quantization aware training tool on pytorch.☆238Updated 4 years ago
- tensorrt int8 量化yolov5 onnx模型☆180Updated 3 years ago
- Neural Network Blocks - Collect all kinds of fancy model blocks for you to build more powerful neural network model.☆238Updated 2 years ago
- A tensorrt implementation of yolov5: https://github.com/ultralytics/yolov5☆190Updated 4 years ago
- ONNX2Pytorch☆160Updated 3 years ago
- Everything in Torch Fx☆341Updated 7 months ago
- A module to convert pytorch model to caffe model.☆66Updated 4 years ago