Models and examples built with TensorFlow

YOLOv5 🚀 in PyTorch > ONNX > CoreML > TFLite

Making large AI models cheaper, faster and more accessible

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

Mask R-CNN for object detection and instance segmentation on Keras and TensorFlow

A set of examples around pytorch in Vision, Text, Reinforcement Learning, etc.

Implementation of Vision Transformer, a simple way to achieve SOTA in vision classification with only a single transformer encoder, in Pytorch

Use ChatGPT to summarize the arXiv papers. 全流程加速科研，利用chatgpt进行论文全文总结+专业翻译+润色+审稿+审稿回复

This is an official implementation for "Swin Transformer: Hierarchical Vision Transformer using Shifted Windows".

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

OpenMMLab Semantic Segmentation Toolbox and Benchmark.

PyTorch implementation of MAE https//arxiv.org/abs/2111.06377

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

Deep Learning and Reinforcement Learning Library for Scientists and Engineers

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 …

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

Google AI 2018 BERT pytorch implementation

95.47% on CIFAR10 with PyTorch

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

A PyTorch Implementation of Single Shot MultiBox Detector

Count the MACs / FLOPs of your PyTorch model.

Differentiable architecture search for convolutional and recurrent networks

OpenMMLab Pre-training Toolbox and Benchmark

RepVGG: Making VGG-style ConvNets Great Again

Image Restoration Toolbox (PyTorch). Training and testing codes for DPIR, USRNet, DnCNN, FFDNet, SRMD, DPSR, BSRGAN, SwinIR

FFCV: Fast Forward Computer Vision (and other ML workloads!)

Summary of related papers on visual attention. Related code will be released based on Jittor gradually.

The state-of-the-art image restoration model without nonlinear activation functions.