NVIDIA-AI-IOT / torch2trt
An easy to use PyTorch to TensorRT converter
☆4,682Updated 6 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,028Updated last week
- PyTorch/TorchScript/FX compiler for NVIDIA GPUs using TensorRT☆2,688Updated this week
- Implementation of popular deep learning networks with TensorRT network definition API☆7,218Updated 2 months ago
- Simplify your onnx model☆3,993Updated 6 months ago
- OpenMMLab Model Deployment Framework☆2,865Updated 5 months ago
- Scaled-YOLOv4: Scaling Cross Stage Partial Network☆2,028Updated 4 months ago
- TensorRT MODNet, YOLOv4, YOLOv3, SSD, MTCNN, and GoogLeNet☆1,766Updated 7 months ago
- PyTorch ,ONNX and TensorRT implementation of YOLOv4☆4,500Updated 8 months ago
- NVIDIA® TensorRT™ is an SDK for high-performance deep learning inference on NVIDIA GPUs. This repository contains the open source compone…☆11,256Updated last month
- YOLOX is a high-performance anchor-free YOLO, exceeding yolov3~v5 with MegEngine, ONNX, TensorRT, ncnn, and OpenVINO supported. Documenta…☆9,664Updated 3 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 last year
- A GPU-accelerated library containing highly optimized building blocks and an execution engine for data processing to accelerate deep lear…☆5,299Updated this week
- implementation of paper - You Only Learn One Representation: Unified Network for Multiple Tasks (https://arxiv.org/abs/2105.04206)☆1,996Updated 4 months ago
- PyTorch implementation of YOLOv4☆1,899Updated 4 months ago
- Deploy your model with TensorRT quickly.☆763Updated last year
- Collection of common code that's shared among different research projects in FAIR computer vision team.☆2,083Updated 3 months ago
- Object detection, 3D detection, and pose estimation using center point detection:☆7,364Updated 2 years ago
- Count the MACs / FLOPs of your PyTorch model.☆4,962Updated 7 months ago
- CV-CUDA™ is an open-source, GPU accelerated library for cloud-scale image processing and computer vision.☆2,449Updated this week
- Fast and accurate object detection with end-to-end GPU optimization☆891Updated 3 years ago
- A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch☆8,566Updated this week
- C++ library based on tensorrt integration☆2,692Updated last year
- RepVGG: Making VGG-style ConvNets Great Again☆3,374Updated 2 years ago
- OpenMMLab Computer Vision Foundation☆6,025Updated this week
- The pytorch re-implement of the official efficientdet with SOTA performance in real time and pretrained weights.☆5,224Updated 3 years ago
- 🔥🔥🔥🔥 (Earlier YOLOv7 not official one) YOLO with Transformers and Instance Segmentation, with TensorRT acceleration! 🔥🔥🔥☆3,126Updated last year
- DeepStream SDK Python bindings and sample applications☆1,565Updated 4 months ago
- Convert TensorFlow, Keras, Tensorflow.js and Tflite models to ONNX☆2,376Updated 3 weeks ago
- A tool to modify ONNX models in a visualization fashion, based on Netron and Flask.☆1,436Updated this week
- Simple samples for TensorRT programming☆1,578Updated 2 months ago