Tensors and Dynamic neural networks in Python with strong GPU acceleration

YOLOv5 🚀 in PyTorch > ONNX > CoreML > TFLite

Deep Learning papers reading roadmap for anyone who are eager to learn this amazing tech!

Ultralytics YOLO11 🚀

The largest collection of PyTorch image encoders / backbones. Including train, eval, inference, export scripts, and pretrained weights -- ResNet, ResNeXT, EfficientNet, NFNet, Vision Transformer (V…

Detectron2 is a platform for object detection, segmentation and other visual recognition tasks.

OpenMMLab Detection Toolbox and Benchmark

Real-ESRGAN aims at developing Practical Algorithms for General Image/Video Restoration.

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

Open standard for machine learning interoperability

Fast and flexible image augmentation library. Paper about the library: https://www.mdpi.com/2078-2489/11/2/125

Image augmentation for machine learning experiments.

Image Polygonal Annotation with Python (polygon, rectangle, circle, line, point and image-level flag annotation).

Transfer learning / domain adaptation / domain generalization / multi-task learning etc. Papers, codes, datasets, applications, tutorials.-迁移学习

🍀 Pytorch implementation of various Attention Mechanisms, MLP, Re-parameter, Convolution, which is helpful to further understand papers.⭐⭐⭐

A paper list of object detection using deep learning.

Advanced AI Explainability for computer vision. Support for CNNs, Vision Transformers, Classification, Object detection, Segmentation, Image similarity and more.

Semantic segmentation models with 500+ pretrained convolutional and transformer-based backbones.

YOLOX is a high-performance anchor-free YOLO, exceeding yolov3~v5 with MegEngine, ONNX, TensorRT, ncnn, and OpenVINO supported. Documentation: https://yolox.readthedocs.io/

Pretrained ConvNets for pytorch: NASNet, ResNeXt, ResNet, InceptionV4, InceptionResnetV2, Xception, DPN, etc.

A python library for user-friendly forecasting and anomaly detection on time series.

A PyTorch implementation of EfficientNet

A Library for Advanced Deep Time Series Models.

Object detection, 3D detection, and pose estimation using center point detection:

Open Source Image and Video Restoration Toolbox for Super-resolution, Denoise, Deblurring, etc. Currently, it includes EDSR, RCAN, SRResNet, SRGAN, ESRGAN, EDVR, BasicVSR, SwinIR, ECBSR, etc. Also …

BoxMOT: pluggable SOTA tracking modules for segmentation, object detection and pose estimation models

PyQt Examples（PyQt各种测试和例子） PyQt4 PyQt5

PySlowFast: video understanding codebase from FAIR for reproducing state-of-the-art video models.

NanoDet-Plus⚡Super fast and lightweight anchor-free object detection model. 🔥Only 980 KB(int8) / 1.8MB (fp16) and run 97FPS on cellphone🔥