Stars
Stable Diffusion web UI
The largest collection of PyTorch image encoders / backbones. Including train, eval, inference, export scripts, and pretrained weights -- ResNet, ResNeXT, EfficientNet, NFNet, Vision Transformer (V…
Interactive deep learning book with multi-framework code, math, and discussions. Adopted at 500 universities from 70 countries including Stanford, MIT, Harvard, and Cambridge.
Image-to-Image Translation in PyTorch
Industry leading face manipulation platform
PyTorch implementations of Generative Adversarial Networks.
Qlib is an AI-oriented quantitative investment platform that aims to realize the potential, empower research, and create value using AI technologies in quantitative investment, from exploring ideas…
Advanced AI Explainability for computer vision. Support for CNNs, Vision Transformers, Classification, Object detection, Segmentation, Image similarity and more.
The AI developer platform. Use Weights & Biases to train and fine-tune models, and manage models from experimentation to production.
PySlowFast: video understanding codebase from FAIR for reproducing state-of-the-art video models.
Deformable DETR: Deformable Transformers for End-to-End Object Detection.
Towhee is a framework that is dedicated to making neural data processing pipelines simple and fast.
Python Library for learning (Structure and Parameter), inference (Probabilistic and Causal), and simulations in Bayesian Networks.
PyTorch implementation of TabNet paper : https://arxiv.org/pdf/1908.07442.pdf
Code release for "Masked-attention Mask Transformer for Universal Image Segmentation"
A pytorch-based deep learning framework for multi-modal 2D/3D medical image segmentation
Pythonic Bayesian Belief Network Package, supporting creation of and exact inference on Bayesian Belief Networks specified as pure python functions.
Research on Tabular Deep Learning: Papers & Packages
FFA-Net: Feature Fusion Attention Network for Single Image Dehazing
Code Release for MViTv2 on Image Recognition.
[IEEE TIP] Vision Transformers for Single Image Dehazing
Rethinking Image Inpainting via a Mutual Encoder Decoder with Feature Equalizations. ECCV 2020 Oral
Contrastive Learning for Compact Single Image Dehazing, CVPR2021
The source code of CVPR 2020 paper "Multi-Scale Boosted Dehazing Network with Dense Feature Fusion"
Single Image Haze Removal Using Dark Channel Prior
no reference image/video quaity assessment(BRISQUE/NIQE/PIQE/DIQA/deepBIQ/VSFA