AICVHub / Pytorch2TensorRT
Pytorch-->onnx-->TensorRT; CUDA11, CUDNN8, TensorRT8
☆194Updated last year
Related projects ⓘ
Alternatives and complementary repositories for Pytorch2TensorRT
- tensorrt int8 量化yolov5 onnx模型☆176Updated 3 years ago
- A tutorial about how to build a TensorRT Engine from a PyTorch Model with the help of ONNX☆243Updated 3 years ago
- NVIDIA TensorRT 加速推断教程！☆132Updated 3 years ago
- www.giantpandacv.com☆149Updated 5 months ago
- NVIDIA-阿里2021 TRT比赛 `二等奖` 代码提交 团队：美迪康 AI Lab☆164Updated 2 years 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
- TensorRT-7 Network Lib 包括常用目标检测、关键点检测、人脸检测、OCR等 可训练自己数据☆523Updated 3 years ago
- This project aims to explore the deployment of Swin-Transformer based on TensorRT, including the test results of FP16 and INT8.☆161Updated last year
- Implement popular deep learning networks in pytorch, used by tensorrtx.☆191Updated 2 years ago
- tensorrt wheels (API)☆77Updated 3 years ago
- ONNX2Pytorch☆159Updated 3 years ago
- TensorRT int8 量化部署 yolov5s 模型，实测3.3ms一帧！☆166Updated 3 years ago
- RepVGG TensorRT int8 量化，实测推理不到1ms一帧！☆61Updated 3 years ago
- pytorch -> onnx -> caffe, pytorch to caffe, or other deep learning framework to onnx and onnx to caffe.☆158Updated 3 years ago
- TensorRT ONNX Plugin、Inference、Compile☆465Updated 3 years ago
- yolov5 onnx caffe☆113Updated 3 years ago
- Code for some onnxruntime projects☆120Updated 3 years ago
- A pytorch to tensorrt convert with dynamic shape support☆257Updated 9 months ago
- convert mmdetection model to tensorrt, support fp16, int8, batch input, dynamic shape etc.☆589Updated 9 months ago
- An Object Detection Knowledge Distillation framework powered by pytorch, now having SSD and yolov5.☆219Updated 2 years ago
- Some tools to operate PaddlePaddle model☆72Updated 2 years ago
- Pytorch and ncnn implementation of PPYOLOE、YOLOX、PPYOLO、PPYOLOv2、PicoDet and so on.☆288Updated 8 months ago
- An Improved One millisecond Mobile Backbone☆142Updated 2 years ago
- ☆62Updated 4 years ago
- 针对pytorch模型的自动化模型结构分析和修改工具集，包含自动分析模型结构的模型压缩算法库☆244Updated last year
- ☆127Updated 8 months ago
- Model Compression 1. Pruning(BN Pruning) 2. Knowledge Distillation (Hinton) 3. Quantization (MNN) 4. Deployment (MNN)☆77Updated 3 years ago
- A Robust Non-IoU Alternative to Non-Maxima Suppression in Object Detection☆50Updated 3 years ago
- RM Operation can equivalently convert ResNet to VGG, which is better for pruning; and can help RepVGG perform better when the depth is la…☆211Updated last year