Stars
Stable Diffusion web UI
Tensors and Dynamic neural networks in Python with strong GPU acceleration
YOLOv5 π in PyTorch > ONNX > CoreML > TFLite
Build and share delightful machine learning apps, all in Python. π Star to support our work!
The largest collection of PyTorch image encoders / backbones. Including train, eval, inference, export scripts, and pretrained weights -- ResNet, ResNeXT, EfficientNet, NFNet, Vision Transformer (Vβ¦
OpenMMLab Detection Toolbox and Benchmark
Image augmentation for machine learning experiments.
Fast and flexible image augmentation library. Paper about the library: https://www.mdpi.com/2078-2489/11/2/125
This is an official implementation for "Swin Transformer: Hierarchical Vision Transformer using Shifted Windows".
Object Detection toolkit based on PaddlePaddle. It supports object detection, instance segmentation, multiple object tracking and real-time multi-person keypoint detection.
Advanced AI Explainability for computer vision. Support for CNNs, Vision Transformers, Classification, Object detection, Segmentation, Image similarity and more.
YOLOX is a high-performance anchor-free YOLO, exceeding yolov3~v5 with MegEngine, ONNX, TensorRT, ncnn, and OpenVINO supported. Documentation: https://yolox.readthedocs.io/
OpenMMLab's next-generation platform for general 3D object detection.
PointNet: Deep Learning on Point Sets for 3D Classification and Segmentation
OpenPCDet Toolbox for LiDAR-based 3D Object Detection.
An easy to use PyTorch to TensorRT converter
PyTorch ,ONNX and TensorRT implementation of YOLOv4
dataset and code for 2016 paper "Learning a Driving Simulator"
Simple, online, and realtime tracking of multiple objects in a video sequence.
PointNet and PointNet++ implemented by pytorch (pure python) and on ModelNet, ShapeNet and S3DIS.
[ECCV 2022] This is the official implementation of BEVFormer, a camera-only framework for autonomous driving perception, e.g., 3D object detection and semantic map segmentation.
π₯π₯π₯π₯ (Earlier YOLOv7 not official one) YOLO with Transformers and Instance Segmentation, with TensorRT acceleration! π₯π₯π₯
A Simple and Versatile Framework for Object Detection and Instance Recognition
Jittor is a high-performance deep learning framework based on JIT compiling and meta-operators.
OpenMMLab YOLO series toolbox and benchmark. Implemented RTMDet, RTMDet-Rotated,YOLOv5, YOLOv6, YOLOv7, YOLOv8,YOLOX, PPYOLOE, etc.
[ICRA'23] BEVFusion: Multi-Task Multi-Sensor Fusion with Unified Bird's-Eye View Representation