NVIDIA-AI-IOT / torch2trt
An easy to use PyTorch to TensorRT converter
☆4,652Updated 5 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☆2,996Updated last month
- PyTorch/TorchScript/FX compiler for NVIDIA GPUs using TensorRT☆2,647Updated this week
- Implementation of popular deep learning networks with TensorRT network definition API☆7,138Updated last month
- Simplify your onnx model☆3,942Updated 4 months ago
- Scaled-YOLOv4: Scaling Cross Stage Partial Network☆2,020Updated 2 months ago
- TensorRT MODNet, YOLOv4, YOLOv3, SSD, MTCNN, and GoogLeNet☆1,761Updated 5 months ago
- NVIDIA® TensorRT™ is an SDK for high-performance deep learning inference on NVIDIA GPUs. This repository contains the open source compone…☆11,056Updated last month
- PyTorch ,ONNX and TensorRT implementation of YOLOv4☆4,496Updated 6 months ago
- Object detection, 3D detection, and pose estimation using center point detection:☆7,334Updated last year
- The pytorch re-implement of the official efficientdet with SOTA performance in real time and pretrained weights.☆5,218Updated 3 years ago
- OpenMMLab Computer Vision Foundation☆5,970Updated 3 weeks ago
- Most popular metrics used to evaluate object detection algorithms.☆4,994Updated 2 weeks ago
- A simple, fully convolutional model for real-time instance segmentation.☆5,063Updated last year
- Neural Network Distiller by Intel AI Lab: a Python package for neural network compression research. https://intellabs.github.io/distille…☆4,364Updated last year
- A PyTorch implementation of EfficientNet☆8,001Updated 2 years 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,191Updated last year
- A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch☆8,502Updated last month
- implementation of paper - You Only Learn One Representation: Unified Network for Multiple Tasks (https://arxiv.org/abs/2105.04206)☆1,994Updated 2 months ago
- AdelaiDet is an open source toolbox for multiple instance-level detection and recognition tasks.☆3,412Updated 4 months ago
- C++ library based on tensorrt integration☆2,664Updated last year
- OpenMMLab Model Deployment Framework☆2,822Updated 3 months ago
- Convert TensorFlow, Keras, Tensorflow.js and Tflite models to ONNX☆2,356Updated this week
- RepVGG: Making VGG-style ConvNets Great Again☆3,360Updated last year
- PyTorch implementation of YOLOv4☆1,896Updated 2 months ago
- FCOS: Fully Convolutional One-Stage Object Detection (ICCV'19)☆3,288Updated last year
- Fast and accurate object detection with end-to-end GPU optimization☆891Updated 3 years ago
- Count the MACs / FLOPs of your PyTorch model.☆4,940Updated 6 months ago
- Flops counter for convolutional networks in pytorch framework☆2,848Updated 3 months ago
- Codebase for Image Classification Research, written in PyTorch.☆2,143Updated 9 months ago
- Simple Online Realtime Tracking with a Deep Association Metric☆5,449Updated 6 months ago