NVIDIA-AI-IOT / torch2trt
An easy to use PyTorch to TensorRT converter
☆4,736Updated 8 months ago
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,066Updated last week
- PyTorch/TorchScript/FX compiler for NVIDIA GPUs using TensorRT☆2,745Updated this week
- Simplify your onnx model☆4,074Updated 8 months ago
- Implementation of popular deep learning networks with TensorRT network definition API☆7,334Updated this week
- NVIDIA® TensorRT™ is an SDK for high-performance deep learning inference on NVIDIA GPUs. This repository contains the open source compone…☆11,575Updated this week
- TensorRT MODNet, YOLOv4, YOLOv3, SSD, MTCNN, and GoogLeNet☆1,775Updated 9 months ago
- Scaled-YOLOv4: Scaling Cross Stage Partial Network☆2,032Updated 6 months ago
- OpenMMLab Model Deployment Framework☆2,939Updated 7 months ago
- PyTorch ,ONNX and TensorRT implementation of YOLOv4☆4,508Updated 10 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,193Updated 2 years ago
- Convert TensorFlow, Keras, Tensorflow.js and Tflite models to ONNX☆2,419Updated 3 months ago
- The pytorch re-implement of the official efficientdet with SOTA performance in real time and pretrained weights.☆5,245Updated 3 years ago
- Deploy your model with TensorRT quickly.☆769Updated last year
- A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch☆8,650Updated last week
- Simple samples for TensorRT programming☆1,598Updated last month
- PyTorch implementation of YOLOv4☆1,908Updated 6 months ago
- 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,416Updated 2 years ago
- Flops counter for neural networks in pytorch framework☆2,891Updated 3 months ago
- OpenMMLab Computer Vision Foundation☆6,124Updated 3 weeks ago
- implementation of paper - You Only Learn One Representation: Unified Network for Multiple Tasks (https://arxiv.org/abs/2105.04206)☆1,999Updated 6 months ago
- Count the MACs / FLOPs of your PyTorch model.☆4,985Updated 10 months ago
- Neural Network Distiller by Intel AI Lab: a Python package for neural network compression research. https://intellabs.github.io/distille…☆4,390Updated 2 years ago
- Fast and accurate object detection with end-to-end GPU optimization☆894Updated 3 years ago
- NanoDet-Plus⚡Super fast and lightweight anchor-free object detection model. 🔥Only 980 KB(int8) / 1.8MB (fp16) and run 97FPS on cellphone…☆5,937Updated 9 months ago
- A tool to modify ONNX models in a visualization fashion, based on Netron and Flask.☆1,492Updated 2 months ago
- Implementation EfficientDet: Scalable and Efficient Object Detection in PyTorch☆1,443Updated 3 years ago
- AdelaiDet is an open source toolbox for multiple instance-level detection and recognition tasks.☆3,432Updated 8 months ago
- Deep neural network library and toolkit to do high performace inference on NVIDIA jetson platforms☆719Updated last year
- RepVGG: Making VGG-style ConvNets Great Again☆3,390Updated 2 years ago
- FCOS: Fully Convolutional One-Stage Object Detection (ICCV'19)☆3,309Updated last year