The devkit of the nuScenes dataset.

State-of-the-Art Deep Learning scripts organized by models - easy to train and deploy with reproducible accuracy and performance on enterprise-grade infrastructure.

构建基于Docker容器的寒武纪开发环境

Simple samples for TensorRT programming

ppl.cv is a high-performance image processing library of openPPL supporting various platforms.

A new tensorrt integrate. 多线程推理

plate detection recognization

an edge-real-time anchor-free object detector with decent performance

🔥🔥🔥YOLOv5, YOLOv6, YOLOv7, YOLOv8, PPYOLOE, YOLOX, YOLOR, YOLOv4, YOLOv3, Transformer, Attention, TOOD and Improved-YOLOv5-YOLOv7... Support to improve backbone, neck, head, loss, IoU, NMS and othe…

deep learning for image processing including classification and object-detection etc.

CVNets: A library for training computer vision networks

EfficientFormerV2 & EfficientFormer(NeurIPs 2022)

Image-to-Image Translation in PyTorch

Compare multiple optimization methods on triton to imporve model service performance

Retinaface get 80.99% in widerface hard val using mobilenet0.25.

Blur Detection of Digital Images using Haar Wavelet

A PyTorch implementation of NeRF (Neural Radiance Fields) that reproduces the results.

fish-kong/Yolov5-Instance-Seg-Tensorrt-CPP

NN-based radar-camera post sensor fusion implemented by TensorRT

Datasets, Transforms and Models specific to Computer Vision

yolort is a runtime stack for yolov5 on specialized accelerators such as tensorrt, libtorch, onnxruntime, tvm and ncnn.

Hello AI World guide to deploying deep-learning inference networks and deep vision primitives with TensorRT and NVIDIA Jetson.

C++/Python Linux utility wrappers for NVIDIA Jetson - camera, codecs, CUDA, GStreamer, HID, OpenGL/XGL

Count the MACs / FLOPs of your PyTorch model.

This repo addresses some of the ONNX export issues in Pytorch Retinaface.

C++ 集锦

TensorRT is a C++ library for high performance inference on NVIDIA GPUs and deep learning accelerators.

DeepStream SDK Python bindings and sample applications

PointNet++: Deep Hierarchical Feature Learning on Point Sets in a Metric Space