NVIDIA-AI-IOT / torch2trt
An easy to use PyTorch to TensorRT converter
☆4,691Updated 7 months ago
Alternatives and similar repositories for torch2trt:
Users that are interested in torch2trt are comparing it to the libraries listed below
- ONNX-TensorRT: TensorRT backend for ONNX☆3,044Updated 2 weeks ago
- PyTorch/TorchScript/FX compiler for NVIDIA GPUs using TensorRT☆2,711Updated this week
- Simplify your onnx model☆4,002Updated 6 months ago
- Implementation of popular deep learning networks with TensorRT network definition API☆7,252Updated 3 months ago
- OpenMMLab Computer Vision Foundation☆6,056Updated this week
- TensorRT MODNet, YOLOv4, YOLOv3, SSD, MTCNN, and GoogLeNet☆1,770Updated 7 months ago
- NVIDIA® TensorRT™ is an SDK for high-performance deep learning inference on NVIDIA GPUs. This repository contains the open source compone…☆11,350Updated last week
- OpenMMLab Model Deployment Framework☆2,889Updated 5 months ago
- A PyTorch implementation of EfficientNet☆8,046Updated 2 years ago
- The pytorch re-implement of the official efficientdet with SOTA performance in real time and pretrained weights.☆5,230Updated 3 years ago
- Google Brain AutoML☆6,323Updated 2 weeks ago
- Count the MACs / FLOPs of your PyTorch model.☆4,964Updated 8 months ago
- AdelaiDet is an open source toolbox for multiple instance-level detection and recognition tasks.☆3,429Updated 7 months ago
- RepVGG: Making VGG-style ConvNets Great Again☆3,376Updated 2 years ago
- FCOS: Fully Convolutional One-Stage Object Detection (ICCV'19)☆3,299Updated last year
- PyTorch ,ONNX and TensorRT implementation of YOLOv4☆4,503Updated 9 months ago
- Scaled-YOLOv4: Scaling Cross Stage Partial Network☆2,030Updated 4 months ago
- 🔥🔥🔥🔥 (Earlier YOLOv7 not official one) YOLO with Transformers and Instance Segmentation, with TensorRT acceleration! 🔥🔥🔥☆3,124Updated last year
- Deploy your model with TensorRT quickly.☆765Updated last year
- Collection of common code that's shared among different research projects in FAIR computer vision team.☆2,093Updated 3 months ago
- Set of methods to ensemble boxes from different object detection models, including implementation of "Weighted boxes fusion (WBF)" method…☆1,757Updated 2 years ago
- ResNeSt: Split-Attention Networks☆3,246Updated 2 years ago
- YOLOX is a high-performance anchor-free YOLO, exceeding yolov3~v5 with MegEngine, ONNX, TensorRT, ncnn, and OpenVINO supported. Documenta…☆9,710Updated 4 months ago
- PyTorch implementation of YOLOv4☆1,902Updated 4 months ago
- Flops counter for convolutional networks in pytorch framework☆2,869Updated 2 months ago
- Convert TensorFlow, Keras, Tensorflow.js and Tflite models to ONNX☆2,386Updated last month
- A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch☆8,585Updated this week
- A simple, fully convolutional model for real-time instance segmentation.☆5,091Updated last year
- Object detection, 3D detection, and pose estimation using center point detection:☆7,375Updated 2 years ago
- SOLO and SOLOv2 for instance segmentation, ECCV 2020 & NeurIPS 2020.☆1,723Updated last year