NVIDIA-AI-IOT / torch2trt
An easy to use PyTorch to TensorRT converter
☆4,707Updated 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,052Updated 3 weeks ago
- PyTorch/TorchScript/FX compiler for NVIDIA GPUs using TensorRT☆2,716Updated this week
- Simplify your onnx model☆4,031Updated 7 months ago
- Implementation of popular deep learning networks with TensorRT network definition API☆7,276Updated 3 months ago
- The pytorch re-implement of the official efficientdet with SOTA performance in real time and pretrained weights.☆5,234Updated 3 years ago
- TensorRT MODNet, YOLOv4, YOLOv3, SSD, MTCNN, and GoogLeNet☆1,772Updated 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,416Updated 3 weeks ago
- Object detection, 3D detection, and pose estimation using center point detection:☆7,379Updated 2 years ago
- OpenMMLab Model Deployment Framework☆2,901Updated 6 months ago
- Scaled-YOLOv4: Scaling Cross Stage Partial Network☆2,031Updated 5 months ago
- PyTorch ,ONNX and TensorRT implementation of YOLOv4☆4,507Updated 9 months ago
- A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch☆8,594Updated last week
- A PyTorch implementation of EfficientNet☆8,058Updated 2 years ago
- A GPU-accelerated library containing highly optimized building blocks and an execution engine for data processing to accelerate deep lear…☆5,337Updated last week
- RepVGG: Making VGG-style ConvNets Great Again☆3,380Updated 2 years ago
- OpenMMLab Computer Vision Foundation☆6,082Updated this week
- Codebase for Image Classification Research, written in PyTorch.☆2,152Updated last year
- A simple, fully convolutional model for real-time instance segmentation.☆5,098Updated last year
- Count the MACs / FLOPs of your PyTorch model.☆4,973Updated 8 months ago
- Flops counter for neural networks in pytorch framework☆2,872Updated 2 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,196Updated 2 years ago
- AdelaiDet is an open source toolbox for multiple instance-level detection and recognition tasks.☆3,430Updated 7 months ago
- 🔥🔥🔥🔥 (Earlier YOLOv7 not official one) YOLO with Transformers and Instance Segmentation, with TensorRT acceleration! 🔥🔥🔥☆3,123Updated last year
- Most popular metrics used to evaluate object detection algorithms.☆5,031Updated 3 months ago
- FCOS: Fully Convolutional One-Stage Object Detection (ICCV'19)☆3,301Updated last year
- ResNeSt: Split-Attention Networks☆3,249Updated 2 years ago
- PyTorch implementation of YOLOv4☆1,904Updated 5 months ago
- YOLOv3 in PyTorch > ONNX > CoreML > TFLite☆10,362Updated this week
- CV-CUDA™ is an open-source, GPU accelerated library for cloud-scale image processing and computer vision.☆2,477Updated last week
- End-to-End Object Detection with Transformers☆14,179Updated last year