Stars
Using pattern matcher in onnx model to match and replace subgraphs.
YOLOv5 Quantization Aware Training with TensorRT
This is 8-bit quantization sample for yolov5. Both PTQ, QAT and Partial Quantization have been implemented, and present the results based on yolov5s.
PPL Quantization Tool (PPQ) is a powerful offline neural network quantization tool.
A tutorial for getting started with the Deep Learning Accelerator (DLA) on NVIDIA Jetson
10000 fps 🚀 for 360 Surround-View CUDA Solution
Simple samples for TensorRT programming
The repository provides code for running inference with the SegmentAnything Model (SAM), links for downloading the trained model checkpoints, and example notebooks that show how to use the model.
SOS IROS 2018 GOOGLE; StereoNet ECCV2018 GOOGLE; ActiveStereoNet ECCV2018 Oral GOOGLE; HITNET CVPR2021 GOOGLE;PLUME Uber ATG
A curated list of neural network pruning resources.
This is the implementation in C++ of the paper: 'A fast and effective ellipse detector for embedded vision applications'
AdelaiDet is an open source toolbox for multiple instance-level detection and recognition tasks.
yolov5 + csl_label.(Oriented Object Detection)(Rotation Detection)(Rotated BBox)基于yolov5的旋转目标检测
Superpoint Implemented in PyTorch: https://arxiv.org/abs/1712.07629
Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)
A simple tool that can generate TensorRT plugin code quickly.
OpenPCDet Toolbox for LiDAR-based 3D Object Detection.
[ICRA'23] BEVFusion: Multi-Task Multi-Sensor Fusion with Unified Bird's-Eye View Representation
MetaPruning: Meta Learning for Automatic Neural Network Channel Pruning. In ICCV 2019.
Implementation of paper - YOLOv7: Trainable bag-of-freebies sets new state-of-the-art for real-time object detectors