NVIDIA-AI-IOT / torch2trt
An easy to use PyTorch to TensorRT converter
☆4,557Updated last month
Related projects: ⓘ
- ONNX-TensorRT: TensorRT backend for ONNX☆2,910Updated last week
- PyTorch/TorchScript/FX compiler for NVIDIA GPUs using TensorRT☆2,499Updated this week
- Simplify your onnx model☆3,777Updated 2 weeks ago
- Implementation of popular deep learning networks with TensorRT network definition API☆6,890Updated 3 weeks ago
- PyTorch ,ONNX and TensorRT implementation of YOLOv4☆4,475Updated 3 months ago
- OpenMMLab Computer Vision Foundation☆5,822Updated 2 weeks ago
- Object detection, 3D detection, and pose estimation using center point detection:☆7,245Updated last year
- The pytorch re-implement of the official efficientdet with SOTA performance in real time and pretrained weights.☆5,197Updated 2 years ago
- Scaled-YOLOv4: Scaling Cross Stage Partial Network☆2,019Updated last year
- OpenMMLab Model Deployment Framework☆2,705Updated last month
- Count the MACs / FLOPs of your PyTorch model.☆4,832Updated 2 months ago
- A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch☆8,303Updated 2 weeks ago
- TensorRT MODNet, YOLOv4, YOLOv3, SSD, MTCNN, and GoogLeNet☆1,738Updated last month
- RepVGG: Making VGG-style ConvNets Great Again☆3,306Updated last year
- NVIDIA® TensorRT™ is an SDK for high-performance deep learning inference on NVIDIA GPUs. This repository contains the open source compone…☆10,552Updated last week
- A PyTorch implementation of EfficientNet☆7,854Updated 2 years ago
- Most popular metrics used to evaluate object detection algorithms.☆4,927Updated 2 weeks ago
- Flops counter for convolutional networks in pytorch framework☆2,783Updated 2 months ago
- PyTorch implementation of YOLOv4☆1,873Updated last month
- A simple, fully convolutional model for real-time instance segmentation.☆4,997Updated 10 months ago
- FCOS: Fully Convolutional One-Stage Object Detection (ICCV'19)☆3,265Updated 9 months ago
- AdelaiDet is an open source toolbox for multiple instance-level detection and recognition tasks.☆3,363Updated 3 weeks ago
- PySlowFast: video understanding codebase from FAIR for reproducing state-of-the-art video models.☆6,510Updated last month
- YOLOX is a high-performance anchor-free YOLO, exceeding yolov3~v5 with MegEngine, ONNX, TensorRT, ncnn, and OpenVINO supported. Documenta…☆9,320Updated last month
- Efficient AI Backbones including GhostNet, TNT and MLP, developed by Huawei Noah's Ark Lab.☆3,986Updated 2 months ago
- implementation of paper - You Only Learn One Representation: Unified Network for Multiple Tasks (https://arxiv.org/abs/2105.04206)☆1,987Updated 9 months ago
- Minimal PyTorch implementation of YOLOv3☆7,309Updated 11 months ago
- Google Brain AutoML☆6,208Updated 5 months ago
- ResNeSt: Split-Attention Networks☆3,223Updated last year
- Neural Network Distiller by Intel AI Lab: a Python package for neural network compression research. https://intellabs.github.io/distille…☆4,341Updated last year