Lists (1)
Sort Name ascending (A-Z)
Stars
🧑🏫 60+ Implementations/tutorials of deep learning papers with side-by-side notes 📝; including transformers (original, xl, switch, feedback, vit, ...), optimizers (adam, adabelief, sophia, ...), ga…
YOLOv5 🚀 in PyTorch > ONNX > CoreML > TFLite
GFPGAN aims at developing Practical Algorithms for Real-world Face Restoration.
The largest collection of PyTorch image encoders / backbones. Including train, eval, inference, export scripts, and pretrained weights -- ResNet, ResNeXT, EfficientNet, NFNet, Vision Transformer (V…
State-of-the-art 2D and 3D Face Analysis Project
PyTorch implementations of Generative Adversarial Networks.
Python package built to ease deep learning on graph, on top of existing DL frameworks.
PyTorch implementation of MAE https//arxiv.org/abs/2111.06377
BoxMOT: pluggable SOTA tracking modules for segmentation, object detection and pose estimation models
PyTorch code for Vision Transformers training with the Self-Supervised learning method DINO
PyTorch implementation of MoCo: https://arxiv.org/abs/1911.05722
A Tensorflow implementation of AnimeGAN for fast photo animation ! This is the Open source of the paper 「AnimeGAN: a novel lightweight GAN for photo animation」, which uses the GAN framwork to trans…
Torchreid: Deep learning person re-identification in PyTorch.
Efficient AI Backbones including GhostNet, TNT and MLP, developed by Huawei Noah's Ark Lab.
DeepFill v1/v2 with Contextual Attention and Gated Convolution, CVPR 2018, and ICCV 2019 Oral
[ICML 2024] Vision Mamba: Efficient Visual Representation Learning with Bidirectional State Space Model
PyTorch implementation of "Supervised Contrastive Learning" (and SimCLR incidentally)
MOT using deepsort and yolov3 with pytorch
Just playing with getting VQGAN+CLIP running locally, rather than having to use colab.
EdgeConnect: Structure Guided Image Inpainting using Edge Prediction, ICCV 2019 https://arxiv.org/abs/1901.00212
VMamba: Visual State Space Models,code is based on mamba
Codebase for Image Classification Research, written in PyTorch.
这是一个YoloV4-pytorch的源码,可以用于训练自己的模型。
MMGeneration is a powerful toolkit for generative models, based on PyTorch and MMCV.
mobilenetv3 with pytorch,provide pre-train model
yolov5 车牌检测 车牌识别 中文车牌识别 检测 支持12种中文车牌 支持双层车牌
solo-learn: a library of self-supervised methods for visual representation learning powered by Pytorch Lightning