scikit-learn: machine learning in Python

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…

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

OpenMMLab Detection Toolbox and Benchmark

Easily train a good VC model with voice data <= 10 mins!

🤗 Diffusers: State-of-the-art diffusion models for image, video, and audio generation in PyTorch and FLAX.

A Lightweight Face Recognition and Facial Attribute Analysis (Age, Gender, Emotion and Race) Library for Python

An advanced Twitter scraping & OSINT tool written in Python that doesn't use Twitter's API, allowing you to scrape a user's followers, following, Tweets and more while evading most API limitations.

Chinese version of GPT2 training code, using BERT tokenizer.

DiffSinger: Singing Voice Synthesis via Shallow Diffusion Mechanism (SVS & TTS); AAAI 2022; Official code

📈 目前最大的工业缺陷检测数据库及论文集 Constantly summarizing open source dataset and critical papers in the field of surface defect research which are of great importance.

TextAttack 🐙 is a Python framework for adversarial attacks, data augmentation, and model training in NLP https://textattack.readthedocs.io/en/master/

A large-scale face dataset for face parsing, recognition, generation and editing.

HiFi-GAN: Generative Adversarial Networks for Efficient and High Fidelity Speech Synthesis

EfficientDet (Scalable and Efficient Object Detection) implementation in Keras and Tensorflow

A Python framework for sequence labeling evaluation(named-entity recognition, pos tagging, etc...)

MobileNetV3 in pytorch and ImageNet pretrained models

MusicTransformer written for MaestroV2 using the Pytorch framework for music generation

Local windowed attention multi-instrumental music transformer for supervised music generation

Tensorflow 2 single shot multibox detector (SSD) implementation from scratch with MobileNetV2 and VGG16 backbones

Code for "Dual Focal Loss for Calibration" (ICML 2023)

Surface Defect Detection as a Tensorflow/Keras model microservice container.