AICVHub / Pytorch2TensorRTLinks
Pytorch-->onnx-->TensorRT; CUDA11, CUDNN8, TensorRT8
☆210Updated last year
Alternatives and similar repositories for Pytorch2TensorRT
Users that are interested in Pytorch2TensorRT are comparing it to the libraries listed below
Sorting:
- tensorrt int8 量化yolov5 onnx模型☆184Updated 4 years ago
- A tutorial about how to build a TensorRT Engine from a PyTorch Model with the help of ONNX☆247Updated 4 years ago
- NVIDIA TensorRT 加速推断教程!☆133Updated 3 years ago
- Implement popular deep learning networks in pytorch, used by tensorrtx.☆197Updated 3 years ago
- TensorRT-7 Network Lib 包括常用目标检测、关键点检测、人脸检测、OCR等 可训练自己数据☆528Updated 3 years ago
- www.giantpandacv.com☆151Updated last year
- RepVGG TensorRT int8 量化,实测推理不到1ms一帧!☆63Updated 4 years ago
- TensorRT int8 量化部署 yolov5s 模型,实测3.3ms一帧!☆170Updated 4 years ago
- tensorrt wheels (API)☆76Updated 4 years ago
- Code for some onnxruntime projects☆122Updated 4 years ago
- pytorch -> onnx -> caffe, pytorch to caffe, or other deep learning framework to onnx and onnx to caffe.☆162Updated 4 years ago
- NVIDIA-阿里2021 TRT比赛 `二等奖` 代码提交 团队:美迪康 AI Lab☆171Updated 2 years ago
- A pytorch to tensorrt convert with dynamic shape support☆262Updated last year
- TensorRT ONNX Plugin、Inference、Compile☆469Updated 3 years ago
- Pytorch and ncnn implementation of PPYOLOE、YOLOX、PPYOLO、PPYOLOv2、PicoDet and so on.☆299Updated last week
- yolov5 onnx caffe☆117Updated 4 years ago
- Neural Network Blocks - Collect all kinds of fancy model blocks for you to build more powerful neural network model.☆240Updated 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
- convert mmdetection model to tensorrt, support fp16, int8, batch input, dynamic shape etc.☆598Updated last year
- 针对pytorch模型的自动化模型结构分析和修改工具集,包含自动分析模型结构的模型压缩算法库☆249Updated 2 years ago
- An Improved One millisecond Mobile Backbone☆145Updated 2 years ago
- ⚡ Useful scripts when using TensorRT☆242Updated 4 years ago
- Tips on how to find suitable object detection anchors☆146Updated 5 years ago
- ONNX2Pytorch☆162Updated 4 years ago
- ☆117Updated last year
- Convert YOLOv3 and YOLOv3-tiny (PyTorch version) into TensorRT models.☆61Updated 5 years ago
- reproduce the YOLO series of papers in pytorch, including YOLOv4, PP-YOLO, YOLOv5,YOLOv3, etc.☆392Updated 4 years ago
- A module to convert pytorch model to caffe model.☆67Updated 5 years ago
- Support Yolov5(4.0)/Yolov5(5.0)/YoloR/YoloX/Yolov4/Yolov3/CenterNet/CenterFace/RetinaFace/Classify/Unet. use darknet/libtorch/pytorch/mxn…☆212Updated 3 years ago
- This project aims to explore the deployment of Swin-Transformer based on TensorRT, including the test results of FP16 and INT8.☆167Updated 2 years ago