NVIDIA-AI-IOT / torch2trtLinks
An easy to use PyTorch to TensorRT converter
☆4,821Updated last year
Alternatives and similar repositories for torch2trt
Users that are interested in torch2trt are comparing it to the libraries listed below
Sorting:
- ONNX-TensorRT: TensorRT backend for ONNX☆3,156Updated last month
- PyTorch/TorchScript/FX compiler for NVIDIA GPUs using TensorRT☆2,873Updated this week
- Simplify your onnx model☆4,192Updated last month
- Implementation of popular deep learning networks with TensorRT network definition API☆7,532Updated last month
- TensorRT MODNet, YOLOv4, YOLOv3, SSD, MTCNN, and GoogLeNet☆1,780Updated last month
- NVIDIA® TensorRT™ is an SDK for high-performance deep learning inference on NVIDIA GPUs. This repository contains the open source compone…☆12,228Updated 3 weeks ago
- OpenMMLab Model Deployment Framework☆3,052Updated last year
- Scaled-YOLOv4: Scaling Cross Stage Partial Network☆2,031Updated 11 months ago
- PyTorch ,ONNX and TensorRT implementation of YOLOv4☆4,523Updated last year
- Object detection, 3D detection, and pose estimation using center point detection:☆7,505Updated 2 years ago
- OpenMMLab Computer Vision Foundation☆6,276Updated 5 months ago
- TensorRT8.Support Yolov5n,s,m,l,x .darknet -> tensorrt. Yolov4 Yolov3 use raw darknet *.weights and *.cfg fils. If the wrapper is usef…☆1,200Updated 2 years ago
- RepVGG: Making VGG-style ConvNets Great Again☆3,425Updated 2 years ago
- PyTorch implementation of YOLOv4☆1,911Updated 11 months ago
- CV-CUDA™ is an open-source, GPU accelerated library for cloud-scale image processing and computer vision.☆2,588Updated 4 months ago
- Flops counter for neural networks in pytorch framework☆2,947Updated last month
- DeepStream SDK Python bindings and sample applications☆1,704Updated last month
- The pytorch re-implement of the official efficientdet with SOTA performance in real time and pretrained weights.☆5,261Updated 3 years ago
- Neural Network Distiller by Intel AI Lab: a Python package for neural network compression research. https://intellabs.github.io/distille…☆4,401Updated 2 years ago
- Fast and accurate object detection with end-to-end GPU optimization☆900Updated 4 years ago
- A simple, fully convolutional model for real-time instance segmentation.☆5,175Updated last month
- Deploy your model with TensorRT quickly.☆766Updated last year
- A tool to modify ONNX models in a visualization fashion, based on Netron and Flask.☆1,563Updated 7 months ago
- Convert TensorFlow, Keras, Tensorflow.js and Tflite models to ONNX☆2,478Updated last month
- Simple samples for TensorRT programming☆1,644Updated 4 months ago
- Count the MACs / FLOPs of your PyTorch model.☆5,048Updated last year
- NanoDet-Plus⚡Super fast and lightweight anchor-free object detection model. 🔥Only 980 KB(int8) / 1.8MB (fp16) and run 97FPS on cellphone…☆6,065Updated last year
- 🔥 🔥🔥🔥 (Earlier YOLOv7 not official one) YOLO with Transformers and Instance Segmentation, with TensorRT acceleration! 🔥🔥🔥☆3,124Updated last year
- MobileNetV1, MobileNetV2, VGG based SSD/SSD-lite implementation in Pytorch 1.0 / Pytorch 0.4. Out-of-box support for retraining on Open I…☆1,430Updated 2 years ago
- micronet, a model compression and deploy lib. compression: 1、quantization: quantization-aware-training(QAT), High-Bit(>2b)(DoReFa/Quantiz…☆2,260Updated 5 months ago